17 lines
551 B
Python
17 lines
551 B
Python
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from backend.app.models.business import AIRecognitionLog
|
|
|
|
|
|
class AIRepository:
|
|
def create_log(self, session: Session, payload: dict) -> AIRecognitionLog:
|
|
log = AIRecognitionLog(**payload)
|
|
session.add(log)
|
|
session.flush()
|
|
return log
|
|
|
|
def get_log(self, session: Session, log_id: int) -> AIRecognitionLog | None:
|
|
stmt = select(AIRecognitionLog).where(AIRecognitionLog.id == log_id)
|
|
return session.execute(stmt).scalar_one_or_none()
|