cloud-api/src/organisation/dependencies.py

"""
Dependencies related to the organisation module

Exports:
 - org_model_query_dependency: org_model: Gets org model from db, if it exists. Uses org_id from query param. Also verifies if the org has been approved.
 - org_model_body_dependency: org_model: Gets org model from db, if it exists. Uses org_id from request body. Also verifies if the org has been approved.
"""

from typing import Annotated, Optional

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: Optional[int] = 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)]