Cleanup temp files at exit #435

1 files changed, 12 insertions, 0 deletions

diff --git a/src/sugar/util.py b/src/sugar/util.py
index d375b87..e027734 100644
--- a/src/sugar/util.py
+++ b/src/sugar/util.py
@@ -28,6 +28,8 @@ import binascii
 import gettext
 import tempfile
 import logging
+import atexit
+import traceback
 
 _ = lambda msg: gettext.dgettext('sugar-toolkit', msg)
 
@@ -291,3 +293,13 @@ class TempFilePath(str):
         else:
             _tracked_paths[self] -= 1
 
+def _cleanup_temp_files():
+    logging.debug('_cleanup_temp_files')
+    for path in _tracked_paths.keys():
+        try:
+            os.unlink(path)
+        except:
+            logging.error(traceback.format_exc())
+
+atexit.register(_cleanup_temp_files)
+