Telegram/modules/custom_filters.py

"""Custom message filters made to improve commands
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 pyrogram import filters
from pyrogram.types import Message

async def admin_func(_, __, msg: Message):
    return await isAnAdmin(msg.from_user.id)

async def member_func(_, __, msg: Message):
    return True if (msg.from_user.id in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))) else False

async def allowed_func(_, __, msg: Message):
    return True if (col_applications.find_one({"user": msg.from_user.id}) is not None) else False

async def enabled_general_func(_, __, msg: Message):
    return configGet("enabled", "features", "general")

async def enabled_applications_func(_, __, msg: Message):
    return configGet("enabled", "features", "applications")

async def enabled_sponsorships_func(_, __, msg: Message):
    return configGet("enabled", "features", "sponsorships")

async def enabled_warnings_func(_, __, msg: Message):
    return configGet("enabled", "features", "warnings")

async def enabled_invites_check_func(_, __, msg: Message):
    return configGet("enabled", "features", "invites_check")

async def enabled_dinovoice_func(_, __, msg: Message):
    return configGet("enabled", "features", "dinovoice")

async def filling_sponsorship_func(_, __, msg: Message):
    return True if col_tmp.find_one({"user": msg.from_user.id, "type": "sponsorship"}) is not None else False

admin = filters.create(admin_func)
member = filters.create(member_func)
allowed = filters.create(allowed_func)

enabled_general = filters.create(enabled_general_func)
enabled_applications = filters.create(enabled_applications_func)
enabled_sponsorships = filters.create(enabled_sponsorships_func)
enabled_warnings = filters.create(enabled_warnings_func)
enabled_invites_check = filters.create(enabled_invites_check_func)
enabled_dinovoice = filters.create(enabled_dinovoice_func)

filling_sponsorship = filters.create(filling_sponsorship_func)