Merge pull request #126 from JinguBangWest/main

Refactor strip_web_protocol()
TagStudioDev · May 4, 2024 · ac3d7c9 · ac3d7c9
2 parents 64514db + 58c773a
commit ac3d7c9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/tagstudio/src/core/utils/web.py b/tagstudio/src/core/utils/web.py
@@ -4,9 +4,7 @@
 
 def strip_web_protocol(string: str) -> str:
     """Strips a leading web protocol (ex. \"https://\") as well as \"www.\" from a string."""
-    new_str = string
-    new_str = new_str.removeprefix('https://')
-    new_str = new_str.removeprefix('http://')
-    new_str = new_str.removeprefix('www.')
-    new_str = new_str.removeprefix('www2.')
-    return new_str
+    prefixes = ['https://','http://','www.','www2.']
+    for prefix in prefixes:
+        string = string.removeprefix(prefix)
+    return string