main.py

import logging
from logging.handlers import RotatingFileHandler
import asyncio
from pyrogram import raw, __version__, idle, types

from tg import handlers, utils
from db import repository
from data import config, clients


# log config
logging.getLogger("pyrogram").setLevel(logging.WARNING)
logging.getLogger("aiosqlite").setLevel(logging.WARNING)

root_logger = logging.getLogger()
root_logger.setLevel(level=logging.DEBUG)

log_format = logging.Formatter("%(asctime)s | %(levelname)s | %(name)s | %(message)s")

console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
console_handler.setFormatter(log_format)
root_logger.addHandler(console_handler)

file_handler = RotatingFileHandler(
    filename="bot.log", maxBytes=20 * (2**20), backupCount=3, mode="D", encoding="utf-8"
)
file_handler.setLevel(logging.DEBUG)
file_handler.setFormatter(log_format)
root_logger.addHandler(file_handler)


_logger = logging.getLogger(__name__)

settings = config.get_settings()


async def main():
    logging.info(
        f"The bot is up and running on Pyrogram v{__version__} (Layer {raw.all.layer})."
    )

    for handler in handlers.HANDLERS:
        clients.bot_1.add_handler(handler)

    for admin in settings.admins:
        if not (user := await repository.get_user(tg_id=admin)):
            await repository.create_user(
                tg_id=admin, name="admin", admin=True, language_code="he"
            )
        else:
            if not user.admin:
                await repository.update_user(tg_id=admin, admin=True)

    await clients.bot_1.start()
    await clients.bot_2.start()

    # await utils.set_bot_info(clients.bot_1)

    await idle()

    await clients.bot_1.stop()
    await clients.bot_2.stop()


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())