diff options
Diffstat (limited to 'amqp_codegen.py')
| -rw-r--r-- | amqp_codegen.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/amqp_codegen.py b/amqp_codegen.py index a10cc6d..c6c6a5f 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: @@ -172,7 +172,7 @@ class AmqpSpec: self.classes = [] for element in self.spec['classes']: - self.classes.append(AmqpClass(self.spec, element)) + self.classes.append(AmqpClass(self, element)) def allClasses(self): return self.classes @@ -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) |