chore: cleanup temp files

DHARPA-Project · Apr 17, 2024 · c6ddbf4 · c6ddbf4
1 parent ead3c86
commit c6ddbf4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/kiara_plugin/onboarding/modules/download/__init__.py b/src/kiara_plugin/onboarding/modules/download/__init__.py
@@ -90,8 +90,12 @@ def retrieve_archive(
             suffix = ""
 
         tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
-        atexit.register(tmp_file.close)
 
+        def rm_tmp_file():
+            tmp_file.close()
+            os.unlink(tmp_file.name)
+
+        atexit.register(rm_tmp_file)
         kiara_file: KiaraFile
 
         kiara_file = download_file(  # type: ignore

diff --git a/src/kiara_plugin/onboarding/utils/download.py b/src/kiara_plugin/onboarding/utils/download.py
@@ -145,10 +145,15 @@ def download_file_bundle(
         suffix = ""
 
     tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
-    atexit.register(tmp_file.close)
+
+    def rm_tmp_file():
+        tmp_file.close()
+        os.unlink(tmp_file.name)
+
+    atexit.register(rm_tmp_file)
 
     history = []
-    datetime.utcnow().replace(tzinfo=pytz.utc)
+    # datetime.utcnow().replace(tzinfo=pytz.utc)
     with open(tmp_file.name, "wb") as f:
         with httpx.stream("GET", url, follow_redirects=True) as r:
             history.append(dict(r.headers))