Files
Ultroid-fork/pyUltroid/dB/antiflood_db.py
2023-07-09 22:25:45 +05:30

33 lines
719 B
Python

# Ultroid - UserBot
# Copyright (C) 2021-2023 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://github.com/TeamUltroid/pyUltroid/blob/main/LICENSE>.
from .. import udB
def get_flood():
return udB.get_key("ANTIFLOOD") or {}
def set_flood(chat_id, limit):
omk = get_flood()
omk.update({chat_id: limit})
return udB.set_key("ANTIFLOOD", omk)
def get_flood_limit(chat_id):
omk = get_flood()
if chat_id in omk.keys():
return omk[chat_id]
def rem_flood(chat_id):
omk = get_flood()
if chat_id in omk.keys():
del omk[chat_id]
return udB.set_key("ANTIFLOOD", omk)