44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
|
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)]
|