getList and small improv

api/v1/_getList.py

@@ -1,31 +0,0 @@
-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",
-                "allowed": allowed_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
-
-    return callback(method_name)
-
-def callback(method):
-    cur = glob.sql.cursor()
-
-    cur.execute("SELECT {} FROM updates WHERE filename = 'osu!.exe' ORDER BY file_version".format(
-        method
-    ))
-
-    data = []
-    for row in cur.fetchall():
-        data.append(row[method])
-
-    return data

api/v1/getFile.py

@@ -1,3 +1,4 @@
+import traceback
 import json
 
 import tornado.gen
@@ -27,7 +28,7 @@ class handler(asyncTornado.asyncRequestHandler):
 		data = {}
 		try:
 			args_filter = asyncTornado.check_arguments(self.request.arguments, ARGS)
-			if False in args_filter:
+			if not args_filter:
 				raise Exception("Missing required arguments")
 
 			method = args_filter[0]
@@ -54,6 +55,7 @@ class handler(asyncTornado.asyncRequestHandler):
 			data["status"] = status_code
 			data["message"] = str(e)
 			print("Err: %s" % e)
+			traceback.print_exc()
 		finally:
 			self.write( json.dumps(data) )
 			self.set_header("Content-Type", "application/json")

api/v1/getList.py

@@ -0,0 +1,52 @@
+import traceback
+import json
+
+import tornado.gen
+import tornado.web
+
+from web import asyncTornado
+from constants import argumentTypes
+
+from objects import glob
+
+ARGS = {
+	("file_hash", "file_version", "timestamp"): argumentTypes.one_required
+}
+
+SQL_STRUCT = {
+	"main": "SELECT %s FROM updates WHERE filename = 'osu!.exe' ORDER BY file_version"
+}
+
+class handler(asyncTornado.asyncRequestHandler):
+	@tornado.web.asynchronous
+	@tornado.gen.engine
+	def asyncGet(self):
+		status_code = 400
+		data = {}
+		try:
+			args_filter = asyncTornado.check_arguments(self.request.arguments, ARGS)
+			if not args_filter:
+				raise Exception("Missing required arguments")
+
+			conn = glob.new_sql()
+			cur = conn.cursor()
+
+			sql = SQL_STRUCT["main"]
+
+			cur.execute(sql % ",".join(args_filter))
+			data = cur.fetchall()
+
+			cur.close()
+			conn.close()
+
+			status_code = 200
+		except Exception as e:
+			status_code = 400
+			data["status"] = status_code
+			data["message"] = str(e)
+			print("Err: %s" % e)
+			traceback.print_exc()
+		finally:
+			self.write( json.dumps(data) )
+			self.set_header("Content-Type", "application/json")
+			self.set_status(status_code)

api/v1/getUpdate.py

@@ -27,7 +27,7 @@ class handler(asyncTornado.asyncRequestHandler):
 		data = {}
 		try:
 			args_filter = asyncTornado.check_arguments(self.request.arguments, ARGS)
-			if False in args_filter:
+			if not args_filter:
 				raise Exception("Missing required arguments")
 
 			method = args_filter[0]