feat: rework completion message for webclient remove process

Signed-off-by: Dominik Willner <th33xitus@gmail.com>
This commit is contained in:
dw-0
2024-10-27 18:41:49 +01:00
parent 12127efa21
commit 1ca1e8ff6f
7 changed files with 128 additions and 69 deletions

View File

@@ -14,6 +14,8 @@ from components.klipper.klipper import Klipper
from components.moonraker.moonraker import Moonraker
from components.webui_client.base_data import BaseWebClientConfig
from core.logger import Logger
from core.services.message_service import Message
from core.types.color import Color
from utils.config_utils import remove_config_section
from utils.fs_utils import run_remove_routines
from utils.instance_utils import get_instances
@@ -23,21 +25,49 @@ def run_client_config_removal(
client_config: BaseWebClientConfig,
kl_instances: List[Klipper],
mr_instances: List[Moonraker],
) -> None:
remove_client_config_dir(client_config)
remove_client_config_symlink(client_config)
remove_config_section(f"update_manager {client_config.name}", mr_instances)
remove_config_section(client_config.config_section, kl_instances)
def remove_client_config_dir(client_config: BaseWebClientConfig) -> None:
) -> Message:
completion_msg = Message(
title=f"{client_config.display_name} Removal Process completed",
color=Color.GREEN,
)
Logger.print_status(f"Removing {client_config.display_name} ...")
run_remove_routines(client_config.config_dir)
if run_remove_routines(client_config.config_dir):
completion_msg.text.append(f"{client_config.display_name} removed")
def remove_client_config_symlink(client_config: BaseWebClientConfig) -> None:
instances: List[Klipper] = get_instances(Klipper)
handled_configs = []
for instance in instances:
run_remove_routines(
if run_remove_routines(
instance.base.cfg_dir.joinpath(client_config.config_filename)
):
handled_configs.append(instance)
if handled_configs:
instance_names = [i.service_file_path.stem for i in handled_configs]
completion_msg.text.append(
f"{client_config.display_name} removed from instances: {', '.join(instance_names)}"
)
mr_section = f"update_manager {client_config.name}"
handled_mr_instances = remove_config_section(mr_section, mr_instances)
if handled_mr_instances:
instance_names = [i.service_file_path.stem for i in handled_mr_instances]
completion_msg.text.append(
f"● Moonraker config section '{mr_section}' removed for instance: {', '.join(instance_names)}"
)
kl_section = client_config.config_section
handled_kl_instances = remove_config_section(kl_section, kl_instances)
if handled_kl_instances:
instance_names = [i.service_file_path.stem for i in handled_kl_instances]
completion_msg.text.append(
f"● Klipper config section '{mr_section}' removed for instance: {', '.join(instance_names)}"
)
if not completion_msg.text:
completion_msg.color = Color.YELLOW
completion_msg.centered = True
completion_msg.text.append("Nothing to remove.")
else:
completion_msg.text.insert(0, "The following actions were performed:")
return completion_msg