X-Git-Url: https://git.armaanb.net/?a=blobdiff_plain;f=lightcards%2Fprogress.py;h=8956b63ab7043063850bb29adf837eb8f4fb2eb0;hb=85327074f6aaa4e46ec36cec6fd6f07317e96ea6;hp=297895d2960735338ec580588d789c6285cdfc15;hpb=65cb608a985f5168edfa6e2833f37bc812274a2f;p=lightcards.git

diff --git a/lightcards/progress.py b/lightcards/progress.py
index 297895d..8956b63 100644
--- a/lightcards/progress.py
+++ b/lightcards/progress.py
@@ -4,17 +4,13 @@
 import hashlib
 import os
 import pickle
-import shutil
 
-global dired
 dired = f"{os.path.expanduser('~')}/.cache/lightcards/"
 
 
 def name_gen(stra):
     """Generate hash of stack for name of pickle file"""
-    hasher = hashlib.md5()
-    hasher.update(str(stra).encode("utf-8"))
-    return hasher.hexdigest()
+    return hashlib.md5(str([str(x) for x in stra]).encode("utf-8")).hexdigest()
 
 
 def make_dirs(dired):
@@ -42,6 +38,6 @@ def dive(stra):
 
 def purge(stra):
     """Delete pickle file"""
-    file = f"{dired}/{name_gen(stra)}/"
+    file = f"{dired}/{name_gen(stra)}.p"
     if os.path.exists(file):
-        shutil.rmtree(file)
+        os.remove(file)