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)