mirror of
https://gitlab.com/KKlochko/tui-rsync.git
synced 2026-08-12 17:57:43 +00:00
(Added) sync one function.
Added Rsync wrappper. Added fzf_prompt, all_labels for choosing a label of source.
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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")
|
||||
|
||||
|
||||
@@ -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())
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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))
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user