MediaManager-maxdorninger-1/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 (
    IndexerQueryResult as IndexerQueryResultSchema,
)
from media_manager.indexer.schemas import (
    IndexerQueryResultId,
)

log = logging.getLogger(__name__)


class IndexerRepository:
    def __init__(self, db: Session) -> None:
        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:
        result_data = result.model_dump()
        result_data["download_url"] = str(
            result.download_url
        )  # this is the needful, because sqlalchemy is too dumb to handle the HttpUrl type

        self.db.add(IndexerQueryResult(**result_data))
        self.db.commit()
        return result