Save RSS state on exit

This commit is contained in:
Robert Miles 2020-07-16 01:07:39 -04:00
parent 333d2cdb76
commit 46fdcc85a0
1 changed files with 2 additions and 1 deletions

3
bot.py
View File

@ -1,4 +1,4 @@
import teambot,tasks,rss,time,sys,subprocess,re,plugin,impmod,traceback,os
import teambot,tasks,rss,time,sys,subprocess,re,plugin,impmod,traceback,os,atexit
feed_url = "https://cosmic.voyage/rss.xml"
bot_op = "khuxkm@sudoers.tilde.team"
@ -25,6 +25,7 @@ class CosmicBot(teambot.Handler):
self.tasks = tasks.TaskPool()
self.tasks.add_coroutine(self.check_rss,12,dict(url=feed_url,known=[],channel="#cosmic"))
self.tasks.load_state(0)
atexit.register(lambda: self.tasks.save_state(0))
def load_modules(self):
plugin.clear()
for module in os.listdir("commands"):