prequeue: less wasteful stdout rerouting

1 files changed, 8 insertions, 5 deletions

diff --git a/prequeue.py b/prequeue.py
index 6ad16d2c..00bcee36 100644
--- a/prequeue.py
+++ b/prequeue.py
@@ -13,17 +13,20 @@ import preprocessor
 def main():
     processor = preprocessor.setup_processor()
 
+    stdin  = sys.stdin
+    stdout = sys.stdout
+
     for source in sys.argv[1:]:
         dest = os.path.splitext(source)[0] + '.tx'
-
-        stdout = sys.stdout
-
         sys.stdin  = open(source, 'r')
         sys.stdout = open(dest, 'w')
-
         processor.preprocess()
 
-        sys.stdout = stdout
+    processor.update_globals()
+
+    sys.stdin  = stdin
+    sys.stdout = stdout
+
 
 if __name__ == '__main__':
     main()