88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
#!/usr/bin/python3
|
|
# API is expected to return:
|
|
# - HTTP status code,
|
|
# - human-readable status message,
|
|
# - json with appropriate data
|
|
import flask, json, time
|
|
import ythdd_globals, ythdd_extractor
|
|
#from flask_sqlalchemy import SQLAlchemy
|
|
#import ythdd_api_v1_stats, ythdd_api_v1_user, ythdd_api_v1_info, ythdd_api_v1_query, ythdd_api_v1_meta, ythdd_api_v1_admin
|
|
|
|
def incrementBadRequests():
|
|
ythdd_globals.apiFailedRequests += 1
|
|
|
|
def notImplemented(data):
|
|
return 501, f"not recognised/implemented: {data[0]}", []
|
|
|
|
def stub_hello():
|
|
return 200, 'hello from v1!', []
|
|
|
|
def stats():
|
|
data_to_send = {
|
|
"starttime": ythdd_globals.starttime,
|
|
"uptime": ythdd_globals.getUptime(),
|
|
"total_api_requests": ythdd_globals.apiRequests,
|
|
"failed_api_requests": ythdd_globals.apiFailedRequests,
|
|
"outside_api_requests": ythdd_globals.outsideApiHits,
|
|
"local_api_requests": ythdd_globals.apiRequests - ythdd_globals.outsideApiHits
|
|
}
|
|
return 200, "OK", data_to_send
|
|
|
|
def hot(data):
|
|
if len(data) <= 3:
|
|
incrementBadRequests()
|
|
return 400, f'error: bad request. supply required arguments.', []
|
|
comment_count = ""
|
|
if data[1] not in ("video", "channel", "handle", "playlist"):
|
|
return notImplemented(data)
|
|
if data[2] not in ("c", "nc", "lc"): # comments, no comments, limited comments
|
|
return notImplemented(data)
|
|
if data[2] == "lc":
|
|
if len(data) <= 4:
|
|
incrementBadRequests()
|
|
return 400, f'error: bad request. limited comments (lc) requires an extra argument specifying amount of comments.', []
|
|
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]
|
|
try:
|
|
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='}
|
|
if data[2] == "nc":
|
|
getcomments = False
|
|
else:
|
|
getcomments = True
|
|
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}', []
|
|
|
|
def lookup(data):
|
|
match data[0]:
|
|
case 'stats':
|
|
return stats()
|
|
case 'hot': # retrieve live, uncached data
|
|
#print(data)
|
|
return hot(data)
|
|
case 'user':
|
|
return stub_hello()
|
|
#do_user()
|
|
case 'info':
|
|
return stub_hello()
|
|
#do_info()
|
|
case 'query':
|
|
return stub_hello()
|
|
case 'meta':
|
|
return stub_hello()
|
|
case 'admin':
|
|
# REQUIRE CREDENTIALS!
|
|
return stub_hello()
|
|
case _:
|
|
incrementBadRequests()
|
|
return notImplemented(data) |