Functional PoC for web service

web_service/manage.py

@@ -14,8 +14,8 @@ api_key TEXT NOT NULL
 
 SQL_CREATE_REQUESTS = \
 """CREATE TABLE requests (
-id INTEGER PRIMARY KEY AUTOINCREMENT,
-username TEXT
+id TEXT PRIMARY KEY,
+username TEXT NOT NULL,
 depth INTEGER NOT NULL,
 width INTEGER NOT NULL,
 int_cycles INTEGER NOT NULL,

web_service/web_service.py

@@ -2,41 +2,68 @@
 import uuid
 from flask import Flask, request, jsonify
 import sqlite3
+import os
+import time
 
 app = Flask(__name__)
 
 DB_FILENAME = "db.sqlite"
-SQL_INSERT_REQUEST = "INSERT INTO requests ('username', 'depth', 'width', 'int_cycles', 'ext_cycles', 'status') VALUES ('{username}', '{depth}', '{width}', '{int_cycles}', '{ext_cycles}', '{status}');"
+SQL_INSERT_REQUEST = "INSERT INTO requests ('id', 'username', 'depth', 'width', 'int_cycles', 'ext_cycles', 'status') VALUES ('{request_id}', '{username}', '{depth}', '{width}', '{int_cycles}', '{ext_cycles}', '{status}');"
 SQL_SELECT_USERNAME = "SELECT username FROM users WHERE api_key='{api_key}'"
+SQL_SELECT_REQUEST_STATUS = "SELECT status FROM requests WHERE id='{request_id}'"
+SQL_UPDATE_STATUS = "UPDATE requests SET status='{status}' WHERE id='{request_id}'"
 
 
 def get_username_from_api_key(api_key):
     conn = sqlite3.connect(DB_FILENAME)
     c = conn.cursor()
-
     sql = SQL_SELECT_USERNAME.format(api_key=api_key)
-    return c.execute(sql)
+    res = c.execute(sql).fetchall()
+    conn.close()
+    return res[0][0] if res else None
+
+
+def get_request_status(request_id):
+    conn = sqlite3.connect(DB_FILENAME)
+    c = conn.cursor()
+    sql = SQL_SELECT_REQUEST_STATUS.format(request_id=request_id)
+    res = c.execute(sql).fetchall()
+    conn.close()
+    return res[0][0] if res else None
 
 
 def submit_devstone(request_id, username, depth, width, int_cycles, ext_cycles):
     conn = sqlite3.connect(DB_FILENAME)
     c = conn.cursor()
-    sql = SQL_INSERT_REQUEST.format(username=username, depth=depth, width=width, int_cycles=int_cycles, ext_cycles=ext_cycles)
+    sql = SQL_INSERT_REQUEST.format(request_id=request_id, username=username, depth=depth, width=width, int_cycles=int_cycles, ext_cycles=ext_cycles, status="pending")
+    c.execute(sql)
+    conn.commit()
+    conn.close()
+    # TODO: here it would be the devstone comparative script call
+    # time.sleep(5)
+    open("results/%s.csv" % request_id, "w").write(username)
+
+    conn = sqlite3.connect(DB_FILENAME)
+    sql = SQL_UPDATE_STATUS.format(request_id=request_id, status="finished")
+    c = conn.cursor()
     c.execute(sql)
     conn.commit()
-    open("%s.csv" % request_id, "w").write(username)
-    #sql = SQL_INSERT_REQUEST.format()
+    conn.close()
 
 
 @app.route('/')
 def hello():
-    return 'Hello, World!'
+    return 'Hey there! Ready to execute some DEVStone models?'
 
 
 @app.route('/run_devstone')
 def run_devstone():
     try:
-        api_key = int(request.args["api_key"])
+        api_key = request.args["api_key"]
+    except Exception as e:
+        return jsonify({"status": "denied", "reason": "No api_key was specified."})
+    
+    try:
         depth = int(request.args["depth"])
         width = int(request.args["width"])
         int_cycles = int(request.args["int_cycles"]) if "int_cycles" in request.args else 0
@@ -51,9 +78,28 @@ def run_devstone():
 
     # Execute task
     request_id = uuid.uuid4().hex
-    run_devstone(request_id, username, depth, width, int_cycles, ext_cycles)
-    return jsonify({"status": "submitted", "request_id": request_id})
+    submit_devstone(request_id, username, depth, width, int_cycles, ext_cycles)
+    status_url = request.host_url + "status?request_id=" + request_id
+    return jsonify({"status": "submitted", "request_id": request_id, "status_url": status_url})
+
+
+@app.route('/status')
+def status():
+    try:
+        request_id = request.args["request_id"]
+    except Exception as e:
+        return jsonify({"status": "denied", "reason": "No request_id was specified."})
+    
+    status = get_request_status(request_id)
+    res = {"status": status, "request_id": request_id}
+    if status == "finished":
+        res["download_link"] = "[still_not_implemented]"
+    
+    return jsonify(res)
 
 
 if __name__ == '__main__':
+    if not os.path.exists('results'):
+        os.makedirs('results')
+        
     app.run(debug=True, port=8080)