52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import inspect
|
|
from typing import Any
|
|
|
|
from fastapi import Request
|
|
from fastapi.exceptions import RequestValidationError
|
|
from pydantic import BaseModel, ValidationError
|
|
|
|
|
|
def BodyOrForm[T: BaseModel](model: type[T]):
|
|
async def dependency(
|
|
request: Request,
|
|
) -> T:
|
|
content_type = request.headers.get("content-type", "")
|
|
|
|
data: dict[str, Any] = {}
|
|
if "application/json" in content_type:
|
|
try:
|
|
data = await request.json()
|
|
except Exception:
|
|
raise RequestValidationError(
|
|
[
|
|
{
|
|
"loc": ("body",),
|
|
"msg": "Invalid JSON body",
|
|
"type": "value_error",
|
|
}
|
|
]
|
|
)
|
|
else:
|
|
form = await request.form()
|
|
data = dict(form)
|
|
|
|
try:
|
|
return model(**data)
|
|
except ValidationError as e:
|
|
raise RequestValidationError(e.errors())
|
|
|
|
dependency.__signature__ = inspect.signature( # pyright: ignore[reportFunctionMemberAccess]
|
|
lambda x: None
|
|
).replace(
|
|
parameters=[
|
|
inspect.Parameter(
|
|
name=model.__name__.lower(),
|
|
kind=inspect.Parameter.POSITIONAL_OR_KEYWORD,
|
|
annotation=model,
|
|
)
|
|
]
|
|
)
|
|
return dependency
|