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):
def purge(stra):
+ """Delete pickle file"""
file = f"{dired}/{name_gen(stra)}/"
if os.path.exists(file):
shutil.rmtree(file)