(Added) update labels option for groups.

dev
KKlochko 2 years ago
parent 5fd1ac1277
commit 9a40962408

@ -45,4 +45,6 @@
Added better label prompt for choosing labels. Added better label prompt for choosing labels.
** 0.8.9 <2023-03-25 Sat> ** 0.8.9 <2023-03-25 Sat>
Updated groups source creation. Updated groups source creation.
** 0.8.10 <2023-03-25 Sat>
Added *update labels* option for groups.

@ -0,0 +1,60 @@
################################################################################
# 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 tui_rsync.models.models import Group, GroupSource
from tui_rsync.cli.label_prompt import LabelPrompt
from tui_rsync.cli.groups.group_prompt import GroupPrompt
from typer.main import get_group
console = Console()
group_update = typer.Typer()
@group_update.command()
def labels(
group_label: str = typer.Option(
None, "--group-label", "-g",
help="[b]The label[/] is a uniq identification of a [b]group[/].",
show_default=False
),
new_labels: str = typer.Option(
None, "--new-labels", "-nl",
help="[b]The new label[/] will replace the [b]old source label[/].",
show_default=False
),
):
"""
[green b]Update[/] [yellow]the group[/] with a [bold]the label[/].
[b]The chosen sources[/] will be updated for [b]the group[/].
"""
if group_label is None:
group_label = GroupPrompt.get_label_fzf()
if new_labels is None:
new_labels = LabelPrompt.get_labels()
group = Group.get_group(group_label)
group.remove_sources()
GroupSource.create_group_sources(group, new_labels)
group.save()

@ -25,10 +25,12 @@ import typer
from tui_rsync.cli.label_prompt import LabelPrompt from tui_rsync.cli.label_prompt import LabelPrompt
from tui_rsync.cli.rsync import Rsync from tui_rsync.cli.rsync import Rsync
from tui_rsync.models.models import Group, count_all_labels_except from tui_rsync.models.models import Group, count_all_labels_except
from tui_rsync.cli.groups.group_update import group_update
from tui_rsync.cli.groups.group_remove import group_remove from tui_rsync.cli.groups.group_remove import group_remove
console = Console() console = Console()
groups = typer.Typer() groups = typer.Typer()
groups.add_typer(group_update, name="update", help="Update groups")
groups.add_typer(group_remove, name="remove", help="Remove groups") groups.add_typer(group_remove, name="remove", help="Remove groups")
@groups.command() @groups.command()

@ -146,6 +146,9 @@ class Group(BaseModel):
""" """
return (group_src.source for group_src in self.sources) return (group_src.source for group_src in self.sources)
def remove_sources(self):
GroupSource.delete().where(GroupSource.group == self.label).execute()
def __str__(self) -> str: def __str__(self) -> str:
return f"{self.label}" return f"{self.label}"

Loading…
Cancel
Save