cloud-api/src/organisation/dependencies.py

"""
Router dependencies for organisation module

Classes:
 - List: Description
 - Classes: Description

Functions:
 - List: Description
 - Functions: Description
"""
from typing import Annotated

from fastapi import Depends, Query

from src.database import db_dependency

from src.organisation.schemas import OrgIDMixin
from src.organisation.models import Organisation as Org
from src.organisation.exceptions import OrgNotFoundException


def get_org_model_query(db: db_dependency, org_id: Annotated[int, Query(gt=0)]) -> type[Org]:
	org_model = db.get(Org, org_id)
	if org_model is None:
		raise OrgNotFoundException(org_id)

	return org_model

org_model_query_dependency = Annotated[type[Org], Depends(get_org_model_query)]


def get_org_model_body(db: db_dependency, request_model: OrgIDMixin) -> type[Org]:
	org_id = getattr(request_model, "organisation_id", None)
	if org_id is None:
		raise OrgNotFoundException
	org_model = db.get(Org, org_id)
	if org_model is None:
		raise OrgNotFoundException(org_id)

	return org_model

org_model_body_dependency = Annotated[type[Org], Depends(get_org_model_body)]