From 42873c371d20ce88d7bc46dca5b9fcd6d572750a Mon Sep 17 00:00:00 2001
From: Aritra Mahatma <128917736+aritra2023@users.noreply.github.com>
Date: Sun, 28 Jun 2026 03:47:02 +0530
Subject: [PATCH] Update database.py

---
 helper/database.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/helper/database.py b/helper/database.py
index 00587c6a..13cb28d7 100644
--- a/helper/database.py
+++ b/helper/database.py
@@ -125,6 +125,24 @@ async def is_banned(self, user_id: int) -> bool:
         user = await self.user_data.find_one({'_id': user_id})
         return user.get('ban', False) if user else False
 
+    # ✅ 24-HOUR VERIFICATION FUNCTIONS
+
+    async def set_verify_status(self, user_id: int):
+        """Set user as verified by saving the current timestamp"""
+        await self.user_data.update_one(
+            {'_id': user_id},
+            {'$set': {'verify_token': datetime.now()}},
+            upsert=True
+        )
+
+    async def check_verify_status(self, user_id: int) -> bool:
+        """Check if user has been verified in the last 24 hours"""
+        user = await self.user_data.find_one({'_id': user_id})
+        if user and 'verify_token' in user:
+            time_diff = datetime.now() - user['verify_token']
+            return time_diff.total_seconds() < 86400  # 24 hours in seconds
+        return False
+
     # ✅ FSUB CHANNELS FUNCTIONS
 
     async def set_fsub_channels(self, fsub_data: dict):