mirror of
https://github.com/TagStudioDev/TagStudio.git
synced 2026-01-28 22:01:24 +00:00
Refactor strip_web_protocol
This allows for more prefixes to be added in the future if needed without repeating code multiple times.
This commit is contained in:
@@ -4,9 +4,8 @@
|
||||
|
||||
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:
|
||||
if string.startswith(prefix):
|
||||
string = string.replace(prefix, '')
|
||||
return string
|
||||
|
||||
Reference in New Issue
Block a user