1 files changed, 4 insertions, 4 deletions
diff --git a/notify.py b/notify.py
index a97656b..94d5037 100755
--- a/notify.py
+++ b/notify.py
@@ -24,23 +24,23 @@ def notify_done(key):
         data=dict()
         with open(CACHE_FILE, 'rb') as f:
             data=pickle.load(f)
-        if datetime.timestamp(datetime.now()) - data["ts"] >= CACHE_EXPIRE*60:
+        if datetime.timestamp(datetime.now()) - data[key] >= CACHE_EXPIRE*60:
             remove(CACHE_FILE)
             return(False)
         else:
-            return(key in data["keys"])
+            return(key in data)
 
 # Save notify key in cache
 def add_key(key):
     if not(path.exists(CACHE_FILE)):
         with open(CACHE_FILE, 'wb') as f:
-            pickle.dump({"ts":datetime.timestamp(datetime.now()),"keys":key},f)
+            pickle.dump({key:datetime.timestamp(datetime.now())},f)
     else:
         data=list()
         with open(CACHE_FILE, 'rb') as f:
             data=pickle.load(f)
         if not(key in data["keys"]):
-            data["keys"].append(key)
+            data[key]=datetime.timestamp(datetime.now())
         with open(CACHE_FILE, 'wb') as f:
             pickle.dump(data,f)