refactor(Klipper): change subprocess imports

Signed-off-by: Dominik Willner <th33xitus@gmail.com>
This commit is contained in:
dw-0
2024-05-11 22:49:26 +02:00
parent 7f79f68209
commit fc8fedc9f6

View File

@@ -11,21 +11,21 @@ import grp
import os import os
import re import re
import shutil import shutil
import subprocess from subprocess import CalledProcessError, run
from typing import List, Union, Literal, Dict, Optional from typing import Dict, List, Literal, Optional, Union
from components.klipper import ( from components.klipper import (
MODULE_PATH, KLIPPER_BACKUP_DIR,
KLIPPER_DIR, KLIPPER_DIR,
KLIPPER_ENV_DIR, KLIPPER_ENV_DIR,
KLIPPER_BACKUP_DIR, MODULE_PATH,
) )
from components.klipper.klipper import Klipper from components.klipper.klipper import Klipper
from components.klipper.klipper_dialogs import ( from components.klipper.klipper_dialogs import (
print_missing_usergroup_dialog,
print_instance_overview, print_instance_overview,
print_select_instance_count_dialog, print_missing_usergroup_dialog,
print_select_custom_name_dialog, print_select_custom_name_dialog,
print_select_instance_count_dialog,
) )
from components.moonraker.moonraker import Moonraker from components.moonraker.moonraker import Moonraker
from components.moonraker.moonraker_utils import moonraker_to_multi_conversion from components.moonraker.moonraker_utils import moonraker_to_multi_conversion
@@ -41,9 +41,9 @@ from core.instance_manager.name_scheme import NameScheme
from utils import PRINTER_CFG_BACKUP_DIR from utils import PRINTER_CFG_BACKUP_DIR
from utils.common import get_install_status_common from utils.common import get_install_status_common
from utils.constants import CURRENT_USER from utils.constants import CURRENT_USER
from utils.git_utils import get_repo_name, get_remote_commit, get_local_commit from utils.git_utils import get_local_commit, get_remote_commit, get_repo_name
from utils.input_utils import get_confirm, get_string_input, get_number_input from utils.input_utils import get_confirm, get_number_input, get_string_input
from utils.logger import Logger, DialogType from utils.logger import DialogType, Logger
from utils.sys_utils import cmd_sysctl_service from utils.sys_utils import cmd_sysctl_service
@@ -232,9 +232,9 @@ def check_user_groups():
for group in missing_groups: for group in missing_groups:
Logger.print_status(f"Adding user '{CURRENT_USER}' to group {group} ...") Logger.print_status(f"Adding user '{CURRENT_USER}' to group {group} ...")
command = ["sudo", "usermod", "-a", "-G", group, CURRENT_USER] command = ["sudo", "usermod", "-a", "-G", group, CURRENT_USER]
subprocess.run(command, check=True) run(command, check=True)
Logger.print_ok(f"Group {group} assigned to user '{CURRENT_USER}'.") Logger.print_ok(f"Group {group} assigned to user '{CURRENT_USER}'.")
except subprocess.CalledProcessError as e: except CalledProcessError as e:
Logger.print_error(f"Unable to add user to usergroups: {e}") Logger.print_error(f"Unable to add user to usergroups: {e}")
raise raise
@@ -246,13 +246,13 @@ def handle_disruptive_system_packages() -> None:
services = [] services = []
command = ["systemctl", "is-enabled", "brltty"] command = ["systemctl", "is-enabled", "brltty"]
brltty_status = subprocess.run(command, capture_output=True, text=True) brltty_status = run(command, capture_output=True, text=True)
command = ["systemctl", "is-enabled", "brltty-udev"] command = ["systemctl", "is-enabled", "brltty-udev"]
brltty_udev_status = subprocess.run(command, capture_output=True, text=True) brltty_udev_status = run(command, capture_output=True, text=True)
command = ["systemctl", "is-enabled", "ModemManager"] command = ["systemctl", "is-enabled", "ModemManager"]
modem_manager_status = subprocess.run(command, capture_output=True, text=True) modem_manager_status = run(command, capture_output=True, text=True)
if "enabled" in brltty_status.stdout: if "enabled" in brltty_status.stdout:
services.append("brltty") services.append("brltty")
@@ -264,7 +264,7 @@ def handle_disruptive_system_packages() -> None:
for service in services if services else []: for service in services if services else []:
try: try:
cmd_sysctl_service(service, "mask") cmd_sysctl_service(service, "mask")
except subprocess.CalledProcessError: except CalledProcessError:
Logger.print_dialog( Logger.print_dialog(
DialogType.WARNING, DialogType.WARNING,
[ [
@@ -286,8 +286,7 @@ def detect_name_scheme(instance_list: List[BaseInstance]) -> NameScheme:
def get_highest_index(instance_list: List[Klipper]) -> int: def get_highest_index(instance_list: List[Klipper]) -> int:
indices = [int(instance.suffix.split("-")[-1]) for instance in instance_list] return max([int(instance.suffix.split("-")[-1]) for instance in instance_list])
return max(indices)
def create_example_printer_cfg( def create_example_printer_cfg(