from typing import Optional

from sqlmodel import Session

from app.models.confluence_site import ConfluenceSite
from app.repositories.base import BaseRepo


class ConfluenceSiteRepo(BaseRepo):
    model_cls = ConfluenceSite

    def get(self, session: Session, site_id: int) -> Optional[ConfluenceSite]:
        return session.get(ConfluenceSite, site_id)

    def must_get(self, session: Session, site_id: int) -> ConfluenceSite:
        confluence_site = self.get(session, site_id)
        if not confluence_site:
            raise Exception(f"Confluence site #{site_id} is not found")
        return confluence_site


confluence_site_repo = ConfluenceSiteRepo()