mirror of https://gitlab.com/KKlochko/tui-rsync
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.9 KiB
71 lines
2.9 KiB
################################################################################
|
|
# 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 Confirm, Prompt, IntPrompt
|
|
from pyfzf import FzfPrompt
|
|
import uuid
|
|
from tui_rsync.models.models import all_labels, all_labels_except
|
|
from tui_rsync.models.models import count_all_labels_except
|
|
|
|
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
|
|
|
|
@staticmethod
|
|
def get_label_fzf() -> str:
|
|
fzf = FzfPrompt()
|
|
return fzf.prompt(all_labels().iterator())[0]
|
|
|
|
@staticmethod
|
|
def get_label_except_fzf(labels = None) -> str:
|
|
fzf = FzfPrompt()
|
|
return fzf.prompt(all_labels_except(labels).iterator())[0]
|
|
|
|
@staticmethod
|
|
def get_labels(labels = None) -> list:
|
|
confirm_question = "Would you like to add a source/sources to the group?"
|
|
is_fzf = Confirm.ask(confirm_question, default=True)
|
|
|
|
if not is_fzf:
|
|
return []
|
|
|
|
count_question = "How much would you like to add sources to the group?"
|
|
count = IntPrompt.ask(count_question, default=1)
|
|
|
|
count_max = count_all_labels_except(labels)
|
|
count = count_max if count > count_max else count
|
|
|
|
labels = []
|
|
for i in range(count):
|
|
option = LabelPrompt.get_label_except_fzf(labels)
|
|
labels.append(option)
|
|
|
|
return labels
|