Start programs and report status

.gitignore

@@ -0,0 +1 @@
+config.py

app.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python3
+import subprocess
+import select
+import time
+
+import config
+import logging.config
+logging.config.dictConfig(config.logging)
+log = logging.getLogger('progrunner')
+
+from flask import Flask, request, session
+from flask_restful import Api, Resource
+
+app = Flask(__name__)
+api = Api(app)
+app.config.from_object(config)
+
+class Authenticator(Resource):
+    def post(self):
+        if request.json and request.json.get("token", None) == config.token:
+            session["authorized"] = True
+            log.info("session: %s", session)
+            return { "authorized": True }
+        else:
+            session["authorized"] = False
+            log.info("session: %s", session)
+            return { "authorized": False }
+
+running = []
+
+class Starter(Resource):
+    def post(self):
+        log.info("session: %s", session)
+        if not session.get("authorized", False):
+            return {}
+
+        log.info("json: %s", request.json)
+        args = [ request.json["program"] ]
+        if "args" in request.json:
+            args += request.json["args"]
+        log.info("args: %s", args)
+        proc = subprocess.Popen(args, stdout=subprocess.PIPE, universal_newlines=True)
+        procinfo = { "popen": proc, "pid": proc.pid, "started": time.time(), "stdout": [] }
+        running.append(procinfo)
+        pid = proc.pid
+        log.info("pid: %s", pid)
+        timeout = request.json.get("timeout", 0.0)
+        if timeout:
+            fd = proc.stdout
+            p = select.poll()
+            p.register(fd)
+            t0 = time.time()
+            t1 = t0 + timeout
+            while True:
+                dt = t1 - time.time()
+                if dt < 0:
+                    break
+                ready = p.poll(dt * 1000)
+                log.debug("ready: %s", ready)
+                if ready:
+                    ln = fd.readline()
+                    log.debug("ln: %s", ln)
+                    if ln == "":
+                        break;
+                    procinfo["stdout"].append(ln)
+        try:
+            procinfo["status"] = proc.wait(0.1)
+            for ln in proc.stdout:
+                procinfo["stdout"].append(ln)
+            procinfo["finished"] = time.time()
+        except subprocess.TimeoutExpired:
+            pass
+        return { x: procinfo[x]
+                    for x in ["pid", "started", "stdout", "status", "finished"]
+                        if x in procinfo }
+
+class Status(Resource):
+    def get(self, pid):
+        for procinfo in running:
+            if "finished" not in procinfo:
+                p = select.poll()
+                p.register(procinfo["popen"].stdout)
+                t1 = time.time() + 1.0
+                while True:
+                    dt = t1 - time.time()
+                    if dt < 0:
+                        break
+                    ready = p.poll(dt * 1000)
+                    log.debug("ready: %s", ready)
+                    if ready:
+                        ln = procinfo["popen"].stdout.readline()
+                        log.debug("ln: %s", ln)
+                        if ln == "":
+                            break;
+                        procinfo["stdout"].append(ln)
+                try:
+                    procinfo["status"] = procinfo["popen"].wait(0.1)
+                    procinfo["finished"] = time.time()
+                except subprocess.TimeoutExpired:
+                    pass
+        for procinfo in running:
+            if procinfo["pid"] == pid:
+                return { x: procinfo[x]
+                            for x in ["pid", "started", "stdout", "status", "finished"]
+                                if x in procinfo }
+
+
+
+
+
+api.add_resource(
+        Authenticator,
+        "/login"
+    )
+api.add_resource(
+        Starter,
+        "/start"
+    )
+api.add_resource(
+        Status,
+        "/status/<int:pid>"
+    )
+
+
+if __name__ == '__main__':
+    app.run()

requirements.txt

@@ -0,0 +1,2 @@
+Flask
+Flask-RESTful