tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed cooldown

Browse Source
This commit is contained in:
BarsTiger
2023-02-14 14:05:06 +02:00
parent 64bbafbb32
commit 3bd34009ae
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bot/utils/cooldown.py
View File

@@ -1,6 +1,5 @@
from functools import wraps
import datetime
from bot.common import bot
from bot.db import db, DBTables
import asyncio
from aiogram import types
@@ -33,10 +32,13 @@ def throttle(cooldown: int = 5, by_id: bool = True, admin_ids: list = None):
if last_time <= delta:
try:
db[DBTables.cooldown][func.__name__][user_id] = now
f_name_dict = db[DBTables.cooldown][func.__name__]
f_name_dict[user_id] = now
db[DBTables.cooldown][func.__name__] = f_name_dict
except KeyError:
db[DBTables.cooldown][func.__name__] = dict()
db[DBTables.cooldown][func.__name__][user_id] = now
f_name_dict = dict()
f_name_dict[user_id] = now
db[DBTables.cooldown][func.__name__] = f_name_dict
try:
return asyncio.create_task(func(*args, **kwargs))
except Exception as e: