diff options
-rw-r--r-- | amqp_codegen.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/amqp_codegen.py b/amqp_codegen.py index a2ad7a8..4ab7bdf 100644 --- a/amqp_codegen.py +++ b/amqp_codegen.py @@ -180,11 +180,12 @@ def do_main(header_fn,body_fn): stdout = sys.stdout f = open(out_file, 'w') try: - sys.stdout = f - fn(amqp_spec) - except: - remove(out_file) - raise + try: + sys.stdout = f + fn(amqp_spec) + except: + remove(out_file) + raise finally: sys.stdout = stdout f.close() |