yasirarism/MissKatyPyro
1
0
Fork 0
mirror of https://github.com/yasirarism/MissKatyPyro.git synced 2025-12-29 17:44:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

One task per user

Browse source
This commit is contained in:
yasir 2023-01-28 12:06:33 +07:00
parent 764b624536
commit f8bcfe898d
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
misskaty/plugins/genss.py
View file

@ -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,