main.py

from bot_instance import bot, dp

from globals import *

from start_functions import *

import editing

import editing.actions

import profile_list

from registration import *

from interactions.globals import *

import interactions.likes.view_likes

from admin import *

from sessions import user_sessions, deserialize_user_sessions

from states import load_states

import db

from docker import Docker

from logger import get_BM_logger

logger = get_BM_logger(__name__)


async def on_startup(dp):
    db.check_db()

    await load_states.load_editing_states()


async def on_shoutdown():
    save_sessions()
    await remove_markups()
    db.close_connection()
    logger.system("Shutting down")


def save_sessions():
    session_values = list(user_sessions.values())
    deserialize_user_sessions(session_values)

    logger.system("Sessions saved")


async def remove_markups():
    for session in user_sessions.values():
        try:
            await bot.edit_message_reply_markup(
                session.telegram_id, session.last_message, reply_markup=None)
        except Exception as e:
            logger.error(e)

containter = Docker()

containter.set_on_shutdown(on_shoutdown)

if __name__ == "__main__":
    from aiogram import executor

    executor.start_polling(dp, skip_updates=True,
                           on_startup=on_startup,
                           allowed_updates=["message", "callback_query"])