d8bd901072
Date: 30-8-2022 Co-Authored-By: Aditya <me@xditya.me> Co-Authored-By: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-Authored-By: CyrusXD <79554993+Ashutosh1478@users.noreply.github.com> Co-Authored-By: Danish <danish@ultroid.tech> Co-Authored-By: TechiError <error@notavailable.live> Co-Authored-By: Arnab Paryali <arnabxd@pm.me>
40 lines
912 B
Python
40 lines
912 B
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2021-2022 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
|