mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-11 17:44:28 +05:00
129 lines
6.2 KiB
Python
129 lines
6.2 KiB
Python
# ======================================================================= #
|
|
# Copyright (C) 2020 - 2024 Dominik Willner <th33xitus@gmail.com> #
|
|
# #
|
|
# This file is part of KIAUH - Klipper Installation And Update Helper #
|
|
# https://github.com/dw-0/kiauh #
|
|
# #
|
|
# This file may be distributed under the terms of the GNU GPLv3 license #
|
|
# ======================================================================= #
|
|
from __future__ import annotations
|
|
|
|
import textwrap
|
|
from typing import Type
|
|
|
|
from components.moonraker import moonraker_remove
|
|
from core.constants import COLOR_CYAN, COLOR_RED, RESET_FORMAT
|
|
from core.menus import Option
|
|
from core.menus.base_menu import BaseMenu
|
|
|
|
|
|
# noinspection PyUnusedLocal
|
|
class MoonrakerRemoveMenu(BaseMenu):
|
|
def __init__(self, previous_menu: Type[BaseMenu] | None = None):
|
|
super().__init__()
|
|
self.previous_menu: Type[BaseMenu] | None = previous_menu
|
|
self.remove_moonraker_service = False
|
|
self.remove_moonraker_dir = False
|
|
self.remove_moonraker_env = False
|
|
self.remove_moonraker_polkit = False
|
|
self.selection_state = False
|
|
|
|
def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
|
|
from core.menus.remove_menu import RemoveMenu
|
|
|
|
self.previous_menu = previous_menu if previous_menu is not None else RemoveMenu
|
|
|
|
def set_options(self) -> None:
|
|
self.options = {
|
|
"a": Option(method=self.toggle_all),
|
|
"1": Option(method=self.toggle_remove_moonraker_service),
|
|
"2": Option(method=self.toggle_remove_moonraker_dir),
|
|
"3": Option(method=self.toggle_remove_moonraker_env),
|
|
"4": Option(method=self.toggle_remove_moonraker_polkit),
|
|
"c": Option(method=self.run_removal_process),
|
|
}
|
|
|
|
def print_menu(self) -> None:
|
|
header = " [ Remove Moonraker ] "
|
|
color = COLOR_RED
|
|
count = 62 - len(color) - len(RESET_FORMAT)
|
|
checked = f"[{COLOR_CYAN}x{RESET_FORMAT}]"
|
|
unchecked = "[ ]"
|
|
o1 = checked if self.remove_moonraker_service else unchecked
|
|
o2 = checked if self.remove_moonraker_dir else unchecked
|
|
o3 = checked if self.remove_moonraker_env else unchecked
|
|
o4 = checked if self.remove_moonraker_polkit else unchecked
|
|
menu = textwrap.dedent(
|
|
f"""
|
|
╔═══════════════════════════════════════════════════════╗
|
|
║ {color}{header:~^{count}}{RESET_FORMAT} ║
|
|
╟───────────────────────────────────────────────────────╢
|
|
║ Enter a number and hit enter to select / deselect ║
|
|
║ the specific option for removal. ║
|
|
╟───────────────────────────────────────────────────────╢
|
|
║ a) {self._get_selection_state_str():37} ║
|
|
╟───────────────────────────────────────────────────────╢
|
|
║ 1) {o1} Remove Service ║
|
|
║ 2) {o2} Remove Local Repository ║
|
|
║ 3) {o3} Remove Python Environment ║
|
|
║ 4) {o4} Remove Policy Kit Rules ║
|
|
╟───────────────────────────────────────────────────────╢
|
|
║ C) Continue ║
|
|
╟───────────────────────────────────────────────────────╢
|
|
"""
|
|
)[1:]
|
|
print(menu, end="")
|
|
|
|
def toggle_all(self, **kwargs) -> None:
|
|
self.selection_state = not self.selection_state
|
|
self.remove_moonraker_service = self.selection_state
|
|
self.remove_moonraker_dir = self.selection_state
|
|
self.remove_moonraker_env = self.selection_state
|
|
self.remove_moonraker_polkit = self.selection_state
|
|
|
|
def toggle_remove_moonraker_service(self, **kwargs) -> None:
|
|
self.remove_moonraker_service = not self.remove_moonraker_service
|
|
|
|
def toggle_remove_moonraker_dir(self, **kwargs) -> None:
|
|
self.remove_moonraker_dir = not self.remove_moonraker_dir
|
|
|
|
def toggle_remove_moonraker_env(self, **kwargs) -> None:
|
|
self.remove_moonraker_env = not self.remove_moonraker_env
|
|
|
|
def toggle_remove_moonraker_polkit(self, **kwargs) -> None:
|
|
self.remove_moonraker_polkit = not self.remove_moonraker_polkit
|
|
|
|
def run_removal_process(self, **kwargs) -> None:
|
|
if (
|
|
not self.remove_moonraker_service
|
|
and not self.remove_moonraker_dir
|
|
and not self.remove_moonraker_env
|
|
and not self.remove_moonraker_polkit
|
|
):
|
|
error = f"{COLOR_RED}Nothing selected! Select options to remove first.{RESET_FORMAT}"
|
|
print(error)
|
|
return
|
|
|
|
moonraker_remove.run_moonraker_removal(
|
|
self.remove_moonraker_service,
|
|
self.remove_moonraker_dir,
|
|
self.remove_moonraker_env,
|
|
self.remove_moonraker_polkit,
|
|
)
|
|
|
|
self.remove_moonraker_service = False
|
|
self.remove_moonraker_dir = False
|
|
self.remove_moonraker_env = False
|
|
self.remove_moonraker_polkit = False
|
|
|
|
self._go_back()
|
|
|
|
def _get_selection_state_str(self) -> str:
|
|
return (
|
|
"Select everything" if not self.selection_state else "Deselect everything"
|
|
)
|
|
|
|
def _go_back(self, **kwargs) -> None:
|
|
if self.previous_menu is not None:
|
|
self.previous_menu().run()
|