spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix UberEats task cancellation

Browse Source
This commit is contained in:
AlberLC
2023-03-03 22:06:47 +01:00
parent 5021bdd7ae
commit d045551db9
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
flanabot/bots/ubereats_bot.py
View File

@@ -113,13 +113,13 @@ class UberEatsBot(MultiBot, ABC):
async def start_ubereats(self, chat: Chat, send_code_now=True):
chat.config['ubereats'] = True
chat.save()
if self.tasks[chat.id]:
self.tasks[chat.id].cancel()
if (task := self.tasks.get(chat.id)) and not task.done():
task.cancel()
self.tasks[chat.id] = await flanautils.do_every(chat.ubereats_seconds, self.send_ubereats_code, chat, do_first_now=send_code_now)
def stop_ubereats(self, chat: Chat):
chat.config['ubereats'] = False
chat.save()
if self.tasks[chat.id]:
self.tasks[chat.id].cancel()
self.tasks[chat.id] = None
if (task := self.tasks.get(chat.id)) and not task.done():
task.cancel()
del self.tasks[chat.id]