evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Only maintain single status message in a chat 

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2019-09-27 20:35:41 +05:30
parent 3fc9d40292
commit 464a633115
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bot/mirror.py
View File

@ -40,7 +40,11 @@ class MirrorListener(listeners.MirrorListeners):
def onUploadComplete(self, link: str, progress_status_list: list, index: int): def onUploadComplete(self, link: str, progress_status_list: list, index: int):
msg = '<a href="{}">{}</a>'.format(link, progress_status_list[index].name()) msg = '<a href="{}">{}</a>'.format(link, progress_status_list[index].name())
try:
deleteMessage(self.context, self.reply_message) deleteMessage(self.context, self.reply_message)
except BadRequest:
# This means that the message has been deleted because of a /status command
pass
sendMessage(msg, self.context, self.update) sendMessage(msg, self.context, self.update)
del download_dict[self.update.update_id] del download_dict[self.update.update_id]
fs_utils.clean_download(progress_status_list[index].path()) fs_utils.clean_download(progress_status_list[index].path())
@ -57,6 +61,7 @@ def mirror(update, context):
message = update.message.text message = update.message.text
link = message.replace('/mirror', '')[1:] link = message.replace('/mirror', '')[1:]
reply_msg = sendMessage('Starting Download', context, update) reply_msg = sendMessage('Starting Download', context, update)
status_reply_dict[update.effective_chat] = reply_msg
listener = MirrorListener(context, update, reply_msg) listener = MirrorListener(context, update, reply_msg)
aria = download_tools.DownloadHelper(listener) aria = download_tools.DownloadHelper(listener)
aria.add_download(link) aria.add_download(link)
@ -77,6 +82,7 @@ def mirror_status(update, context):
try: try:
deleteMessage(context, status_reply_dict[update.effective_chat]) deleteMessage(context, status_reply_dict[update.effective_chat])
del status_reply_dict[update.effective_chat] del status_reply_dict[update.effective_chat]
sendMessage(message, context, update)
except KeyError: except KeyError:
pass pass
break break
@ -87,6 +93,10 @@ def mirror_status(update, context):
sleep(DOWNLOAD_STATUS_UPDATE_INTERVAL) sleep(DOWNLOAD_STATUS_UPDATE_INTERVAL)
@run_async
def cancel_mirror(update, context):
pass
mirror_handler = CommandHandler('mirror', mirror) mirror_handler = CommandHandler('mirror', mirror)
mirror_status_handler = CommandHandler('status', mirror_status) mirror_status_handler = CommandHandler('status', mirror_status)
dispatcher.add_handler(mirror_handler) dispatcher.add_handler(mirror_handler)