Merge 9342c94096 into a374ac8fac

kiauh/components/klipper/klipper_setup.py

@@ -74,7 +74,6 @@ def install_klipper() -> None:
 
     try:
         if not kl_im.instances:
-            check_install_dependencies(["git"])
             setup_klipper_prerequesites()
 
         count = 0

kiauh/components/klipper/klipper_utils.py

@@ -11,21 +11,22 @@ import grp
 import os
 import re
 import shutil
-from subprocess import CalledProcessError, run
+import subprocess
-from typing import Dict, List, Literal, Optional, Union
+import textwrap
+from typing import List, Union, Literal, Dict, Optional
 
 from components.klipper import (
-    KLIPPER_BACKUP_DIR,
+    MODULE_PATH,
     KLIPPER_DIR,
     KLIPPER_ENV_DIR,
-    MODULE_PATH,
+    KLIPPER_BACKUP_DIR,
 )
 from components.klipper.klipper import Klipper
 from components.klipper.klipper_dialogs import (
-    print_instance_overview,
     print_missing_usergroup_dialog,
-    print_select_custom_name_dialog,
+    print_instance_overview,
     print_select_instance_count_dialog,
+    print_select_custom_name_dialog,
 )
 from components.moonraker.moonraker import Moonraker
 from components.moonraker.moonraker_utils import moonraker_to_multi_conversion
@@ -41,9 +42,9 @@ from core.instance_manager.name_scheme import NameScheme
 from utils import PRINTER_CFG_BACKUP_DIR
 from utils.common import get_install_status_common
 from utils.constants import CURRENT_USER
-from utils.git_utils import get_local_commit, get_remote_commit, get_repo_name
+from utils.git_utils import get_repo_name, get_remote_commit, get_local_commit
-from utils.input_utils import get_confirm, get_number_input, get_string_input
+from utils.input_utils import get_confirm, get_string_input, get_number_input
-from utils.logger import DialogType, Logger
+from utils.logger import Logger
 from utils.sys_utils import cmd_sysctl_service
 
 
@@ -97,9 +98,8 @@ def update_name_scheme(
     klipper_instances: List[Klipper],
     moonraker_instances: List[Moonraker],
 ) -> NameScheme:
-    # if there are more moonraker instances installed
+    # if there are more moonraker instances installed than klipper, we
-    # than klipper, we load their names into the name_dict,
+    # load their names into the name_dict, as we will detect and enforce that naming scheme
-    # as we will detect and enforce that naming scheme
     if len(moonraker_instances) > len(klipper_instances):
         update_name_dict(name_dict, moonraker_instances)
         return detect_name_scheme(moonraker_instances)
@@ -141,11 +141,7 @@ def get_install_count() -> Union[int, None]:
     """
     kl_instances = InstanceManager(Klipper).instances
     print_select_instance_count_dialog()
-    question = (
+    question = f"Number of{' additional' if len(kl_instances) > 0 else ''} Klipper instances to set up"
-        f"Number of"
-        f"{' additional' if len(kl_instances) > 0 else ''} "
-        f"Klipper instances to set up"
-    )
     return get_number_input(question, 1, default=1, allow_go_back=True)
 
 
@@ -197,8 +193,7 @@ def klipper_to_multi_conversion(new_name: str) -> None:
     else:
         Logger.print_info(f"Existing '{new_instance.data_dir}' found ...")
 
-    # patch the virtual_sdcard sections path
+    # patch the virtual_sdcard sections path value to match the new printer_data foldername
-    # value to match the new printer_data foldername
     cm = ConfigManager(new_instance.cfg_file)
     if cm.config.has_section("virtual_sdcard"):
         cm.set_value("virtual_sdcard", "path", str(new_instance.gcodes_dir))
@@ -232,9 +227,9 @@ def check_user_groups():
         for group in missing_groups:
             Logger.print_status(f"Adding user '{CURRENT_USER}' to group {group} ...")
             command = ["sudo", "usermod", "-a", "-G", group, CURRENT_USER]
-            run(command, check=True)
+            subprocess.run(command, check=True)
             Logger.print_ok(f"Group {group} assigned to user '{CURRENT_USER}'.")
-    except CalledProcessError as e:
+    except subprocess.CalledProcessError as e:
         Logger.print_error(f"Unable to add user to usergroups: {e}")
         raise
 
@@ -246,13 +241,13 @@ def handle_disruptive_system_packages() -> None:
     services = []
 
     command = ["systemctl", "is-enabled", "brltty"]
-    brltty_status = run(command, capture_output=True, text=True)
+    brltty_status = subprocess.run(command, capture_output=True, text=True)
 
     command = ["systemctl", "is-enabled", "brltty-udev"]
-    brltty_udev_status = run(command, capture_output=True, text=True)
+    brltty_udev_status = subprocess.run(command, capture_output=True, text=True)
 
     command = ["systemctl", "is-enabled", "ModemManager"]
-    modem_manager_status = run(command, capture_output=True, text=True)
+    modem_manager_status = subprocess.run(command, capture_output=True, text=True)
 
     if "enabled" in brltty_status.stdout:
         services.append("brltty")
@@ -264,16 +259,15 @@ def handle_disruptive_system_packages() -> None:
     for service in services if services else []:
         try:
             cmd_sysctl_service(service, "mask")
-        except CalledProcessError:
+        except subprocess.CalledProcessError:
-            Logger.print_dialog(
+            warn_msg = textwrap.dedent(
-                DialogType.WARNING,
+                f"""
-                [
+                KIAUH was unable to mask the {service} system service. 
-                    f"KIAUH was unable to mask the {service} system service. "
+                Please fix the problem manually. Otherwise, this may have 
-                    "Please fix the problem manually. Otherwise, this may have "
+                undesirable effects on the operation of Klipper.
-                    "undesirable effects on the operation of Klipper."
+                """
-                ],
+            )[1:]
-                end="",
+            Logger.print_warn(warn_msg)
-            )
 
 
 def detect_name_scheme(instance_list: List[BaseInstance]) -> NameScheme:
@@ -286,7 +280,8 @@ def detect_name_scheme(instance_list: List[BaseInstance]) -> NameScheme:
 
 
 def get_highest_index(instance_list: List[Klipper]) -> int:
-    return max([int(instance.suffix.split("-")[-1]) for instance in instance_list])
+    indices = [int(instance.suffix.split("-")[-1]) for instance in instance_list]
+    return max(indices)
 
 
 def create_example_printer_cfg(

kiauh/components/klipperscreen/klipperscreen.py

@@ -52,17 +52,14 @@ def install_klipperscreen() -> None:
     mr_im = InstanceManager(Moonraker)
     mr_instances = mr_im.instances
     if not mr_instances:
-        Logger.print_dialog(
+        warn_msg = [
-            DialogType.WARNING,
+            "Moonraker not found! KlipperScreen will not properly work "
-            [
+            "without a working Moonraker installation.",
-                "Moonraker not found! KlipperScreen will not properly work "
+            "\n\n",
-                "without a working Moonraker installation.",
+            "KlipperScreens update manager configuration for Moonraker "
-                "\n\n",
+            "will not be added to any moonraker.conf.",
-                "KlipperScreens update manager configuration for Moonraker "
+        ]
-                "will not be added to any moonraker.conf.",
+        Logger.print_dialog(DialogType.WARNING, warn_msg)
-            ],
-            end="",
-        )
         if not get_confirm(
             "Continue KlipperScreen installation?",
             default_choice=False,
@@ -70,7 +67,7 @@ def install_klipperscreen() -> None:
         ):
             return
 
-    package_list = ["git", "wget", "curl", "unzip", "dfu-util"]
+    package_list = ["wget", "curl", "unzip", "dfu-util"]
     check_install_dependencies(package_list)
 
     git_clone_wrapper(KLIPPERSCREEN_REPO, KLIPPERSCREEN_DIR)
@@ -83,8 +80,7 @@ def install_klipperscreen() -> None:
             mr_im.restart_all_instance()
         else:
             Logger.print_info(
-                "Moonraker is not installed! Cannot add "
+                "Moonraker is not installed! Cannot add KlipperScreen to update manager!"
-                "KlipperScreen to update manager!"
             )
         Logger.print_ok("KlipperScreen successfully installed!")
     except CalledProcessError as e:

kiauh/components/mobileraker/mobileraker.py

@@ -52,16 +52,13 @@ def install_mobileraker() -> None:
     mr_im = InstanceManager(Moonraker)
     mr_instances = mr_im.instances
     if not mr_instances:
-        Logger.print_dialog(
+        warn_msg = [
-            DialogType.WARNING,
+            "Moonraker not found! Mobileraker's companion will not properly work "
-            [
+            "without a working Moonraker installation.",
-                "Moonraker not found! Mobileraker's companion will not properly work "
+            "Mobileraker's companion's update manager configuration for Moonraker "
-                "without a working Moonraker installation.",
+            "will not be added to any moonraker.conf.",
-                "Mobileraker's companion's update manager configuration for Moonraker "
+        ]
-                "will not be added to any moonraker.conf.",
+        Logger.print_dialog(DialogType.WARNING, warn_msg)
-            ],
-            end="",
-        )
         if not get_confirm(
             "Continue Mobileraker's companion installation?",
             default_choice=False,
@@ -69,7 +66,7 @@ def install_mobileraker() -> None:
         ):
             return
 
-    package_list = ["git", "wget", "curl", "unzip", "dfu-util"]
+    package_list = ["wget", "curl", "unzip", "dfu-util"]
     check_install_dependencies(package_list)
 
     git_clone_wrapper(MOBILERAKER_REPO, MOBILERAKER_DIR)
@@ -82,8 +79,7 @@ def install_mobileraker() -> None:
             mr_im.restart_all_instance()
         else:
             Logger.print_info(
-                "Moonraker is not installed! Cannot add Mobileraker's "
+                "Moonraker is not installed! Cannot add Mobileraker's companion to update manager!"
-                "companion to update manager!"
             )
         Logger.print_ok("Mobileraker's companion successfully installed!")
     except CalledProcessError as e:

kiauh/components/moonraker/moonraker_setup.py

@@ -6,7 +6,7 @@
 #                                                                         #
 #  This file may be distributed under the terms of the GNU GPLv3 license  #
 # ======================================================================= #
-import json
+
 import subprocess
 from pathlib import Path
 
@@ -86,40 +86,32 @@ def install_moonraker() -> None:
         instance_names.append(klipper_instances[index].suffix)
 
     create_example_cfg = get_confirm("Create example moonraker.conf?")
+    setup_moonraker_prerequesites()
+    install_moonraker_polkit()
 
-    try:
+    used_ports_map = {
-        check_install_dependencies(["git"])
+        instance.suffix: instance.port for instance in moonraker_instances
-        setup_moonraker_prerequesites()
+    }
-        install_moonraker_polkit()
+    for name in instance_names:
+        current_instance = Moonraker(suffix=name)
 
-        used_ports_map = {
+        mr_im.current_instance = current_instance
-            instance.suffix: instance.port for instance in moonraker_instances
+        mr_im.create_instance()
-        }
+        mr_im.enable_instance()
-        for name in instance_names:
-            current_instance = Moonraker(suffix=name)
 
-            mr_im.current_instance = current_instance
+        if create_example_cfg:
-            mr_im.create_instance()
+            # if a webclient and/or it's config is installed, patch its update section to the config
-            mr_im.enable_instance()
+            clients = get_existing_clients()
+            create_example_moonraker_conf(current_instance, used_ports_map, clients)
 
-            if create_example_cfg:
+        mr_im.start_instance()
-                # if a webclient and/or it's config is installed, patch
-                # its update section to the config
-                clients = get_existing_clients()
-                create_example_moonraker_conf(current_instance, used_ports_map, clients)
 
-            mr_im.start_instance()
+    mr_im.reload_daemon()
 
-        mr_im.reload_daemon()
+    # if mainsail is installed, and we installed
-
+    # multiple moonraker instances, we enable mainsails remote mode
-        # if mainsail is installed, and we installed
+    if MainsailData().client_dir.exists() and len(mr_im.instances) > 1:
-        # multiple moonraker instances, we enable mainsails remote mode
+        enable_mainsail_remotemode()
-        if MainsailData().client_dir.exists() and len(mr_im.instances) > 1:
-            enable_mainsail_remotemode()
-
-    except Exception as e:
-        Logger.print_error(f"Error while installing Moonraker: {e}")
-        return
 
 
 def check_moonraker_install_requirements() -> bool:
@@ -148,19 +140,9 @@ def setup_moonraker_prerequesites() -> None:
 
 
 def install_moonraker_packages(moonraker_dir: Path) -> None:
-    install_script = moonraker_dir.joinpath("scripts/install-moonraker.sh")
+    script = moonraker_dir.joinpath("scripts/install-moonraker.sh")
-    deps_json = MOONRAKER_DIR.joinpath("scripts/system-dependencies.json")
+    packages = parse_packages_from_file(script)
-    moonraker_deps = []
+    check_install_dependencies(packages)
-
-    if deps_json.exists():
-        moonraker_deps = json.load(deps_json).get("debian", [])
-    elif install_script.exists():
-        moonraker_deps = parse_packages_from_file(install_script)
-
-    if not moonraker_deps:
-        raise ValueError("Error reading Moonraker dependencies!")
-
-    check_install_dependencies(moonraker_deps)
 
 
 def install_moonraker_polkit() -> None:

kiauh/components/webui_client/client_setup.py

@@ -119,7 +119,7 @@ def install_client(client: BaseWebClient) -> None:
         )
         valid_port = is_valid_port(port, ports_in_use)
 
-    check_install_dependencies(["nginx", "unzip"])
+    check_install_dependencies(["nginx"])
 
     try:
         download_client(client)

kiauh/core/menus/settings_menu.py

@@ -9,7 +9,7 @@
 import shutil
 import textwrap
 from pathlib import Path
-from typing import Optional, Tuple, Type
+from typing import Type, Optional, Tuple
 
 from components.klipper import KLIPPER_DIR
 from components.klipper.klipper import Klipper
@@ -19,10 +19,10 @@ from core.instance_manager.instance_manager import InstanceManager
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
 from core.settings.kiauh_settings import KiauhSettings
-from utils.constants import COLOR_CYAN, COLOR_GREEN, RESET_FORMAT
+from utils.constants import COLOR_CYAN, RESET_FORMAT, COLOR_GREEN, COLOR_YELLOW
 from utils.git_utils import git_clone_wrapper
-from utils.input_utils import get_confirm, get_string_input
+from utils.input_utils import get_string_input, get_confirm
-from utils.logger import DialogType, Logger
+from utils.logger import Logger
 
 
 # noinspection PyUnusedLocal
@@ -88,7 +88,7 @@ class SettingsMenu(BaseMenu):
             | 3) Toggle unstable Mainsail releases                  |
             | 4) Toggle unstable Fluidd releases                    |
             |                                                       |
-            | 5) Toggle automatic backups before updates            |
+            | 5) Toggle automatic backups before updates            |   
             """
         )[1:]
         print(menu, end="")
@@ -100,17 +100,12 @@ class SettingsMenu(BaseMenu):
         self._format_repo_str("moonraker")
 
         self.auto_backups_enabled = self.kiauh_settings.get(
-            "kiauh",
+            "kiauh", "backup_before_update"
-            "backup_before_update",
         )
         self.mainsail_unstable = self.kiauh_settings.get(
-            "mainsail",
+            "mainsail", "unstable_releases"
-            "unstable_releases",
-        )
-        self.fluidd_unstable = self.kiauh_settings.get(
-            "fluidd",
-            "unstable_releases",
         )
+        self.fluidd_unstable = self.kiauh_settings.get("fluidd", "unstable_releases")
 
     def _format_repo_str(self, repo_name: str) -> None:
         repo = self.kiauh_settings.get(repo_name, "repo_url")
@@ -120,16 +115,20 @@ class SettingsMenu(BaseMenu):
         setattr(self, f"{repo_name}_repo", f"{COLOR_CYAN}{repo}{RESET_FORMAT} {branch}")
 
     def _gather_input(self) -> Tuple[str, str]:
-        Logger.print_dialog(
+        l2 = "Make sure your input is correct!"
-            DialogType.ATTENTION,
+        error = textwrap.dedent(
-            [
+            f"""\n
-                "There is no input validation in place! Make sure your"
+            {COLOR_YELLOW}/=======================================================\\
-                " input is valid and has no typos! For any change to"
+            | ATTENTION:                                            |
-                " take effect, the repository must be cloned again. "
+            | There is no input validation in place! Make sure your |
-                "Make sure you don't have any ongoing prints running, "
+            | input is valid and has no typos! For any change to    | 
-                "as the services will be restarted!"
+            | take effect, the repository must be cloned again.     |
-            ],
+            | Make sure you don't have any ongoing prints running,  | 
-        )
+            | as the services will be restarted!                    |
+            \=======================================================/{RESET_FORMAT}
+            """
+        )[1:]
+        print(error, end="\n")
         repo = get_string_input(
             "Enter new repository URL",
             allow_special_chars=True,
@@ -141,35 +140,44 @@ class SettingsMenu(BaseMenu):
 
         return repo, branch
 
+    def _display_summary(self, name: str, repo: str, branch: str):
+        l1 = f"New {name} repository URL:"
+        l2 = f"● {repo}"
+        l3 = f"New {name} repository branch:"
+        l4 = f"● {branch}"
+        summary = textwrap.dedent(
+            f"""\n
+            /=======================================================\\
+            |  {l1:<52} |
+            |  {l2:<52} |
+            |  {l3:<52} |
+            |  {l4:<52} |
+            \=======================================================/
+            """
+        )[1:]
+        print(summary, end="")
+
     def _set_repo(self, repo_name: str):
-        repo_url, branch = self._gather_input()
+        repo, branch = self._gather_input()
-        display_name = repo_name.capitalize()
+        self._display_summary(repo_name.capitalize(), repo, branch)
-        Logger.print_dialog(
-            DialogType.CUSTOM,
-            [
-                f"New {display_name} repository URL:",
-                f"● {repo_url}",
-                f"New {display_name} repository branch:",
-                f"● {branch}",
-            ],
-            end="",
-        )
 
         if get_confirm("Apply changes?", allow_go_back=True):
-            self.kiauh_settings.set(repo_name, "repo_url", repo_url)
+            self.kiauh_settings.set(repo_name, "repo_url", repo)
             self.kiauh_settings.set(repo_name, "branch", branch)
             self.kiauh_settings.save()
             self._load_settings()
             Logger.print_ok("Changes saved!")
         else:
             Logger.print_info(
-                f"Skipping change of {display_name} source repository  ..."
+                f"Skipping change of {repo_name.capitalize()} source repository  ..."
             )
             return
 
-        Logger.print_status(f"Switching to {display_name}'s new source repository ...")
+        Logger.print_status(
+            f"Switching to {repo_name.capitalize()}'s new source repository ..."
+        )
         self._switch_repo(repo_name)
-        Logger.print_ok(f"Switched to {repo_url} at branch {branch}!")
+        Logger.print_ok(f"Switched to {repo} at branch {branch}!")
 
     def _switch_repo(self, name: str) -> None:
         target_dir: Path
@@ -203,27 +211,17 @@ class SettingsMenu(BaseMenu):
 
     def toggle_mainsail_release(self, **kwargs):
         self.mainsail_unstable = not self.mainsail_unstable
-        self.kiauh_settings.set(
+        self.kiauh_settings.set("mainsail", "unstable_releases", self.mainsail_unstable)
-            "mainsail",
-            "unstable_releases",
-            self.mainsail_unstable,
-        )
         self.kiauh_settings.save()
 
     def toggle_fluidd_release(self, **kwargs):
         self.fluidd_unstable = not self.fluidd_unstable
-        self.kiauh_settings.set(
+        self.kiauh_settings.set("fluidd", "unstable_releases", self.fluidd_unstable)
-            "fluidd",
-            "unstable_releases",
-            self.fluidd_unstable,
-        )
         self.kiauh_settings.save()
 
     def toggle_backup_before_update(self, **kwargs):
         self.auto_backups_enabled = not self.auto_backups_enabled
         self.kiauh_settings.set(
-            "kiauh",
+            "kiauh", "backup_before_update", self.auto_backups_enabled
-            "backup_before_update",
-            self.auto_backups_enabled,
         )
         self.kiauh_settings.save()

pyproject.toml

@@ -17,5 +17,3 @@ indent-style = "space"
 line-ending = "lf"
 quote-style = "double"
 
-[tool.ruff.lint]
-extend-select = ["I"]