WIP: #17

modules/custom_filters.py

@@ -4,10 +4,11 @@ usage in context of Holo Users."""
 from os import path
 from app import isAnAdmin
 from modules.utils import configGet, jsonLoad
-from modules.database import col_applications, col_tmp
+from modules.database import col_applications, col_tmp, col_bans
 from pyrogram import filters
 from pyrogram.types import Message
 
+
 async def admin_func(_, __, msg: Message):
     return await isAnAdmin(msg.from_user.id)
 
@@ -21,6 +22,11 @@ async def allowed_func(_, __, msg: Message):
         output = False
     return output
 
+
+async def banned_func(_, __, msg: Message):
+    return True if col_bans.find_one( {"user": msg.from_user.id} ) is not None else False
+
+
 async def enabled_general_func(_, __, msg: Message):
     return configGet("enabled", "features", "general")
 
@@ -49,6 +55,8 @@ admin = filters.create(admin_func)
 member = filters.create(member_func)
 allowed = filters.create(allowed_func)
 
+banned = filters.create(banned_func)
+
 enabled_general = filters.create(enabled_general_func)
 enabled_applications = filters.create(enabled_applications_func)
 enabled_sponsorships = filters.create(enabled_sponsorships_func)

modules/database.py

@@ -28,11 +28,12 @@ db = db_client.get_database(name=db_config["name"])
 
 collections = db.list_collection_names()
 
-for collection in ["tmp", "users", "context", "youtube", "spoilers", "messages", "warnings", "applications", "sponsorships"]:
+for collection in ["tmp", "bans", "users", "context", "youtube", "spoilers", "messages", "warnings", "applications", "sponsorships"]:
     if not collection in collections:
         db.create_collection(collection)
 
 col_tmp = db.get_collection("tmp")
+col_bans = db.get_collection("bans")
 col_users = db.get_collection("users")
 col_context = db.get_collection("context")
 col_youtube = db.get_collection("youtube")