GPQBot

1
from functools import partial
2

3
from fastapi import FastAPI
4
from pybotx import Bot
5

6
from db.sqlalchemy import build_db_session_factory
7
from dependencies.bot import get_bot
8
from api.routers import router
9

10
async def startup(bot: Bot) -> None:
11
    # -- Bot --
12
    await bot.startup()
13

14
    # -- Database --
15
    bot.state.db_session_factory = await build_db_session_factory()
16
    
17

18

19
async def shutdown(application: FastAPI) -> None:
20
    bot: Bot = application.state.bot
21
    await bot.shutdown()
22

23
def get_application() -> FastAPI:
24
    """Create configured server application instance."""
25
    bot = get_bot()
26

27
    application = FastAPI(title='BOT')
28
    application.state.bot = bot
29

30
    application.add_event_handler("startup", partial(startup, bot))
31
    application.add_event_handler("shutdown", partial(shutdown, bot))
32

33
    application.include_router(router)
34

35
    return application
36

37

38
app = get_application()