fixes in regex in name_normalize

2 years ago · 30ad5a510d
parent 1764d54e5d
commit 30ad5a510d
1 changed files with 5 additions and 2 deletions
--- a/video_creation/final_video.py
+++ b/video_creation/final_video.py
@ -21,8 +21,11 @@ def name_normalize(
        name: str
 ) -> str:
    name = re.sub(r'[?\\"%*:|<>]', '', name)
-    name = re.sub(r'(\D+)/(\D+)', r'\1 or\ \2', name)
-    name = re.sub(r'(\d+)/(\d+)', r'\1 of\ \2', name)
+    name = re.sub(r'( [w,W]\s?\/\s?[o,O,0])', r' without', name)
+    name = re.sub(r'( [w,W]\s?\/)', r' with', name)
+    name = re.sub(r'([0-9]+)\s?\/\s?([0-9]+)', r'\1 of \2', name)
+    name = re.sub(r'(\w+)\s?\/\s?(\w+)', r'\1 or \2', name)
+    name = re.sub(r'\/', r'', name)
    return name