From e235fe0ed20c2884fa649b8a48d2bb6915a19f37 Mon Sep 17 00:00:00 2001
From: profitroll <vozhd.kk@gmail.com>
Date: Sun, 2 Apr 2023 18:42:11 +0200
Subject: [PATCH] Improved bans

---
 modules/callbacks/ban.py |  5 ++++-
 validation/bans.json     | 29 +++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 1 deletion(-)
 create mode 100644 validation/bans.json

diff --git a/modules/callbacks/ban.py b/modules/callbacks/ban.py
index a96602a..8b2dae3 100644
--- a/modules/callbacks/ban.py
+++ b/modules/callbacks/ban.py
@@ -1,3 +1,4 @@
+from datetime import datetime
 from app import app, isAnAdmin
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery
 from pyrogram import filters
@@ -12,7 +13,9 @@ async def callback_query_reject(app: Client, clb: CallbackQuery):
     fullclb = clb.data.split("_")
 
     if not await isAnAdmin(int(fullclb[1])):
-        col_bans.insert_one({"user": int(fullclb[1])})
+        col_bans.insert_one(
+            {"user": int(fullclb[1]), "admin": clb.from_user.id, "date": datetime.now()}
+        )
 
     edited_markup = [
         [
diff --git a/validation/bans.json b/validation/bans.json
new file mode 100644
index 0000000..3bbc392
--- /dev/null
+++ b/validation/bans.json
@@ -0,0 +1,29 @@
+{
+    "$jsonSchema": {
+        "required": [
+            "user",
+            "admin",
+            "date"
+        ],
+        "properties": {
+            "user": {
+                "bsonType": [
+                    "int",
+                    "long"
+                ],
+                "description": "Telegram ID of user"
+            },
+            "admin": {
+                "bsonType": [
+                    "int",
+                    "long"
+                ],
+                "description": "Telegram ID of admin"
+            },
+            "date": {
+                "bsonType": "date",
+                "description": "Date and time of getting"
+            }
+        }
+    }
+}
\ No newline at end of file