MediaManager/media_manager/indexer/repository.py

import logging

from sqlalchemy.orm import Session

from media_manager.indexer.models import IndexerQueryResult
from media_manager.indexer.schemas import (
    IndexerQueryResultId,
    IndexerQueryResult as IndexerQueryResultSchema,
)

log = logging.getLogger(__name__)


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

    def get_result(self, result_id: IndexerQueryResultId) -> IndexerQueryResultSchema:
        return IndexerQueryResultSchema.model_validate(
            self.db.get(IndexerQueryResult, result_id)
        )

    def save_result(self, result: IndexerQueryResultSchema) -> IndexerQueryResultSchema:
        log.debug("Saving indexer query result: %s", result)
        self.db.add(IndexerQueryResult(**result.model_dump()))
        self.db.commit()
        return result