import pickle
import shutil
+global dired
+dired = f"{os.path.expanduser('~')}/.cache/lightcards/"
-def gen_hash(inp):
+def name_gen(stra):
hasher = hashlib.md5()
- hasher.update(inp)
-
+ hasher.update(str(stra).encode("utf-8"))
return(hasher.hexdigest())
-def name_gen(stra):
- return gen_hash(str(stra).encode("utf-8"))
+def make_dirs(dired):
+ if not os.path.exists(dired):
+ os.makedirs(dired)
-def dump(obj, typer, stra):
- dired = f"{os.path.expanduser('~')}/.cache/lightcards/{name_gen(stra)}/"
- if os.path.exists(dired):
- shutil.rmtree(dired)
- os.makedirs(dired)
+def dump(obj, stra):
+ make_dirs(dired)
- pickle.dump(obj, open(f"{dired}/{typer}.p", "wb"))
+ pickle.dump(obj, open(f"{dired}/{name_gen(stra)}.p", "wb"))
-def dive(typer, stra):
- file = f"{os.path.expanduser('~')}/.cache/lightcards/{name_gen(stra)}/" + \
- f"{typer}.p"
+def dive(stra):
+ file = f"{dired}/{name_gen(stra)}.p"
+ make_dirs(dired)
if os.path.exists(file):
return pickle.load(open(file, "rb"))
else:
def purge(stra):
- dired = f"{os.path.expanduser('~')}/.cache/lightcards/{name_gen(stra)}/"
- shutil.rmtree(dired)
+ file = f"{dired}/{name_gen(stra)}/"
+ if os.path.exists(file):
+ shutil.rmtree(file)
-def purge_all():
- dired = f"{os.path.expanduser('~')}/.cache/lightcards/"
- shutil.rmtree(dired)
-def main():
- pass
+def purge_all():
+ if os.path.exists(dired):
+ shutil.rmtree(dired)
if __name__ == "__main__":