augustin_s/pier
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
augustin_s
2022-05-10 10:26:01 +02:00
parent 700eb1e762
commit 009a10a4cd
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pier.py
View File

@ -25,11 +25,7 @@ class ScriptServer(http.server.SimpleHTTPRequestHandler):
self.send_result(f"{scr} does not exist. choose from:\n", printable_scripts) self.send_result(f"{scr} does not exist. choose from:\n", printable_scripts)
return return
sio = StringIO() res = run_script(scr)
with redirect_stdout(sio), redirect_stderr(sio):
with open(scr) as f:
exec(f.read())
res = sio.getvalue()
msg = f"{scr} output:\n" msg = f"{scr} output:\n"
msg += "_" * 80 msg += "_" * 80
@ -46,6 +42,12 @@ class ScriptServer(http.server.SimpleHTTPRequestHandler):
print(msg) print(msg)
def run_script(fn):
sio = StringIO()
with redirect_stdout(sio), redirect_stderr(sio):
with open(fn) as f:
exec(f.read())
return sio.getvalue()
def encode_html(*msg): def encode_html(*msg):
msg = (str(i) for i in msg) msg = (str(i) for i in msg)