osu-wayback/handlers/getupdate.py
2018-05-29 13:28:42 +02:00

34 lines
1.1 KiB
Python

from objects import glob
allowed_args = ["file_hash", "file_version", "timestamp"]
def handle(request):
if len([x for x in request.args if x in allowed_args]) == 0:
return {"error": "Missing valid args"}
for i in range(len(allowed_args)): # Gets the first valid argument and sets it as the method handler
method = request.args.get(allowed_args[i])
method_name = allowed_args[i]
if method is not None:
break
output = {
"method": method_name,
"response": callback(method_name, request.args.get(method_name))
}
return output
def callback(method, data):
cur = glob.sql.cursor()
if method is "file_hash":
cur.execute("SELECT * FROM updates WHERE file_hash = '{}'".format(
data
))
else:
cur.execute("SELECT a.* FROM updates a INNER JOIN ( SELECT MAX(file_version) file_version, filename FROM updates WHERE {} < {} GROUP BY filename) b ON a.file_version = b.file_version;".format(
method,
data
))
return cur.fetchall()