From 9e0108f5fe04077d3f18bd984a8a172bbb7d68f9 Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Tue, 10 May 2022 10:35:42 +0200 Subject: [PATCH] split open and run --- pier.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/pier.py b/pier.py index 08e4067..701234f 100755 --- a/pier.py +++ b/pier.py @@ -45,12 +45,16 @@ class ScriptServer(http.server.SimpleHTTPRequestHandler): def run_script(fn): sio = StringIO() with redirect_stdout(sio), redirect_stderr(sio): - with open(fn) as f: - src = f.read() - code = compile(src, fn, "exec") - exec(code) + code = open_script(fn) + exec(code) return sio.getvalue() +def open_script(fn): + with open(fn) as f: + src = f.read() + code = compile(src, fn, "exec") + return code + def encode_html(*msg): msg = (str(i) for i in msg) msg = " ".join(msg)