From 5e80d07e3d20b8b7b788fe7bdd5792378d18da46 Mon Sep 17 00:00:00 2001 From: yasir Date: Sat, 28 Jan 2023 12:13:13 +0700 Subject: [PATCH] One task per user --- misskaty/plugins/genss.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/misskaty/plugins/genss.py b/misskaty/plugins/genss.py index af4636c8..33db41ff 100644 --- a/misskaty/plugins/genss.py +++ b/misskaty/plugins/genss.py @@ -41,11 +41,12 @@ async def genss(client, message): if media is None: return await message.reply("Reply to a Telegram Video or document as video to generate screenshoot!") process = await message.reply_text("`Processing, please wait..`") - if not DL_TASK.get(message.from_user.id): - DL_TASK[message.from_user.id] = True - - if DL_TASK.get(message.from_user.id) == True: + if DL_TASK.get(message.from_user.id): return await process.edit("Sorry to avoid flood and error, bot only process one task at a time.") + + if not DL_TASK.get(message.from_user.id): + DL_TASK[message.from_user.id] = 1 + c_time = time.time() the_real_download_location = await replied.download( progress=progress_for_pyrogram,