Make textgen script more genericHEADmaster

1 files changed, 35 insertions, 0 deletions

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..eaad486
--- /dev/null
+++ b/main.py
@@ -0,0 +1,35 @@
+import os
+import subprocess
+import sys
+import threading
+import time
+from flask import Flask, Response
+
+
+app = Flask(__name__)
+
+@app.route('/', defaults={'path': ''})
+@app.route('/<path:path>')
+def handler(path):
+    if path == 'favicon.ico':
+        return Response(status=204)
+
+    def run():
+        try:
+            proc = subprocess.Popen(
+                map(lambda x: x.replace('{PATH}', path), sys.argv[2:]),
+                stdout=subprocess.PIPE
+            )
+            for c in iter(lambda: proc.stdout.read(1), b''):
+                yield c
+        finally:
+            proc.kill()
+
+    return Response(run(), mimetype='text/plain')
+
+def fixperms():
+    time.sleep(0.05)
+    os.chmod(sys.argv[1], 660)
+
+threading.Thread(target=fixperms).start()
+app.run(host='unix://' + sys.argv[1])