KKlochko/tui-rsync
1
0
Fork 0
mirror of https://gitlab.com/KKlochko/tui-rsync.git synced 2026-08-12 17:57:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

First release

Browse Source 
Added source add function.
This commit is contained in:
KKlochko
2023-02-26 14:14:40 +02:00
commit b9d8131c3b
13 changed files with 1058 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tui_rsync/__init__.py
View File
tui_rsync/cli/__init__.py
+3
View File
@@ -0,0 +1,3 @@
from cli.cli import cli_app
from cli.source import source
from cli.label_prompt import LabelPrompt
tui_rsync/cli/cli.py
+27
View File
@@ -0,0 +1,27 @@
################################################################################
# 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
import typer
from cli.source import source
console = Console()
cli_app = typer.Typer(rich_markup_mode="rich")
cli_app.add_typer(source, name="source", help="Manage sources")
tui_rsync/cli/label_prompt.py
+36
View File
@@ -0,0 +1,36 @@
################################################################################
# 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
import uuid
console = Console()
class LabelPrompt:
@staticmethod
def ask_uuid(
question: str = "Would you like to change [yellow b]the label[/]?"
) -> str:
"""
Return the label or the default uuid value.
"""
uid = uuid.uuid4().hex
label = Prompt.ask(question, default=uid)
return label
tui_rsync/cli/source.py
+65
View File
@@ -0,0 +1,65 @@
################################################################################
# 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
console = Console()
source = typer.Typer()
@source.command()
def add(
label: str = typer.Option(
None, "--label", "-l",
help="[b]The label[/] is a uniq identification of a [b]source[/].",
show_default=False
),
source: str = typer.Option(
None, "--source", "-s",
help="[b]A source[/] of the data.",
show_default=False
),
destinations: Optional[List[str]] = typer.Option(
None, "--destination", "-d", help="[b]The backup[/] destinations.",
show_default=False
),
args: str = typer.Option(
None, "--args", "-a",
help="[b i yellow]Additional[/] rsync [b]arguments[/].",
show_default=False
)
):
"""
[green b]Create[/] a [yellow]new source[/] with a [bold]uniq[/] label.
[b]The source[/] will be connected to [b]backup destinations[/].
[yellow i]Optionally, additional[/] arguments for rsync can be added.g
"""
if label is None:
label = LabelPrompt.ask_uuid()
if source is None:
source = console.input("What is the [yellow b]path to the source[/]? ")
if args is None:
args = console.input("What is the [yellow b]rsync args of source[/]? ")
Source.create_save(label, source, destinations, args)
tui_rsync/main.py
+6
View File
@@ -0,0 +1,6 @@
from models.models import create_tables
from cli.cli import cli_app
if __name__ == "__main__":
create_tables()
cli_app()
tui_rsync/models/__init__.py
+5
View File
@@ -0,0 +1,5 @@
from models.models import Source
from models.models import Path
from models.models import Destination
from models.models import SyncCommand
from models.models import create_tables
tui_rsync/models/models.py
+67
View File
@@ -0,0 +1,67 @@
################################################################################
# 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 peewee import *
db = SqliteDatabase('sync.db')
class BaseModel(Model):
class Meta:
database = db
class Path(BaseModel):
path = CharField(unique=True)
def __str__(self) -> str:
return f"Path({self.path})"
def __repr__(self) -> str:
return f"Path({self.path})"
class SyncCommand(BaseModel):
command = CharField()
class Source(BaseModel):
label = CharField(unique=True)
source = ForeignKeyField(Path)
args = ForeignKeyField(SyncCommand)
@staticmethod
def create_save(label:str, source:str, destinations:list[str], args:str):
src_path, _ = Path.get_or_create(path=source)
src_sync_cmd, _ = SyncCommand.get_or_create(command=args)
src = Source.create(
label=label,
source=src_path,
destinations=[],
args=src_sync_cmd
)
for destination in destinations:
des_path, _ = Path.get_or_create(path=destination)
src_des, _ = Destination.get_or_create(source=src, path=des_path)
src.save()
return src
class Destination(BaseModel):
source = ForeignKeyField(Source, backref='destinations')
path = ForeignKeyField(Path)
def create_tables():
db.connect()
db.create_tables([Source, Path, Destination, SyncCommand], safe=True)