Triplehx3/contentdb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add YouTube thumbnail support

Browse Source 
Fixes #359
This commit is contained in:
rubenwardy
2024-06-07 06:25:32 +01:00
parent ee83a7b5ce
commit 5bd97598a8
6 changed files with 67 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/utils/url.py
View File

@@ -31,16 +31,22 @@ def url_get_query(parsed_url: urlparse.ParseResult) -> Dict[str, List[str]]:
return urlparse.parse_qs(parsed_url.query)
def clean_youtube_url(url: str) -> Optional[str]:
def get_youtube_id(url: str) -> Optional[str]:
parsed = urlparse.urlparse(url)
print(parsed)
if (parsed.netloc == "www.youtube.com" or parsed.netloc == "youtube.com") and parsed.path == "/watch":
print(url_get_query(parsed))
video_id = url_get_query(parsed).get("v", [None])[0]
if video_id:
return url_set_query("https://www.youtube.com/watch", {"v": video_id})
return video_id
elif parsed.netloc == "youtu.be":
return url_set_query("https://www.youtube.com/watch", {"v": parsed.path[1:]})
return parsed.path[1:]
return None
def clean_youtube_url(url: str) -> Optional[str]:
id_ = get_youtube_id(url)
if id_:
return url_set_query("https://www.youtube.com/watch", {"v": id_})
return None