Update database.py

database.py

@@ -4,6 +4,7 @@ from os import getenv
 from dotenv import load_dotenv
 
 import datetime
+from datetime import datetime as dt
 import time
 
 from motor import motor_asyncio
@@ -38,6 +39,28 @@ class Database:
             LOGS.info(f"DatabaseErr: {e} ")
             quit(1)
 
+    async def process_unfreezes(client):
+        while True:
+            now = dt.now()
+            expired = await self.user_blacklists.find({
+                "is_frozen": True,
+                "unfreeze_at": {"$lte": now}
+            })
+            for user in expired:
+                await self.user_blacklists.update_one(
+                    {"_id": user["_id"]},
+                    {"$set": {"is_frozen": False}}
+                )
+                try:
+                    await client.send_message(
+                        user["user_id"],
+                        "✅ Your restrictions have been lifted"
+                    )
+                except:
+                    pass
+
+            await asyncio.sleep(300)
+
     async def _close(self):
         await self.client.close()