database.py

from motor.motor_asyncio import AsyncIOMotorClient
from config import Config

class Database:
    def __init__(self):
        self.client = AsyncIOMotorClient(Config.MONGODB_URI)
        self.db = self.client['file_renamer_bot']
        self.users = self.db['users']
        self.settings = self.db['settings']

    async def add_user(self, user_id):
        user = await self.users.find_one({'user_id': user_id})
        if not user:
            await self.users.insert_one({
                'user_id': user_id,
                'thumbnail': None,
                'prefix': '',
                'suffix': '',
                'metadata': {}
            })

    async def get_user(self, user_id):
        return await self.users.find_one({'user_id': user_id})

    async def set_thumbnail(self, user_id, file_id):
        await self.users.update_one(
            {'user_id': user_id},
            {'$set': {'thumbnail': file_id}},
            upsert=True
        )

    async def del_thumbnail(self, user_id):
        await self.users.update_one(
            {'user_id': user_id},
            {'$set': {'thumbnail': None}}
        )

    async def set_prefix(self, user_id, prefix):
        await self.users.update_one(
            {'user_id': user_id},
            {'$set': {'prefix': prefix}},
            upsert=True
        )

    async def set_suffix(self, user_id, suffix):
        await self.users.update_one(
            {'user_id': user_id},
            {'$set': {'suffix': suffix}},
            upsert=True
        )

db = Database()