mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-15 03:24:29 +05:00
133 lines
4.9 KiB
Python
133 lines
4.9 KiB
Python
# ======================================================================= #
|
|
# Copyright (C) 2020 - 2025 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 typing import List
|
|
|
|
from components.moonraker.moonraker import Moonraker
|
|
from core.instance_manager.instance_manager import InstanceManager
|
|
from core.logger import DialogType, Logger
|
|
from core.services.backup_service import BackupService
|
|
from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
|
|
SimpleConfigParser,
|
|
)
|
|
from extensions.base_extension import BaseExtension
|
|
from utils.common import moonraker_exists
|
|
from utils.input_utils import get_confirm
|
|
|
|
|
|
# noinspection PyMethodMayBeStatic
|
|
class SimplyPrintExtension(BaseExtension):
|
|
def install_extension(self, **kwargs) -> None:
|
|
Logger.print_status("Installing SimplyPrint ...")
|
|
|
|
if not (mr_instances := moonraker_exists("SimplyPrint Installer")):
|
|
return
|
|
|
|
Logger.print_dialog(
|
|
DialogType.INFO,
|
|
self._construct_dialog(mr_instances, True),
|
|
)
|
|
|
|
if not get_confirm(
|
|
"Continue SimplyPrint installation?",
|
|
default_choice=True,
|
|
allow_go_back=True,
|
|
):
|
|
Logger.print_info("Exiting SimplyPrint installation ...")
|
|
return
|
|
|
|
try:
|
|
self._patch_moonraker_confs(mr_instances, True)
|
|
|
|
except Exception as e:
|
|
Logger.print_error(f"Error during SimplyPrint installation:\n{e}")
|
|
|
|
def remove_extension(self, **kwargs) -> None:
|
|
Logger.print_status("Removing SimplyPrint ...")
|
|
|
|
if not (mr_instances := moonraker_exists("SimplyPrint Uninstaller")):
|
|
return
|
|
|
|
Logger.print_dialog(
|
|
DialogType.INFO,
|
|
self._construct_dialog(mr_instances, False),
|
|
)
|
|
|
|
if not get_confirm(
|
|
"Do you really want to uninstall SimplyPrint?",
|
|
default_choice=True,
|
|
allow_go_back=True,
|
|
):
|
|
Logger.print_info("Exiting SimplyPrint uninstallation ...")
|
|
return
|
|
|
|
try:
|
|
self._patch_moonraker_confs(mr_instances, False)
|
|
|
|
except Exception as e:
|
|
Logger.print_error(f"Error during SimplyPrint installation:\n{e}")
|
|
|
|
def _construct_dialog(
|
|
self, mr_instances: List[Moonraker], is_install: bool
|
|
) -> List[str]:
|
|
mr_names = [f"● {m.service_file_path.name}" for m in mr_instances]
|
|
_type = "install" if is_install else "uninstall"
|
|
|
|
return [
|
|
"The following Moonraker instances were found:",
|
|
*mr_names,
|
|
"\n\n",
|
|
f"The setup will {_type} SimplyPrint for all Moonraker instances. "
|
|
f"After {_type}ation, all Moonraker services will be restarted!",
|
|
]
|
|
|
|
def _patch_moonraker_confs(
|
|
self, mr_instances: List[Moonraker], is_install: bool
|
|
) -> None:
|
|
section = "simplyprint"
|
|
_type, _ft = ("Adding", "to") if is_install else ("Removing", "from")
|
|
|
|
patched_files = []
|
|
for moonraker in mr_instances:
|
|
Logger.print_status(
|
|
f"{_type} section 'simplyprint' {_ft} {moonraker.cfg_file} ..."
|
|
)
|
|
scp = SimpleConfigParser()
|
|
scp.read_file(moonraker.cfg_file)
|
|
|
|
install_and_has_section = is_install and scp.has_section(section)
|
|
uninstall_and_has_no_section = not is_install and not scp.has_section(
|
|
section
|
|
)
|
|
|
|
if install_and_has_section or uninstall_and_has_no_section:
|
|
status = "already" if is_install else "does not"
|
|
Logger.print_info(
|
|
f"Section 'simplyprint' {status} exists! Skipping ..."
|
|
)
|
|
continue
|
|
|
|
if is_install and not scp.has_section("simplyprint"):
|
|
BackupService().backup_printer_config_dir()
|
|
scp.add_section(section)
|
|
elif not is_install and scp.has_section("simplyprint"):
|
|
BackupService().backup_printer_config_dir()
|
|
scp.remove_section(section)
|
|
scp.write_file(moonraker.cfg_file)
|
|
patched_files.append(moonraker.cfg_file)
|
|
|
|
if patched_files:
|
|
InstanceManager.restart_all(mr_instances)
|
|
|
|
install_state = "successfully" if patched_files else "was already"
|
|
Logger.print_dialog(
|
|
DialogType.SUCCESS,
|
|
[f"SimplyPrint {install_state} {'' if is_install else 'un'}installed!"],
|
|
center_content=True,
|
|
)
|