Make textgen script more generic
This commit is contained in:
parent
03d6b66341
commit
7c22096681
35
main.py
Normal file
35
main.py
Normal file
|
@ -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])
|
Loading…
Reference in a new issue