from typing import List

from sqlmodel import Session, select
from entity.frame_analysis_result import FrameAnalysisResultCreate, FrameAnalysisResult


class FrameAnalysisResultService:

    def __init__(self, db: Session):
        self.db = db

    def add_frame_analysis_results(self, results: List[FrameAnalysisResultCreate]):
        new_results = [FrameAnalysisResult.model_validate(result) for result in results]
        self.db.add_all(new_results)
        self.db.commit()
        for result in new_results:
            self.db.refresh(result)
        return new_results

    def get_results_by_frame(self, frame_id):
        statement = select(FrameAnalysisResult).where(FrameAnalysisResult.frame_id == frame_id)
        results = self.db.exec(statement)
        return results.all()