afiqzudinhadi/TagStudio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor strip_web_protocol

Browse source 
This allows for more prefixes to be added in the future if needed without repeating code multiple times.
This commit is contained in:
JinguBangWest 2024-05-02 22:05:57 -04:00
parent ea8d954548
commit bb1161baa9
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tagstudio/src/core/utils/web.py
View file

@ -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