Add missing helper method

zmike808 · Jun 19, 2022 · b0779b1 · b0779b1
1 parent 279d417
commit b0779b1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -71,3 +71,5 @@ venv/
 *_downloads\
 
 *.txt
+
+*_live/
diff --git a/pyinstalive/helpers.py b/pyinstalive/helpers.py
@@ -293,3 +293,9 @@ def test_proxy():
             return proxy if response.status_code == 200 else None
         except Exception as e:
             return None
+
+def remove_temp_folder():
+    try:
+        shutil.rmtree(globals.download.segments_path)
+    except Exception as e:
+        logger.error("Could not remove segment folder: {:s}".format(str(e)))