feat: initialize aivideo project
This commit is contained in:
22
backend/app/modules/providers/repository.py
Normal file
22
backend/app/modules/providers/repository.py
Normal file
@@ -0,0 +1,22 @@
|
||||
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))
|
||||
|
||||
Reference in New Issue
Block a user