from sqlalchemy import select
from sqlalchemy.orm import Session

from app.models.entities import ProviderAccount, ProviderModel


class ProvidersRepository:
    def __init__(self, db: Session) -> None:
        self.db = db

    def list_accounts(self):
        return self.db.query(ProviderAccount).order_by(ProviderAccount.id.desc())

    def get_account(self, account_id: int) -> ProviderAccount | None:
        return self.db.scalar(select(ProviderAccount).where(ProviderAccount.id == account_id))

    def list_models(self):
        return self.db.query(ProviderModel).order_by(ProviderModel.id.desc())

    def get_model(self, model_id: int) -> ProviderModel | None:
        return self.db.scalar(select(ProviderModel).where(ProviderModel.id == model_id))