luxferre
657f91d73d
Org endpoints use query/body model dependencies to perform initial db lookups. Issue #6 Org ID path params have been replaced with either query params (get endpoints) or body values. Resolves #10 Endpoints in other modules that rely on an org model lookup have also been updated.
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
"""
|
|
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)]
|