from anydi.ext.fastapi import get_container
from debug_toolbar.panels.sqlalchemy import SQLAlchemyPanel
from fansifter_common.adapters.db.base import Database
from starlette.requests import Request


class BaseDBPanel(SQLAlchemyPanel):
    db_cls: type[Database]

    async def add_engines(self, request: Request) -> None:
        container = get_container(request)
        db = container.resolve(self.db_cls)
        self.engines.add(db.engine)