cofin / litestar-granian Goto Github PK

   def init_app(
    init_api_routers: Callable or None,
    structlog_config: StructLoggingConfig,
    context: str,
    middlewares: list[Middleware],
) -> Litestar:
    set_event_loop_policy(EventLoopPolicy())

    if init_api_routers is None:

        def init_api_routers():
            return []

    structlog_plugin = StructlogPlugin(StructlogConfig(structlog_logging_config=structlog_config))
    app = Litestar(
        route_handlers=[*init_api_routers()],
        middleware=middlewares,
        plugins=[structlog_plugin, GranianPlugin()],
    )
    return app


   def init_apps() -> Litestar:
    context = "schedule_service"
    structlog_config = init_logging(context)
    logger = get_logger()
    app: Litestar = init_app(
        None, structlog_config, context, [GZipMiddleware]
    )
    app.on_startup = []
    app.on_shutdown = []
    app.logger = logger
    return app


app = init_apps()

from __future__ import annotations

from litestar import Controller, Litestar, get

from litestar_granian import GranianPlugin


class SampleController(Controller):
    @get(path="/sample")
    async def sample_route(self) -> dict[str, str]:
        """Sample Route."""
        return {"sample": "hello-world"}


app = Litestar(plugins=[GranianPlugin()], route_handlers=[SampleController])