(Added) sync one function.

Added Rsync wrappper.
Added fzf_prompt, all_labels for choosing a label of source.
This commit is contained in:
2023-02-27 16:21:10 +02:00
parent b9d8131c3b
commit 467989bbd5
9 changed files with 126 additions and 4 deletions
+2
View File
@@ -1,3 +1,5 @@
from cli.cli import cli_app
from cli.source import source
from cli.sync import sync
from cli.rsync import Rsync
from cli.label_prompt import LabelPrompt
+2
View File
@@ -20,8 +20,10 @@
from rich.console import Console
import typer
from cli.source import source
from cli.sync import sync
console = Console()
cli_app = typer.Typer(rich_markup_mode="rich")
cli_app.add_typer(source, name="source", help="Manage sources")
cli_app.add_typer(sync, name="sync", help="Sync sources")
+6
View File
@@ -19,7 +19,9 @@
from rich.console import Console
from rich.prompt import Prompt
from pyfzf import FzfPrompt
import uuid
from models.models import all_labels
console = Console()
@@ -34,3 +36,7 @@ class LabelPrompt:
uid = uuid.uuid4().hex
label = Prompt.ask(question, default=uid)
return label
def get_label_fzf() -> str:
fzf = FzfPrompt()
return fzf.prompt(all_labels().iterator())
+31
View File
@@ -0,0 +1,31 @@
################################################################################
# Copyright (C) 2023 Kostiantyn Klochko <kostya_klochko@ukr.net> #
# #
# This file is part of tui-rsync. #
# #
# tui-rsync is free software: you can redistribute it and/or modify it under #
# uthe terms of the GNU General Public License as published by the Free #
# Software Foundation, either version 3 of the License, or (at your option) #
# any later version. #
# #
# tui-rsync is distributed in the hope that it will be useful, but WITHOUT ANY #
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS #
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more #
# details. #
# #
# You should have received a copy of the GNU General Public License along with #
# tui-rsync. If not, see <https://www.gnu.org/licenses/>. #
################################################################################
import shlex
from subprocess import Popen, PIPE
class Rsync:
def __init__(self, args:str):
self.__args = ["rsync"] + shlex.split(args)
def run_one(self, source, destination):
args = self.__args + [source, destination]
output = Popen(args, stdout=PIPE)
response = output.communicate()
+49
View File
@@ -0,0 +1,49 @@
################################################################################
# Copyright (C) 2023 Kostiantyn Klochko <kostya_klochko@ukr.net> #
# #
# This file is part of tui-rsync. #
# #
# tui-rsync is free software: you can redistribute it and/or modify it under #
# uthe terms of the GNU General Public License as published by the Free #
# Software Foundation, either version 3 of the License, or (at your option) #
# any later version. #
# #
# tui-rsync is distributed in the hope that it will be useful, but WITHOUT ANY #
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS #
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more #
# details. #
# #
# You should have received a copy of the GNU General Public License along with #
# tui-rsync. If not, see <https://www.gnu.org/licenses/>. #
################################################################################
from rich.console import Console
from rich.prompt import Prompt
from typing import List, Optional
import typer
from models.models import Source, Destination, SyncCommand, Path
from cli.label_prompt import LabelPrompt
from cli.rsync import Rsync
console = Console()
sync = typer.Typer()
@sync.command()
def one(
label: str = typer.Option(
None, "--label", "-l",
help="[b]The label[/] is a uniq identification of a [b]source[/].",
show_default=False
),
):
"""
[green b]Sync[/] a [yellow]source[/] with the [yellow b]label[/] and its backups.
"""
if label is None:
label = LabelPrompt.get_label_fzf()
src = Source.get(Source.label == label)
rsync = Rsync(str(src.args))
for dest in src.destinations:
rsync.run_one(str(src.source), str(dest))
+18 -3
View File
@@ -29,7 +29,7 @@ class Path(BaseModel):
path = CharField(unique=True)
def __str__(self) -> str:
return f"Path({self.path})"
return f"{self.path}"
def __repr__(self) -> str:
return f"Path({self.path})"
@@ -37,6 +37,9 @@ class Path(BaseModel):
class SyncCommand(BaseModel):
command = CharField()
def __str__(self) -> str:
return self.command
class Source(BaseModel):
label = CharField(unique=True)
source = ForeignKeyField(Path)
@@ -58,10 +61,22 @@ class Source(BaseModel):
src.save()
return src
def __str__(self) -> str:
return f"{self.label}"
def __repr__(self) -> str:
return f"{self.label}"
class Destination(BaseModel):
source = ForeignKeyField(Source, backref='destinations')
path = ForeignKeyField(Path)
def __str__(self) -> str:
return f"{self.path}"
def create_tables():
db.connect()
db.create_tables([Source, Path, Destination, SyncCommand], safe=True)
with db:
db.create_tables([Source, Path, Destination, SyncCommand], safe=True)
def all_labels():
with db:
return Source.select(Source.label)