bootstrap.py

from typing import cast

from dependency_injector.containers import DynamicContainer
from dependency_injector.providers import Object
from pydantic import BaseModel, ConfigDict

from .asyncapi import init_asyncapi_info
from .config import AppConfig
from .di_container import Container
from .dramatiq import init_dramatiq
from .logs import init_logger
from .storage import init_storage


class InitReference(BaseModel):
    di_container: DynamicContainer

    model_config = ConfigDict(arbitrary_types_allowed=True)


def application_init(app_config: AppConfig) -> InitReference:
    container = cast(
        DynamicContainer,  # Make mypy happy
        Container(
            config=Object(app_config),
        ),
    )
    init_logger(app_config)
    init_storage()
    init_dramatiq(app_config)
    init_asyncapi_info(app_config.APP_NAME)

    return InitReference(
        di_container=container,
    )