tgbot-collection/ytdlbot
1
1
Fork 0
mirror of https://github.com/tgbot-collection/ytdlbot.git synced 2024-09-20 15:05:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Prevent race condition

Browse source
This commit is contained in:
jasonkhew96 2021-05-04 20:43:34 +08:00 committed by BennyThink
parent 50263b8cad
commit a783709355
No known key found for this signature in database
GPG key ID: 6CD0DBDA5235D481
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ytdl.py
View file

@ -38,17 +38,17 @@ async def upload_callback(current, total, chat_id, message):
key = f"{chat_id}-{message.id}" key = f"{chat_id}-{message.id}"
# if the key exists, we shouldn't send edit message # if the key exists, we shouldn't send edit message
if not r.exists(key): if not r.exists(key):
r.set(key, "ok", ex=EXPIRE)
msg = f'Uploading {round(current / total * 100, 2)}%: {current}/{total}' msg = f'Uploading {round(current / total * 100, 2)}%: {current}/{total}'
await bot.edit_message(chat_id, message, msg) await bot.edit_message(chat_id, message, msg)
r.set(key, "ok", ex=EXPIRE)
async def sync_edit_message(chat_id, message, msg): async def sync_edit_message(chat_id, message, msg):
# try to avoid flood # try to avoid flood
key = f"{chat_id}-{message.id}" key = f"{chat_id}-{message.id}"
if not r.exists(key): if not r.exists(key):
await bot.edit_message(chat_id, message, msg)
r.set(key, "ok", ex=EXPIRE) r.set(key, "ok", ex=EXPIRE)
await bot.edit_message(chat_id, message, msg)
def go(chat_id, message, msg): def go(chat_id, message, msg):