One task per user

This commit is contained in:
yasir 2023-01-28 12:06:33 +07:00
parent 764b624536
commit f8bcfe898d

View file

@ -8,7 +8,7 @@
import os import os
import time import time
import traceback import traceback
from asyncio import gather, sleep, Lock from asyncio import gather, sleep
from logging import getLogger from logging import getLogger
from shutil import rmtree from shutil import rmtree
@ -41,11 +41,10 @@ async def genss(client, message):
if media is None: if media is None:
return await message.reply("Reply to a Telegram Video or document as video to generate screenshoot!") return await message.reply("Reply to a Telegram Video or document as video to generate screenshoot!")
process = await message.reply_text("`Processing, please wait..`") process = await message.reply_text("`Processing, please wait..`")
async with Lock():
if not DL_TASK.get(message.from_user.id): if not DL_TASK.get(message.from_user.id):
DL_TASK[message.from_user.id] = Lock() DL_TASK[message.from_user.id] = True
if DL_TASK.get(message.from_user.id).unlocked: 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.") return await process.edit("Sorry to avoid flood and error, bot only process one task at a time.")
c_time = time.time() c_time = time.time()
the_real_download_location = await replied.download( the_real_download_location = await replied.download(