Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Actually grab resume timestamp before it's overwritten (resume.py)

Browse source
This commit is contained in:
jesopo 2019-02-15 10:26:33 +00:00
parent a50b8c5694
commit f6179c46ab
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/resume.py
View file

@ -15,6 +15,11 @@ class Module(ModuleManager.BaseModule):
def _get_timestamp(self, server): def _get_timestamp(self, server):
return server.get_setting("last-read", None) return server.get_setting("last-read", None)
@utils.hook("new.server")
def new_server(self, event):
resume_timestamp = self._get_timestamp(event["server"])
event["server"]._resume_timestamp = resume_timestamp
@utils.hook("received.cap.ls") @utils.hook("received.cap.ls")
def on_cap_ls(self, event): def on_cap_ls(self, event):
if CAP in event["capabilities"]: if CAP in event["capabilities"]:
@ -29,18 +34,21 @@ class Module(ModuleManager.BaseModule):
def on_resume(self, event): def on_resume(self, event):
if event["args"][0] == "SUCCESS": if event["args"][0] == "SUCCESS":
self.log.info("Successfully resumed session", []) self.log.info("Successfully resumed session", [])
elif event["args"][0] == "ERR": elif event["args"][0] == "ERR":
self.log.info("Failed to resume session: %s", [event["args"][1]]) self.log.info("Failed to resume session: %s", [event["args"][1]])
elif event["args"][0] == "TOKEN": elif event["args"][0] == "TOKEN":
token = self._get_token(event["server"]) token = self._get_token(event["server"])
self._set_token(event["server"], event["args"][1], new=True) self._set_token(event["server"], event["args"][1], new=True)
if token: if token:
timestamp = self._get_timestamp(event["server"]) timestamp = event["server"]._resume_timestamp
event["server"].send("RESUME %s%s" % event["server"].send("RESUME %s%s" %
(token, " %s" % timestamp if timestamp else "")) (token, " %s" % timestamp if timestamp else ""))
event["server"].cap_started = False event["server"].cap_started = False
event["server"].capability_done("resume") event["server"].capability_done("resume")
@utils.hook("received.numeric.001") @utils.hook("received.numeric.001")