Class TechifyBots

plugins/fsub.py:9–43

from config import Config

class TechifyBots:
    def __init__(self):
        client = AsyncIOMotorClient(Config.DB_URL)
        db = client[Config.DB_NAME]
        self.join_requests = db["join_requests"]
        self.fsub_cache = db["fsub_cache"]

    async def add_join_req(self, user_id: int, channel_id: int):
        await self.join_requests.update_one(
            {"user_id": user_id},
            {"$addToSet": {"channels": channel_id}, "$set": {"created_at": datetime.datetime.utcnow()}},
            upsert=True
        )

    async def has_joined_channel(self, user_id: int, channel_id: int) -> bool:
        doc = await self.join_requests.find_one({"user_id": user_id})
        return bool(doc and channel_id in doc.get("channels", []))

    async def del_join_req(self):
        await self.join_requests.drop()
        await self.fsub_cache.drop()

    async def save_fsub_msg(self, user_id: int, message_id: int):
        await self.fsub_cache.update_one(
            {"user_id": user_id},
            {"$set": {"message_id": message_id, "created_at": datetime.datetime.utcnow()}},
            upsert=True
        )

    async def get_fsub_msg(self, user_id: int):
        doc = await self.fsub_cache.find_one({"user_id": user_id})
        return doc.get("message_id") if doc else None

    async def delete_fsub_msg_db(self, user_id: int):
        await self.fsub_cache.delete_one({"user_id": user_id})

tb = TechifyBots()