From 69b4102c996f1793dbc7280924ae023281f94d30 Mon Sep 17 00:00:00 2001
From: Pedro de Medeiros <pedro.medeiros@gmail.com>
Date: Wed, 2 Jun 2021 22:31:12 -0300
Subject: map.py: remove generated files on error

---
 tools/map.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'tools')

diff --git a/tools/map.py b/tools/map.py
index c2e2212..53c98b7 100755
--- a/tools/map.py
+++ b/tools/map.py
@@ -318,6 +318,7 @@ def main():
     if args.bin:
         for i, block in enumerate(out):
             filename = path.join(args.dir, "%s%02d.bin" % (args.id, i))
+            remove_list.append(filename)
             with open(filename, "wb") as fd:
                 if i in empty:
                     fd.write(struct.pack("<B", 0))
@@ -379,9 +380,16 @@ def main():
 
 
 if __name__ == "__main__":
+    remove_list = []
+
     try:
         main()
     except Exception as ex:
         print("FATAL: %s\n***" % ex, file=sys.stderr)
-        traceback.print_exc(ex)
+        traceback.print_exc()
+
+        for filename in remove_list:
+            if os.path.exists(filename):
+                os.unlink(filename)
+
         sys.exit(1)
-- 
cgit v1.2.3