sherl/ythdd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new boilerplate extractor for related videos

Browse Source
This commit is contained in:
sherl
2024-09-26 14:42:06 +02:00
parent 43c7b7c213
commit 24209a20a8
3 changed files with 113 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
ythdd_api_v1.py
View File

@@ -12,6 +12,7 @@ def incrementBadRequests():
ythdd_globals.apiFailedRequests += 1
def notImplemented(data):
# TODO: change list to string -> data, not data[0]
return 501, f"not recognised/implemented: {data[0]}", []
def stub_hello():
@@ -29,13 +30,75 @@ def stats():
return 200, "OK", data_to_send
def hot(data):
if len(data) <= 3:
#print(data)
if len(data) <= 2:
incrementBadRequests()
return 400, f'error: bad request. supply required arguments.', []
comment_count = ""
if data[1] not in ("video", "channel", "handle", "playlist"):
match data[1]:
case "video" | "channel" | "handle" | "playlist":
url_lookup = {'video': 'https://www.youtube.com/watch?v=', 'channel': 'https://www.youtube.com/channel/', 'handle': 'https://www.youtube.com/@', 'playlist': 'https://www.youtube.com/playlist?list='}
comment_count = ""
if len(data) <= 3:
incrementBadRequests()
return 400, f'error: bad request. supply required arguments.', []
if data[2] not in ("c", "nc", "lc"):
incrementBadRequests()
return notImplemented(data)
if data[2] == "lc" and len(data) <= 4:
incrementBadRequests()
return 400, f'error: bad request. limited comments (lc) requires an extra argument specifying amount of comments.', []
elif data[2] == "lc":
try:
comment_count = str(int(data[3]))
except:
incrementBadRequests()
return 400, f'error: bad request. {data[3]} is not a number.', []
videoId = data[4]
else:
videoId = data[3]
if len(videoId) != 11: # videoId sanity check
incrementBadRequests()
return 400, f'error: bad request. wrong videoId: {videoId} is {len(videoId)} characters long, but should be 11.', []
getcomments = True
if data[2] == "nc":
getcomments = False
try:
started = int(time.time())
extracted_dict = ythdd_extractor.extract(url_lookup[data[1]] + videoId, getcomments=getcomments, maxcomments=comment_count)
extracted_dict["took"] = int(time.time()) - started
return 200, "OK", extracted_dict
except Exception as e:
incrementBadRequests()
return 400, f'error: failed to get "{videoId}" ({data[2]}). {e}', []
case "related":
videoId = data[2]
if len(videoId) != 11: # videoId sanity check
incrementBadRequests()
return 400, f'error: bad request. wrong videoId: {videoId} is {len(videoId)} characters long, but should be 11.', []
started = int(time.time())
try:
extracted_related = ythdd_extractor.related('https://www.youtube.com/watch?v=' + videoId)
extracted_related['took'] = int(time.time()) - started
return 200, "OK", extracted_related
except Exception as e:
incrementBadRequests()
return 400, f'error: unknown error while parsing {videoId}: {e}', []
case _:
incrementBadRequests()
return notImplemented(data)
'''
if data[1] not in ("video", "channel", "handle", "playlist", "related"):
incrementBadRequests()
return notImplemented(data)
if data[2] not in ("c", "nc", "lc"): # comments, no comments, limited comments
incrementBadRequests()
return notImplemented(data)
if data[2] == "lc":
if len(data) <= 4:
@@ -62,6 +125,7 @@ def hot(data):
except Exception as e:
incrementBadRequests()
return 400, f'error: failed to get "{videoId}" ({data[2]}). {e}', []
'''
def lookup(data):
match data[0]: