diff --git a/modules/commands/warn.py b/modules/commands/warn.py index 3a8419d..6f8d4fe 100644 --- a/modules/commands/warn.py +++ b/modules/commands/warn.py @@ -1,7 +1,8 @@ -from os import sep +from datetime import datetime from app import app, isAnAdmin from pyrogram import filters -from modules.utils import jsonLoad, jsonSave, configGet, locale +from modules.utils import configGet, locale +from modules.database import col_warnings # Warn command ================================================================================================================= @app.on_message(~ filters.scheduled & filters.command(["warn"], prefixes=["/"])) @@ -10,11 +11,10 @@ async def cmd_warn(app, msg): if msg.chat.id == configGet("destination_group"): if msg.reply_to_message_id != None: if isAnAdmin(msg.from_user.id): - warnings = jsonLoad(f"{configGet('data', 'locations')}{sep}warnings.json") - if str(msg.reply_to_message.from_user.id) not in warnings: - warnings[str(msg.reply_to_message.from_user.id)] = 1 + message = " ".join(msg.command[2:]) if len(msg.command) > 1 else "" + col_warnings.insert_one({"user": msg.reply_to_message.from_user.id, "admin": msg.from_user.id, "date": datetime.now(), "reason": message}) + if message == "": + await msg.reply_text(locale("warned", "message").format(msg.reply_to_message.from_user.first_name, msg.reply_to_message.from_user.id)) else: - warnings[str(msg.reply_to_message.from_user.id)] += 1 - jsonSave(warnings, f"{configGet('data', 'locations')}{sep}warnings.json") - await msg.reply_text(locale("warned", "message").format(msg.reply_to_message.from_user.first_name, msg.reply_to_message.from_user.id)) + await msg.reply_text(locale("warned_reason", "message").format(msg.reply_to_message.from_user.first_name, msg.reply_to_message.from_user.id, message)) # ============================================================================================================================== \ No newline at end of file