danvics/ytptube
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

FEAT: Enhance URL handling in Item class to support YouTube video IDs

Browse source
This commit is contained in:
arabcoders 2025-09-03 21:04:28 +03:00
parent 551daf6b70
commit ff9e173e18
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
app/library/ItemDTO.py
View file

@ -1,5 +1,6 @@
import json import json
import logging import logging
import re
import time import time
import uuid import uuid
from dataclasses import dataclass, field from dataclasses import dataclass, field
@ -115,6 +116,12 @@ class Item:
msg = "url param is required." msg = "url param is required."
raise ValueError(msg) raise ValueError(msg)
url = url.strip()
# If it's only a YouTube video ID, convert to a full URL.
if len(url) >= 11 and re.fullmatch(r"[A-Za-z0-9_-]{11}", url):
url = f"https://www.youtube.com/watch?v={url}"
data: dict[str, str] = {"url": url} data: dict[str, str] = {"url": url}
preset: str | None = item.get("preset") preset: str | None = item.get("preset")