Fix error while creating zip

ruchernchong · Jul 6, 2024 · 0a96f5b · 0a96f5b
1 parent e17b19d
commit 0a96f5b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/__main__.py b/__main__.py
@@ -15,12 +15,13 @@
 
 
 def create_zip(files_to_zip):
-    with tempfile.TemporaryDirectory() as temp_dir:
-        zip_file = os.path.join(temp_dir, "function.zip")
-        with zipfile.ZipFile(zip_file, "w") as zipf:
-            for file in files_to_zip:
-                zipf.write(file, os.path.basename(file))
-        return zip_file
+    temp_dir = tempfile.TemporaryDirectory().name
+    os.makedirs(temp_dir)
+    zip_file = os.path.join(temp_dir, "function.zip")
+    with zipfile.ZipFile(zip_file, "w") as zipf:
+        for file in files_to_zip:
+            zipf.write(file, os.path.basename(file))
+    return zip_file
 
 
 zip_path = create_zip(["main.py", "calculator.py", "requirements.txt"])