loic/lichess-notify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add cache expiration

Browse source
This commit is contained in:
Loic Guegan 2020-05-22 21:21:14 +02:00
parent 242682843e
commit 404db42aea
1 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
notify.py
View file

@ -1,12 +1,14 @@
#!/usr/bin/env python #!/usr/bin/env python
import berserk, subprocess, pickle import berserk, subprocess, pickle
from os import path from datetime import datetime
from os import path,remove
# Change ACCESS TOKEN according to your need # Change ACCESS TOKEN according to your need
ACCESS_TOKEN="" ACCESS_TOKEN=""
NOTIFY_DURATION=15*60 # Notification duration in seconds NOTIFY_DURATION=15*60 # Notification duration in seconds
CACHE_FILE="/tmp/lichess_notify_cache" # Change this according to your needs CACHE_FILE="/tmp/lichess_notify_cache" # Change this according to your needs
CACHE_EXPIRE=300 # Change cache expiration in minutes
# Notify using notify-send # Notify using notify-send
def notify_send(summary, message): def notify_send(summary, message):
@ -19,26 +21,28 @@ def notify_done(key):
if not(path.exists(CACHE_FILE)): if not(path.exists(CACHE_FILE)):
return(False) return(False)
else: else:
data=dict()
with open(CACHE_FILE, 'rb') as f: with open(CACHE_FILE, 'rb') as f:
keys=pickle.load(f) data=pickle.load(f)
return(key in keys) if datetime.timestamp(datetime.now()) - data["ts"] >= CACHE_EXPIRE*60:
remove(CACHE_FILE)
return(False)
else:
return(key in data["keys"])
# Save notify key in cache # Save notify key in cache
def add_key(key): def add_key(key):
if not(path.exists(CACHE_FILE)): if not(path.exists(CACHE_FILE)):
with open(CACHE_FILE, 'wb') as f: with open(CACHE_FILE, 'wb') as f:
pickle.dump([key],f) pickle.dump({"ts":datetime.timestamp(datetime.now()),"keys":key},f)
f.close()
else: else:
keys=list() data=list()
with open(CACHE_FILE, 'rb') as f: with open(CACHE_FILE, 'rb') as f:
keys=pickle.load(f) data=pickle.load(f)
f.close() if not(key in data["keys"]):
if not(key in keys): data["keys"].append(key)
keys.append(key)
with open(CACHE_FILE, 'wb') as f: with open(CACHE_FILE, 'wb') as f:
pickle.dump(keys,f) pickle.dump(data,f)
f.close()
# Fetch data and notify # Fetch data and notify
session = berserk.TokenSession(ACCESS_TOKEN) session = berserk.TokenSession(ACCESS_TOKEN)