Don't delete the output file before closing it

Windows won't let you, and produces an error, disguising whatever
error led us to try to delete the output file in the first place.

1 files changed, 7 insertions, 7 deletions

diff --git a/amqp_codegen.py b/amqp_codegen.py
index a10cc6d..f265962 100644
--- a/amqp_codegen.py
+++ b/amqp_codegen.py
@@ -32,7 +32,7 @@
 from __future__ import nested_scopes
 import re
 import sys
-from os import remove
+import os
 from optparse import OptionParser
 
 try:
@@ -271,16 +271,16 @@ def do_main_dict(funcDict):
     def execute(fn, amqp_specs, out_file):
         stdout = sys.stdout
         f = open(out_file, 'w')
+        success = False
         try:
-            try:
-                sys.stdout = f
-                fn(amqp_specs)
-            except:
-                remove(out_file)
-                raise
+            sys.stdout = f
+            fn(amqp_specs)
+            success = True
         finally:
             sys.stdout = stdout
             f.close()
+            if not success:
+                os.remove(out_file)
 
     parser = OptionParser()
     parser.add_option("--ignore-conflicts", action="store_true", dest="ignore_conflicts", default=False)