From a1ca12c375db1ffbd7b71c1bb10752cb38e3d19d Mon Sep 17 00:00:00 2001 From: Michael Kuperfish Steinberg <36902556+Michael-K-Stein@users.noreply.github.com> Date: Fri, 20 Jan 2023 14:15:00 +0200 Subject: [PATCH] Fix illegal file path cleaning --- utils/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 13e5cb4..615eabf 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -1,4 +1,7 @@ +ILLEGAL_FILE_NAME_CHARACTERS = '/?"*|\\:;><#%{}$!\'@`=' def clean_file_path(prompt: str): - return prompt.replace('/', '').replace('?', '').replace('"', '').replace('*', '').replace('|', '').replace('\\', '').replace(':', '').replace(';', '').replace('>', '').replace('<', '') + for illegal_char in ILLEGAL_FILE_NAME_CHARACTERS: + prompt = prompt.replace(illegal_char, '') + return prompt