X-Git-Url: https://git.armaanb.net/?a=blobdiff_plain;f=lightcards%2Fprogress.py;h=3be0247e42170c3243cf03814dc09b51b0ab38bc;hb=6226519694970400eb98632369d042b0499ecabe;hp=bf78e0161a8349584ca7966ab77194a7876274a2;hpb=f8dbb2a02277188e677bb89ffdacb13a270df35e;p=lightcards.git

diff --git a/lightcards/progress.py b/lightcards/progress.py
index bf78e01..3be0247 100644
--- a/lightcards/progress.py
+++ b/lightcards/progress.py
@@ -6,28 +6,29 @@ import os
 import pickle
 import shutil
 
-global dired
 dired = f"{os.path.expanduser('~')}/.cache/lightcards/"
 
 
 def name_gen(stra):
-    hasher = hashlib.md5()
-    hasher.update(str(stra).encode("utf-8"))
-    return hasher.hexdigest()
+    """Generate hash of stack for name of pickle file"""
+    return hashlib.md5(str([str(x) for x in stra]).encode("utf-8")).hexdigest()
 
 
 def make_dirs(dired):
+    """mkdir -p equivalent"""
     if not os.path.exists(dired):
         os.makedirs(dired)
 
 
 def dump(obj, stra):
+    """Write pickle file"""
     make_dirs(dired)
 
     pickle.dump(obj, open(f"{dired}/{name_gen(stra)}.p", "wb"))
 
 
 def dive(stra):
+    """Get pickle file"""
     file = f"{dired}/{name_gen(stra)}.p"
     make_dirs(dired)
     if os.path.exists(file):
@@ -37,6 +38,7 @@ def dive(stra):
 
 
 def purge(stra):
+    """Delete pickle file"""
     file = f"{dired}/{name_gen(stra)}/"
     if os.path.exists(file):
         shutil.rmtree(file)