Lomanic/NewLeaf
1
0
Fork 0
mirror of https://git.sr.ht/~cadence/NewLeaf synced 2024-11-21 23:27:29 +00:00
Code Issues Releases Wiki Activity

Captions: Error checking

Browse Source
This commit is contained in:
Cadence Ember 2021-01-20 17:35:24 +13:00
parent 8e69928756
commit c837828a22
No known key found for this signature in database
GPG Key ID: BC1C2C61CF521B17
2 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
extractors/captions.py
View File

@ -29,6 +29,12 @@ def extract_captions_from_video(id):
def extract_captions_from_api(id): def extract_captions_from_api(id):
url = "https://video.google.com/timedtext?hl=en&type=list&v=%s" % id url = "https://video.google.com/timedtext?hl=en&type=list&v=%s" % id
with requests.get(url) as r: with requests.get(url) as r:
if r.status_code == 404:
return {
"error": "Video unavailable",
"identifier": "NOT_FOUND"
}
r.raise_for_status() r.raise_for_status()
transcript = ET.fromstring(r.content.decode("utf8")) transcript = ET.fromstring(r.content.decode("utf8"))

24
index.py
View File

@ -92,17 +92,25 @@ class Second(object):
@cherrypy.tools.json_out() @cherrypy.tools.json_out()
def suggestions(self, *, q, **kwargs): def suggestions(self, *, q, **kwargs):
return extract_search_suggestions(q) return extract_search_suggestions(q)
@cherrypy.expose @cherrypy.expose
def captions(self, id, **kwargs): def captions(self, id, **kwargs):
result = extract_captions(id, **kwargs) try:
if type(result) is dict: result = extract_captions(id, **kwargs)
cherrypy.response.headers["content-type"] = "application/json" if type(result) is dict:
return bytes(json.dumps(result), "utf8") cherrypy.response.headers["content-type"] = "application/json"
else: return bytes(json.dumps(result), "utf8")
cherrypy.response.headers["content-type"] = "text/vtt; charset=UTF-8" else:
return result cherrypy.response.headers["content-type"] = "text/vtt; charset=UTF-8"
return result
except StopIteration:
cherrypy.response.status = "400"
cherrypy.response.headers["content-type"] = "application/json"
return bytes(json.dumps({
"error": "No captions matching that language or label",
"identifier": "NO_MATCHING_CAPTIONS"
}), "utf8")
@cherrypy.expose @cherrypy.expose
def vi(self, id, file): def vi(self, id, file):