postgres version

server/api.py

@@ -0,0 +1,34 @@
+import time
+
+from flask import request, jsonify
+from __init__ import app
+import auth, database, reloader
+from models import FlagStatus
+from spam import is_spam_flag
+
+
+@app.route('/api/get_config')
+@auth.api_auth_required
+def get_config():
+    config = reloader.get_config()
+    return jsonify({key: value for key, value in config.items()
+                    if 'PASSWORD' not in key and 'TOKEN' not in key})
+
+
+@app.route('/api/post_flags', methods=['POST'])
+@auth.api_auth_required
+def post_flags():
+    flags = request.get_json()
+    flags = [item for item in flags if not is_spam_flag(item['flag'])]
+
+    cur_time = round(time.time())
+    rows = [(item['flag'], item['sploit'], item['team'], cur_time, FlagStatus.QUEUED.name)
+            for item in flags]
+
+    db = database.get()
+    cursor = db.cursor()
+    cursor.executemany("INSERT OR IGNORE INTO flags (flag, sploit, team, time, status) "
+                   "VALUES (%s, %s, %s, %s, %s)", rows)
+    db.commit()
+
+    return ''