mirror of https://github.com/ekimekim/wubloader
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
3 years ago
|
import flask as flask
|
||
|
|
from common import dateutil
|
||
|
|
from dateutil.parser import ParserError
|
||
|
|
from flask import request
|
||
|
|
|
||
|
|
app = flask.Flask('buscribe')
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/buscribe/vtt')
|
||
|
|
def get_vtt():
|
||
|
|
"""Returns WebVTT subtitle file for the period between start_time and end_time."""
|
||
|
|
try:
|
||
|
|
start_time_string = request.args.get('start_time')
|
||
|
|
start_time = dateutil.parse(start_time_string)
|
||
|
|
except ParserError:
|
||
|
|
return "Invalid start time!", 400
|
||
|
|
except ValueError:
|
||
|
|
return "Missing start time!", 400
|
||
|
|
|
||
|
|
try:
|
||
|
|
end_time_string = request.args.get('end_time')
|
||
|
|
end_time = dateutil.parse(end_time_string)
|
||
|
|
except ParserError:
|
||
|
|
return "Invalid end time!", 400
|
||
|
|
except ValueError:
|
||
|
|
return "Missing end time!", 400
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/buscribe/json')
|
||
|
|
def get_json():
|
||
|
|
"""Searches the line database for *query*, with optional start_time and end_time boundaries.
|
||
|
|
|
||
|
|
Search is done using PostgreSQL websearch_to_tsquery() (https://www.postgresql.org/docs/13/functions-textsearch.html)"""
|
||
|
|
start_time_string = request.args.get('start_time')
|
||
|
|
try:
|
||
|
|
start_time = dateutil.parse(start_time_string)
|
||
|
|
except ParserError:
|
||
|
|
return "Invalid start time!", 400
|
||
|
|
|
||
|
|
end_time_string = request.args.get('end_time', default=None)
|
||
|
|
try:
|
||
|
|
end_time = dateutil.parse(end_time_string)
|
||
|
|
except ParserError:
|
||
|
|
return "Invalid end time!", 400
|
||
|
|
|
||
|
|
# I think websearch_to_tsquery() sanitizes its own input.
|
||
|
|
query = request.args.get('end_time', default=None)
|