Added getFile endpoint
This commit is contained in:
parent
8752f2a431
commit
fc3c0f44c5
37
handlers/file.py
Normal file
37
handlers/file.py
Normal file
|
@ -0,0 +1,37 @@
|
|||
from objects import glob
|
||||
|
||||
allowed_args = ["file_hash", "id", "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 "timestamp":
|
||||
cur.execute("SELECT * FROM updates WHERE timestamp <= '{}' ORDER BY timestamp DESC LIMIT 1".format(
|
||||
data
|
||||
))
|
||||
else:
|
||||
query = "SELECT * FROM updates WHERE {} = {} LIMIT 1"
|
||||
if method is "file_hash":
|
||||
query = "SELECT * FROM updates WHERE {} = '{}' LIMIT 1"
|
||||
cur.execute(query.format(
|
||||
method,
|
||||
data
|
||||
))
|
||||
|
||||
return cur.fetchone()
|
6
web.py
6
web.py
|
@ -2,7 +2,7 @@ import json
|
|||
import MySQLdb
|
||||
import MySQLdb.cursors
|
||||
from flask import Flask, make_response, request, render_template, jsonify
|
||||
from handlers import update
|
||||
from handlers import update, file
|
||||
from objects import glob
|
||||
|
||||
app = Flask(__name__)
|
||||
|
@ -30,5 +30,9 @@ def api_index():
|
|||
def api_update():
|
||||
return update.handle(request)
|
||||
|
||||
@app.route("/api/getFile", methods=["GET", "POST"])
|
||||
def api_file():
|
||||
return file.handle(request)
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(**config["web"])
|
Loading…
Reference in New Issue
Block a user