0afa1ade15
* Ultroid 2025 Co-authored-by: New-dev0 <New-dev0@users.noreply.github.com> Co-authored-by: Amit Sharma <buddhhu@users.noreply.github.com> Co-authored-by: TechiError <techierror@users.noreply.github.com> Co-authored-by: Aditya <me@xditya.me> Co-authored-by: Arnab Paryali <Arnabxd@users.noreply.github.com> Co-authored-by: hellboi_atul <hellboi-atul@users.noreply.github.com> Co-authored-by: sppidy <sppidy@users.noreply.github.com>
40 lines
912 B
Python
40 lines
912 B
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2021-2025 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_stuff():
|
|
return udB.get_key("WARNS") or {}
|
|
|
|
|
|
def add_warn(chat, user, count, reason):
|
|
x = get_stuff()
|
|
try:
|
|
x[chat].update({user: [count, reason]})
|
|
except BaseException:
|
|
x.update({chat: {user: [count, reason]}})
|
|
return udB.set_key("WARNS", x)
|
|
|
|
|
|
def warns(chat, user):
|
|
x = get_stuff()
|
|
try:
|
|
count, reason = x[chat][user][0], x[chat][user][1]
|
|
return count, reason
|
|
except BaseException:
|
|
return 0, None
|
|
|
|
|
|
def reset_warn(chat, user):
|
|
x = get_stuff()
|
|
try:
|
|
x[chat].pop(user)
|
|
return udB.set_key("WARNS", x)
|
|
except BaseException:
|
|
return
|