From f8bcfe898d3d99740becc2a33063b2132d9f1100 Mon Sep 17 00:00:00 2001 From: yasir Date: Sat, 28 Jan 2023 12:06:33 +0700 Subject: [PATCH] One task per user --- misskaty/plugins/genss.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/misskaty/plugins/genss.py b/misskaty/plugins/genss.py index 2086cec3..a8b0fc94 100644 --- a/misskaty/plugins/genss.py +++ b/misskaty/plugins/genss.py @@ -8,7 +8,7 @@ import os import time import traceback -from asyncio import gather, sleep, Lock +from asyncio import gather, sleep from logging import getLogger from shutil import rmtree @@ -41,12 +41,11 @@ 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..`") - async with Lock(): - if not DL_TASK.get(message.from_user.id): - DL_TASK[message.from_user.id] = Lock() + if not DL_TASK.get(message.from_user.id): + DL_TASK[message.from_user.id] = True - if DL_TASK.get(message.from_user.id).unlocked: - return await process.edit("Sorry to avoid flood and error, bot only process one task at a time.") + 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.") c_time = time.time() the_real_download_location = await replied.download( progress=progress_for_pyrogram,