From 75ac8a22d506d079775b36e4f78f978561e9cde6 Mon Sep 17 00:00:00 2001 From: dw-0 Date: Sun, 7 Jul 2024 17:25:08 +0200 Subject: [PATCH] refactor: add regex pattern to assign custom names Signed-off-by: Dominik Willner --- kiauh/components/klipper/klipper_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/kiauh/components/klipper/klipper_utils.py b/kiauh/components/klipper/klipper_utils.py index 3132ad1..98c0ce7 100644 --- a/kiauh/components/klipper/klipper_utils.py +++ b/kiauh/components/klipper/klipper_utils.py @@ -141,8 +141,10 @@ def assign_custom_name(key: int, name_dict: Dict[int, str]) -> None: existing_names = [] existing_names.extend(Klipper.blacklist()) existing_names.extend(name_dict[n] for n in name_dict) + pattern = r"^[a-zA-Z0-9]+$" + question = f"Enter name for instance {key + 1}" - name_dict[key] = get_string_input(question, exclude=existing_names) + name_dict[key] = get_string_input(question, exclude=existing_names, regex=pattern) def handle_to_multi_instance_conversion(new_name: str) -> None: