mirror of
https://github.com/fastapi/fastapi.git
synced 2026-09-14 13:36:21 +08:00
0976185af9
* ✨ Pydantic v2 migration, initial implementation (#9500) * ✨ Add compat layer, for Pydantic v1 and v2 * ✨ Re-export Pydantic needed internals from compat, to later patch them for v1 * ♻️ Refactor internals to use new compatibility layers and run with Pydantic v2 * 📝 Update examples to run with Pydantic v2 * ✅ Update tests to use Pydantic v2 * 🎨 [pre-commit.ci] Auto format from pre-commit.com hooks * ✅ Temporarily disable Peewee tests, afterwards I'll enable them only for Pydantic v1 * 🐛 Fix JSON Schema generation and OpenAPI ref template * 🐛 Fix model field creation with defaults from Pydantic v2 * 🐛 Fix body field creation, with new FieldInfo * ✨ Use and check new ResponseValidationError for server validation errors * ✅ Fix test_schema_extra_examples tests with ResponseValidationError * ✅ Add dirty-equals to tests for compatibility with Pydantic v1 and v2 * ✨ Add util to regenerate errors with custom loc * ✨ Generate validation errors with loc * ✅ Update tests for compatibility with Pydantic v1 and v2 * ✅ Update tests for Pydantic v2 in tests/test_filter_pydantic_sub_model.py * ✅ Refactor tests in tests/test_dependency_overrides.py for Pydantic v2, separate parameterized into independent tests to use insert_assert * ✅ Refactor OpenAPI test for tests/test_infer_param_optionality.py for consistency, and make it compatible with Pydantic v1 and v2 * ✅ Update tests for tests/test_multi_query_errors.py for Pydantic v1 and v2 * ✅ Update tests for tests/test_multi_body_errors.py for Pydantic v1 and v2 * ✅ Update tests for tests/test_multi_body_errors.py for Pydantic v1 and v2 * 🎨 [pre-commit.ci] Auto format from pre-commit.com hooks * ♻️ Refactor tests for tests/test_path.py to inline pytest parameters, to make it easier to make them compatible with Pydantic v2 * ✅ Refactor and udpate tests for tests/test_path.py for Pydantic v1 and v2 * ♻️ Refactor and update tests for tests/test_query.py with compatibility for Pydantic v1 and v2 * ✅ Fix test with optional field without default None * ✅ Update tests for compatibility with Pydantic v2 * ✅ Update tutorial tests for Pydantic v2 * ♻️ Update OAuth2 dependencies for Pydantic v2 * ♻️ Refactor str check when checking for sequence types * ♻️ Rename regex to pattern to keep in sync with Pydantic v2 * ♻️ Refactor _compat.py, start moving conditional imports and declarations to specifics of Pydantic v1 or v2 * ✅ Update tests for OAuth2 security optional * ✅ Refactor tests for OAuth2 optional for Pydantic v2 * ✅ Refactor tests for OAuth2 security for compatibility with Pydantic v2 * 🐛 Fix location in compat layer for Pydantic v2 ModelField * ✅ Refactor tests for Pydantic v2 in tests/test_tutorial/test_bigger_applications/test_main_an_py39.py * 🐛 Add missing markers in Python 3.9 tests * ✅ Refactor tests for bigger apps for consistency with annotated ones and with support for Pydantic v2 * 🐛 Fix jsonable_encoder with new Pydantic v2 data types and Url * 🐛 Fix invalid JSON error for compatibility with Pydantic v2 * ✅ Update tests for behind_a_proxy for Pydantic v2 * ✅ Update tests for tests/test_tutorial/test_body/test_tutorial001_py310.py for Pydantic v2 * ✅ Update tests for tests/test_tutorial/test_body/test_tutorial001.py with Pydantic v2 and consistency with Python 3.10 tests * ✅ Fix tests for tutorial/body_fields for Pydantic v2 * ✅ Refactor tests for tutorial/body_multiple_params with Pydantic v2 * ✅ Update tests for tutorial/body_nested_models for Pydantic v2 * ✅ Update tests for tutorial/body_updates for Pydantic v2 * ✅ Update test for tutorial/cookie_params for Pydantic v2 * ✅ Fix tests for tests/test_tutorial/test_custom_request_and_route/test_tutorial002.py for Pydantic v2 * ✅ Update tests for tutorial/dataclasses for Pydantic v2 * ✅ Update tests for tutorial/dependencies for Pydantic v2 * ✅ Update tests for tutorial/extra_data_types for Pydantic v2 * ✅ Update tests for tutorial/handling_errors for Pydantic v2 * ✅ Fix test markers for Python 3.9 * ✅ Update tests for tutorial/header_params for Pydantic v2 * ✅ Update tests for Pydantic v2 in tests/test_tutorial/test_openapi_callbacks/test_tutorial001.py * ✅ Fix extra tests for Pydantic v2 * ✅ Refactor test for parameters, to later fix Pydantic v2 * ✅ Update tests for tutorial/query_params for Pydantic v2 * ♻️ Update examples in docs to use new pattern instead of the old regex * ✅ Fix several tests for Pydantic v2 * ✅ Update and fix test for ResponseValidationError * 🐛 Fix check for sequences vs scalars, include bytes as scalar * 🐛 Fix check for complex data types, include UploadFile * 🐛 Add list to sequence annotation types * 🐛 Fix checks for uploads and add utils to find if an annotation is an upload (or bytes) * ✨ Add UnionType and NoneType to compat layer * ✅ Update tests for request_files for compatibility with Pydantic v2 and consistency with other tests * ✅ Fix testsw for request_forms for Pydantic v2 * ✅ Fix tests for request_forms_and_files for Pydantic v2 * ✅ Fix tests in tutorial/security for compatibility with Pydantic v2 * ⬆️ Upgrade required version of email_validator * ✅ Fix tests for params repr * ✅ Add Pydantic v2 pytest markers * Use match_pydantic_error_url * 🎨 [pre-commit.ci] Auto format from pre-commit.com hooks * Use field_serializer instead of encoders in some tests * Show Undefined as ... in repr * Mark custom encoders test with xfail * Update test to reflect new serialization of Decimal as str * Use `model_validate` instead of `from_orm` * Update JSON schema to reflect required nullable * Add dirty-equals to pyproject.toml * Fix locs and error creation for use with pydantic 2.0a4 * Use the type adapter for serialization. This is hacky. * 🎨 [pre-commit.ci] Auto format from pre-commit.com hooks * ✅ Refactor test_multi_body_errors for compatibility with Pydantic v1 and v2 * ✅ Refactor test_custom_encoder for Pydantic v1 and v2 * ✅ Set input to None for now, for compatibility with current tests * 🐛 Fix passing serialization params to model field when handling the response * ♻️ Refactor exceptions to not depend on Pydantic ValidationError class * ♻️ Revert/refactor params to simplify repr * ✅ Tweak tests for custom class encoders for Pydantic v1 and v2 * ✅ Tweak tests for jsonable_encoder for Pydantic v1 and v2 * ✅ Tweak test for compatibility with Pydantic v1 and v2 * 🐛 Fix filtering data with subclasses * 🐛 Workaround examples in OpenAPI schema * ✅ Add skip marker for SQL tutorial, needs to be updated either way * ✅ Update test for broken JSON * ✅ Fix test for broken JSON * ✅ Update tests for timedeltas * ✅ Fix test for plain text validation errors * ✅ Add markers for Pydantic v1 exclusive tests (for now) * ✅ Update test for path_params with enums for compatibility with Pydantic v1 and v2 * ✅ Update tests for extra examples in OpenAPI * ✅ Fix tests for response_model with compatibility with Pydantic v1 and v2 * 🐛 Fix required double serialization for different types of models * ✅ Fix tests for response model with compatibility with new Pydantic v2 * 🐛 Import Undefined from compat layer * ✅ Fix tests for response_model for Pydantic v2 * ✅ Fix tests for schema_extra for Pydantic v2 * ✅ Add markers and update tests for Pydantic v2 * 💡 Comment out logic for double encoding that breaks other usecases * ✅ Update errors for int parsing * ♻️ Refactor re-enabling compatibility for Pydantic v1 * ♻️ Refactor OpenAPI utils to re-enable support for Pydantic v1 * ♻️ Refactor dependencies/utils and _compat for compatibility with Pydantic v1 * 🐛 Fix and tweak compatibility with Pydantic v1 and v2 in dependencies/utils * ✅ Tweak tests and examples for Pydantic v1 * ♻️ Tweak call to ModelField.validate for compatibility with Pydantic v1 * ✨ Use new global override TypeAdapter from_attributes * ✅ Update tests after updating from_attributes * 🔧 Update pytest config to avoid collecting tests from docs, useful for editor-integrated tests * ✅ Add test for data filtering, including inheritance and models in fields or lists of models * ♻️ Make OpenAPI models compatible with both Pydantic v1 and v2 * ♻️ Fix compatibility for Pydantic v1 and v2 in jsonable_encoder * ♻️ Fix compatibility in params with Pydantic v1 and v2 * ♻️ Fix compatibility when creating a FieldInfo in Pydantic v1 and v2 in utils.py * ♻️ Fix generation of flat_models and JSON Schema definitions in _compat.py for Pydantic v1 and v2 * ♻️ Update handling of ErrorWrappers for Pydantic v1 * ♻️ Refactor checks and handling of types an sequences * ♻️ Refactor and cleanup comments with compatibility for Pydantic v1 and v2 * ♻️ Update UploadFile for compatibility with both Pydantic v1 and v2 * 🔥 Remove commented out unneeded code * 🐛 Fix mock of get_annotation_from_field_info for Pydantic v1 and v2 * 🐛 Fix params with compatibility for Pydantic v1 and v2, with schemas and new pattern vs regex * 🐛 Fix check if field is sequence for Pydantic v1 * ✅ Fix tests for custom_schema_fields, for compatibility with Pydantic v1 and v2 * ✅ Simplify and fix tests for jsonable_encoder with compatibility for Pydantic v1 and v2 * ✅ Fix tests for orm_mode with Pydantic v1 and compatibility with Pydantic v2 * ♻️ Refactor logic for normalizing Pydantic v1 ErrorWrappers * ♻️ Workaround for params with examples, before defining what to deprecate in Pydantic v1 and v2 for examples with JSON Schema vs OpenAPI * ✅ Fix tests for Pydantic v1 and v2 for response_by_alias * ✅ Fix test for schema_extra with compatibility with Pydantic v1 and v2 * ♻️ Tweak error regeneration with loc * ♻️ Update error handling and serializationwith compatibility for Pydantic v1 and v2 * ♻️ Re-enable custom encoders for Pydantic v1 * ♻️ Update ErrorWrapper reserialization in Pydantic v1, do it outside of FastAPI ValidationExceptions * ✅ Update test for filter_submodel, re-structure to simplify testing while keeping division of Pydantic v1 and v2 * ✅ Refactor Pydantic v1 only test that requires modifying environment variables * 🔥 Update test for plaintext error responses, for Pydantic v1 and v2 * ⏪️ Revert changes in DB tutorial to use Pydantic v1 (the new guide will have SQLModel) * ✅ Mark current SQL DB tutorial tests as Pydantic only * ♻️ Update datastructures for compatibility with Pydantic v1, not requiring pydantic-core * ♻️ Update encoders.py for compatibility with Pydantic v1 * ⏪️ Revert changes to Peewee, the docs for that are gonna live in a new HowTo section, not in the main tutorials * ♻️ Simplify response body kwargs generation * 🔥 Clean up comments * 🔥 Clean some tests and comments * ✅ Refactor tests to match new Pydantic error string URLs * ✅ Refactor tests for recursive models for Pydantic v1 and v2 * ✅ Update tests for Peewee, re-enable, Pydantic-v1-only * ♻️ Update FastAPI params to take regex and pattern arguments * ⏪️ Revert tutorial examples for pattern, it will be done in a subsequent PR * ⏪️ Revert changes in schema extra examples, it will be added later in a docs-specific PR * 💡 Add TODO comment to document str validations with pattern * 🔥 Remove unneeded comment * 📌 Upgrade Pydantic pin dependency * ⬆️ Upgrade email_validator dependency * 🐛 Tweak type annotations in _compat.py * 🔇 Tweak mypy errors for compat, for Pydantic v1 re-imports * 🐛 Tweak and fix type annotations * ➕ Update requirements-test.txt, re-add dirty-equals * 🔥 Remove unnecessary config * 🐛 Tweak type annotations * 🔥 Remove unnecessary type in dependencies/utils.py * 💡 Update comment in routing.py --------- Co-authored-by: David Montague <35119617+dmontagu@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> * 👷 Add CI for both Pydantic v1 and v2 (#9688) * 👷 Test and install Pydantic v1 and v2 in CI * 💚 Tweak CI config for Pydantic v1 and v2 * 💚 Fix Pydantic v2 specification in CI * 🐛 Fix type annotations for compatibility with Python 3.7 * 💚 Install Pydantic v2 for lints * 🐛 Fix type annotations for Pydantic v2 * 💚 Re-use test cache for lint * ♻️ Refactor internals for test coverage and performance (#9691) * ♻️ Tweak import of Annotated from typing_extensions, they are installed anyway * ♻️ Refactor _compat to define functions for Pydantic v1 or v2 once instead of checking inside * ✅ Add test for UploadFile for Pydantic v2 * ♻️ Refactor types and remove logic for impossible cases * ✅ Add missing tests from test refactor for path params * ✅ Add tests for new decimal encoder * 💡 Add TODO comment for decimals in encoders * 🔥 Remove unneeded dummy function * 🔥 Remove section of code in field_annotation_is_scalar covered by sub-call to field_annotation_is_complex * ♻️ Refactor and tweak variables and types in _compat * ✅ Add tests for corner cases and compat with Pydantic v1 and v2 * ♻️ Refactor type annotations * 🔖 Release version 0.100.0-beta1 * ♻️ Refactor parts that use optional requirements to make them compatible with installations without them (#9707) * ♻️ Refactor parts that use optional requirements to make them compatible with installations without them * ♻️ Update JSON Schema for email field without email-validator installed * 🐛 Fix support for Pydantic v2.0, small changes in their final release (#9771) Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com> * 🔖 Release version 0.100.0-beta2 * ✨ OpenAPI 3.1.0 with Pydantic v2, merge `master` (#9773) * ➕ Add dirty-equals as a testing dependency (#9778) ➕ Add dirty-equals as a testing dependency, it seems it got lsot at some point * 🔀 Merge master, fix valid JSON Schema accepting bools (#9782) * ⏪️ Revert usage of custom logic for TypeAdapter JSON Schema, solved on the Pydantic side (#9787) ⏪️ Revert usage of custom logic for TypeAdapter JSON Schema, solved on Pydantic side * ♻️ Deprecate parameter `regex`, use `pattern` instead (#9786) * 📝 Update docs to deprecate regex, recommend pattern * ♻️ Update examples to use new pattern instead of regex * 📝 Add new example with deprecated regex * ♻️ Add deprecation notes and warnings for regex * ✅ Add tests for regex deprecation * ✅ Update tests for compatibility with Pydantic v1 * ✨ Update docs to use Pydantic v2 settings and add note and example about v1 (#9788) * ➕ Add pydantic-settings to all extras * 📝 Update docs for Pydantic settings * 📝 Update Settings source examples to use Pydantic v2, and add a Pydantic v1 version * ✅ Add tests for settings with Pydantic v1 and v2 * 🔥 Remove solved TODO comment * ♻️ Update conditional OpenAPI to use new Pydantic v2 settings * ✅ Update tests to import Annotated from typing_extensions for Python < 3.9 (#9795) * ➕ Add pydantic-extra-types to fastapi[extra] * ➕ temp: Install Pydantic from source to test JSON Schema metadata fixes (#9777) * ➕ Install Pydantic from source, from branch for JSON Schema with metadata * ➕ Update dependencies, install Pydantic main * ➕ Fix dependency URL for Pydantic from source * ➕ Add pydantic-settings for test requirements * 💡 Add TODO comments to re-enable Pydantic main (not from source) (#9796) * ✨ Add new Pydantic Field param options to Query, Cookie, Body, etc. (#9797) * 📝 Add docs for Pydantic v2 for `docs/en/docs/advanced/path-operation-advanced-configuration.md` (#9798) * 📝 Update docs in examples for settings with Pydantic v2 (#9799) * 📝 Update JSON Schema `examples` docs with Pydantic v2 (#9800) * ♻️ Use new Pydantic v2 JSON Schema generator (#9813) Co-authored-by: David Montague <35119617+dmontagu@users.noreply.github.com> * ♻️ Tweak type annotations and Pydantic version range (#9801) * 📌 Re-enable GA Pydantic, for v2, require minimum 2.0.2 (#9814) * 🔖 Release version 0.100.0-beta3 * 🔥 Remove duplicate type declaration from merge conflicts (#9832) * 👷♂️ Run tests with Pydantic v2 GA (#9830) 👷 Run tests for Pydantic v2 GA * 📝 Add notes to docs expecting Pydantic v2 and future updates (#9833) * 📝 Update index with new extras * 📝 Update release notes --------- Co-authored-by: David Montague <35119617+dmontagu@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Pastukhov Nikita <diementros@yandex.ru>
1359 lines
56 KiB
Python
1359 lines
56 KiB
Python
import asyncio
|
|
import dataclasses
|
|
import email.message
|
|
import inspect
|
|
import json
|
|
from contextlib import AsyncExitStack
|
|
from enum import Enum, IntEnum
|
|
from typing import (
|
|
Any,
|
|
Callable,
|
|
Coroutine,
|
|
Dict,
|
|
List,
|
|
Optional,
|
|
Sequence,
|
|
Set,
|
|
Tuple,
|
|
Type,
|
|
Union,
|
|
)
|
|
|
|
from fastapi import params
|
|
from fastapi._compat import (
|
|
ModelField,
|
|
Undefined,
|
|
_get_model_config,
|
|
_model_dump,
|
|
_normalize_errors,
|
|
lenient_issubclass,
|
|
)
|
|
from fastapi.datastructures import Default, DefaultPlaceholder
|
|
from fastapi.dependencies.models import Dependant
|
|
from fastapi.dependencies.utils import (
|
|
get_body_field,
|
|
get_dependant,
|
|
get_parameterless_sub_dependant,
|
|
get_typed_return_annotation,
|
|
solve_dependencies,
|
|
)
|
|
from fastapi.encoders import jsonable_encoder
|
|
from fastapi.exceptions import (
|
|
FastAPIError,
|
|
RequestValidationError,
|
|
ResponseValidationError,
|
|
WebSocketRequestValidationError,
|
|
)
|
|
from fastapi.types import DecoratedCallable, IncEx
|
|
from fastapi.utils import (
|
|
create_cloned_field,
|
|
create_response_field,
|
|
generate_unique_id,
|
|
get_value_or_default,
|
|
is_body_allowed_for_status_code,
|
|
)
|
|
from pydantic import BaseModel
|
|
from starlette import routing
|
|
from starlette.concurrency import run_in_threadpool
|
|
from starlette.exceptions import HTTPException
|
|
from starlette.requests import Request
|
|
from starlette.responses import JSONResponse, Response
|
|
from starlette.routing import (
|
|
BaseRoute,
|
|
Match,
|
|
compile_path,
|
|
get_name,
|
|
request_response,
|
|
websocket_session,
|
|
)
|
|
from starlette.routing import Mount as Mount # noqa
|
|
from starlette.types import ASGIApp, Lifespan, Scope
|
|
from starlette.websockets import WebSocket
|
|
|
|
|
|
def _prepare_response_content(
|
|
res: Any,
|
|
*,
|
|
exclude_unset: bool,
|
|
exclude_defaults: bool = False,
|
|
exclude_none: bool = False,
|
|
) -> Any:
|
|
if isinstance(res, BaseModel):
|
|
read_with_orm_mode = getattr(_get_model_config(res), "read_with_orm_mode", None)
|
|
if read_with_orm_mode:
|
|
# Let from_orm extract the data from this model instead of converting
|
|
# it now to a dict.
|
|
# Otherwise there's no way to extract lazy data that requires attribute
|
|
# access instead of dict iteration, e.g. lazy relationships.
|
|
return res
|
|
return _model_dump(
|
|
res,
|
|
by_alias=True,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
elif isinstance(res, list):
|
|
return [
|
|
_prepare_response_content(
|
|
item,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
for item in res
|
|
]
|
|
elif isinstance(res, dict):
|
|
return {
|
|
k: _prepare_response_content(
|
|
v,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
for k, v in res.items()
|
|
}
|
|
elif dataclasses.is_dataclass(res):
|
|
return dataclasses.asdict(res)
|
|
return res
|
|
|
|
|
|
async def serialize_response(
|
|
*,
|
|
field: Optional[ModelField] = None,
|
|
response_content: Any,
|
|
include: Optional[IncEx] = None,
|
|
exclude: Optional[IncEx] = None,
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = False,
|
|
exclude_defaults: bool = False,
|
|
exclude_none: bool = False,
|
|
is_coroutine: bool = True,
|
|
) -> Any:
|
|
if field:
|
|
errors = []
|
|
if not hasattr(field, "serialize"):
|
|
# pydantic v1
|
|
response_content = _prepare_response_content(
|
|
response_content,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
if is_coroutine:
|
|
value, errors_ = field.validate(response_content, {}, loc=("response",))
|
|
else:
|
|
value, errors_ = await run_in_threadpool(
|
|
field.validate, response_content, {}, loc=("response",)
|
|
)
|
|
if isinstance(errors_, list):
|
|
errors.extend(errors_)
|
|
elif errors_:
|
|
errors.append(errors_)
|
|
if errors:
|
|
raise ResponseValidationError(
|
|
errors=_normalize_errors(errors), body=response_content
|
|
)
|
|
|
|
if hasattr(field, "serialize"):
|
|
return field.serialize(
|
|
value,
|
|
include=include,
|
|
exclude=exclude,
|
|
by_alias=by_alias,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
|
|
return jsonable_encoder(
|
|
value,
|
|
include=include,
|
|
exclude=exclude,
|
|
by_alias=by_alias,
|
|
exclude_unset=exclude_unset,
|
|
exclude_defaults=exclude_defaults,
|
|
exclude_none=exclude_none,
|
|
)
|
|
else:
|
|
return jsonable_encoder(response_content)
|
|
|
|
|
|
async def run_endpoint_function(
|
|
*, dependant: Dependant, values: Dict[str, Any], is_coroutine: bool
|
|
) -> Any:
|
|
# Only called by get_request_handler. Has been split into its own function to
|
|
# facilitate profiling endpoints, since inner functions are harder to profile.
|
|
assert dependant.call is not None, "dependant.call must be a function"
|
|
|
|
if is_coroutine:
|
|
return await dependant.call(**values)
|
|
else:
|
|
return await run_in_threadpool(dependant.call, **values)
|
|
|
|
|
|
def get_request_handler(
|
|
dependant: Dependant,
|
|
body_field: Optional[ModelField] = None,
|
|
status_code: Optional[int] = None,
|
|
response_class: Union[Type[Response], DefaultPlaceholder] = Default(JSONResponse),
|
|
response_field: Optional[ModelField] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
dependency_overrides_provider: Optional[Any] = None,
|
|
) -> Callable[[Request], Coroutine[Any, Any, Response]]:
|
|
assert dependant.call is not None, "dependant.call must be a function"
|
|
is_coroutine = asyncio.iscoroutinefunction(dependant.call)
|
|
is_body_form = body_field and isinstance(body_field.field_info, params.Form)
|
|
if isinstance(response_class, DefaultPlaceholder):
|
|
actual_response_class: Type[Response] = response_class.value
|
|
else:
|
|
actual_response_class = response_class
|
|
|
|
async def app(request: Request) -> Response:
|
|
try:
|
|
body: Any = None
|
|
if body_field:
|
|
if is_body_form:
|
|
body = await request.form()
|
|
stack = request.scope.get("fastapi_astack")
|
|
assert isinstance(stack, AsyncExitStack)
|
|
stack.push_async_callback(body.close)
|
|
else:
|
|
body_bytes = await request.body()
|
|
if body_bytes:
|
|
json_body: Any = Undefined
|
|
content_type_value = request.headers.get("content-type")
|
|
if not content_type_value:
|
|
json_body = await request.json()
|
|
else:
|
|
message = email.message.Message()
|
|
message["content-type"] = content_type_value
|
|
if message.get_content_maintype() == "application":
|
|
subtype = message.get_content_subtype()
|
|
if subtype == "json" or subtype.endswith("+json"):
|
|
json_body = await request.json()
|
|
if json_body != Undefined:
|
|
body = json_body
|
|
else:
|
|
body = body_bytes
|
|
except json.JSONDecodeError as e:
|
|
raise RequestValidationError(
|
|
[
|
|
{
|
|
"type": "json_invalid",
|
|
"loc": ("body", e.pos),
|
|
"msg": "JSON decode error",
|
|
"input": {},
|
|
"ctx": {"error": e.msg},
|
|
}
|
|
],
|
|
body=e.doc,
|
|
) from e
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(
|
|
status_code=400, detail="There was an error parsing the body"
|
|
) from e
|
|
solved_result = await solve_dependencies(
|
|
request=request,
|
|
dependant=dependant,
|
|
body=body,
|
|
dependency_overrides_provider=dependency_overrides_provider,
|
|
)
|
|
values, errors, background_tasks, sub_response, _ = solved_result
|
|
if errors:
|
|
raise RequestValidationError(_normalize_errors(errors), body=body)
|
|
else:
|
|
raw_response = await run_endpoint_function(
|
|
dependant=dependant, values=values, is_coroutine=is_coroutine
|
|
)
|
|
|
|
if isinstance(raw_response, Response):
|
|
if raw_response.background is None:
|
|
raw_response.background = background_tasks
|
|
return raw_response
|
|
response_args: Dict[str, Any] = {"background": background_tasks}
|
|
# If status_code was set, use it, otherwise use the default from the
|
|
# response class, in the case of redirect it's 307
|
|
current_status_code = (
|
|
status_code if status_code else sub_response.status_code
|
|
)
|
|
if current_status_code is not None:
|
|
response_args["status_code"] = current_status_code
|
|
if sub_response.status_code:
|
|
response_args["status_code"] = sub_response.status_code
|
|
content = await serialize_response(
|
|
field=response_field,
|
|
response_content=raw_response,
|
|
include=response_model_include,
|
|
exclude=response_model_exclude,
|
|
by_alias=response_model_by_alias,
|
|
exclude_unset=response_model_exclude_unset,
|
|
exclude_defaults=response_model_exclude_defaults,
|
|
exclude_none=response_model_exclude_none,
|
|
is_coroutine=is_coroutine,
|
|
)
|
|
response = actual_response_class(content, **response_args)
|
|
if not is_body_allowed_for_status_code(response.status_code):
|
|
response.body = b""
|
|
response.headers.raw.extend(sub_response.headers.raw)
|
|
return response
|
|
|
|
return app
|
|
|
|
|
|
def get_websocket_app(
|
|
dependant: Dependant, dependency_overrides_provider: Optional[Any] = None
|
|
) -> Callable[[WebSocket], Coroutine[Any, Any, Any]]:
|
|
async def app(websocket: WebSocket) -> None:
|
|
solved_result = await solve_dependencies(
|
|
request=websocket,
|
|
dependant=dependant,
|
|
dependency_overrides_provider=dependency_overrides_provider,
|
|
)
|
|
values, errors, _, _2, _3 = solved_result
|
|
if errors:
|
|
raise WebSocketRequestValidationError(_normalize_errors(errors))
|
|
assert dependant.call is not None, "dependant.call must be a function"
|
|
await dependant.call(**values)
|
|
|
|
return app
|
|
|
|
|
|
class APIWebSocketRoute(routing.WebSocketRoute):
|
|
def __init__(
|
|
self,
|
|
path: str,
|
|
endpoint: Callable[..., Any],
|
|
*,
|
|
name: Optional[str] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
dependency_overrides_provider: Optional[Any] = None,
|
|
) -> None:
|
|
self.path = path
|
|
self.endpoint = endpoint
|
|
self.name = get_name(endpoint) if name is None else name
|
|
self.dependencies = list(dependencies or [])
|
|
self.path_regex, self.path_format, self.param_convertors = compile_path(path)
|
|
self.dependant = get_dependant(path=self.path_format, call=self.endpoint)
|
|
for depends in self.dependencies[::-1]:
|
|
self.dependant.dependencies.insert(
|
|
0,
|
|
get_parameterless_sub_dependant(depends=depends, path=self.path_format),
|
|
)
|
|
|
|
self.app = websocket_session(
|
|
get_websocket_app(
|
|
dependant=self.dependant,
|
|
dependency_overrides_provider=dependency_overrides_provider,
|
|
)
|
|
)
|
|
|
|
def matches(self, scope: Scope) -> Tuple[Match, Scope]:
|
|
match, child_scope = super().matches(scope)
|
|
if match != Match.NONE:
|
|
child_scope["route"] = self
|
|
return match, child_scope
|
|
|
|
|
|
class APIRoute(routing.Route):
|
|
def __init__(
|
|
self,
|
|
path: str,
|
|
endpoint: Callable[..., Any],
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
name: Optional[str] = None,
|
|
methods: Optional[Union[Set[str], List[str]]] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Union[Type[Response], DefaultPlaceholder] = Default(
|
|
JSONResponse
|
|
),
|
|
dependency_overrides_provider: Optional[Any] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Union[
|
|
Callable[["APIRoute"], str], DefaultPlaceholder
|
|
] = Default(generate_unique_id),
|
|
) -> None:
|
|
self.path = path
|
|
self.endpoint = endpoint
|
|
if isinstance(response_model, DefaultPlaceholder):
|
|
return_annotation = get_typed_return_annotation(endpoint)
|
|
if lenient_issubclass(return_annotation, Response):
|
|
response_model = None
|
|
else:
|
|
response_model = return_annotation
|
|
self.response_model = response_model
|
|
self.summary = summary
|
|
self.response_description = response_description
|
|
self.deprecated = deprecated
|
|
self.operation_id = operation_id
|
|
self.response_model_include = response_model_include
|
|
self.response_model_exclude = response_model_exclude
|
|
self.response_model_by_alias = response_model_by_alias
|
|
self.response_model_exclude_unset = response_model_exclude_unset
|
|
self.response_model_exclude_defaults = response_model_exclude_defaults
|
|
self.response_model_exclude_none = response_model_exclude_none
|
|
self.include_in_schema = include_in_schema
|
|
self.response_class = response_class
|
|
self.dependency_overrides_provider = dependency_overrides_provider
|
|
self.callbacks = callbacks
|
|
self.openapi_extra = openapi_extra
|
|
self.generate_unique_id_function = generate_unique_id_function
|
|
self.tags = tags or []
|
|
self.responses = responses or {}
|
|
self.name = get_name(endpoint) if name is None else name
|
|
self.path_regex, self.path_format, self.param_convertors = compile_path(path)
|
|
if methods is None:
|
|
methods = ["GET"]
|
|
self.methods: Set[str] = {method.upper() for method in methods}
|
|
if isinstance(generate_unique_id_function, DefaultPlaceholder):
|
|
current_generate_unique_id: Callable[
|
|
["APIRoute"], str
|
|
] = generate_unique_id_function.value
|
|
else:
|
|
current_generate_unique_id = generate_unique_id_function
|
|
self.unique_id = self.operation_id or current_generate_unique_id(self)
|
|
# normalize enums e.g. http.HTTPStatus
|
|
if isinstance(status_code, IntEnum):
|
|
status_code = int(status_code)
|
|
self.status_code = status_code
|
|
if self.response_model:
|
|
assert is_body_allowed_for_status_code(
|
|
status_code
|
|
), f"Status code {status_code} must not have a response body"
|
|
response_name = "Response_" + self.unique_id
|
|
self.response_field = create_response_field(
|
|
name=response_name,
|
|
type_=self.response_model,
|
|
# TODO: This should actually set mode='serialization', just, that changes the schemas
|
|
# mode="serialization",
|
|
mode="validation",
|
|
)
|
|
# Create a clone of the field, so that a Pydantic submodel is not returned
|
|
# as is just because it's an instance of a subclass of a more limited class
|
|
# e.g. UserInDB (containing hashed_password) could be a subclass of User
|
|
# that doesn't have the hashed_password. But because it's a subclass, it
|
|
# would pass the validation and be returned as is.
|
|
# By being a new field, no inheritance will be passed as is. A new model
|
|
# will be always created.
|
|
# TODO: remove when deprecating Pydantic v1
|
|
self.secure_cloned_response_field: Optional[
|
|
ModelField
|
|
] = create_cloned_field(self.response_field)
|
|
else:
|
|
self.response_field = None # type: ignore
|
|
self.secure_cloned_response_field = None
|
|
self.dependencies = list(dependencies or [])
|
|
self.description = description or inspect.cleandoc(self.endpoint.__doc__ or "")
|
|
# if a "form feed" character (page break) is found in the description text,
|
|
# truncate description text to the content preceding the first "form feed"
|
|
self.description = self.description.split("\f")[0].strip()
|
|
response_fields = {}
|
|
for additional_status_code, response in self.responses.items():
|
|
assert isinstance(response, dict), "An additional response must be a dict"
|
|
model = response.get("model")
|
|
if model:
|
|
assert is_body_allowed_for_status_code(
|
|
additional_status_code
|
|
), f"Status code {additional_status_code} must not have a response body"
|
|
response_name = f"Response_{additional_status_code}_{self.unique_id}"
|
|
response_field = create_response_field(name=response_name, type_=model)
|
|
response_fields[additional_status_code] = response_field
|
|
if response_fields:
|
|
self.response_fields: Dict[Union[int, str], ModelField] = response_fields
|
|
else:
|
|
self.response_fields = {}
|
|
|
|
assert callable(endpoint), "An endpoint must be a callable"
|
|
self.dependant = get_dependant(path=self.path_format, call=self.endpoint)
|
|
for depends in self.dependencies[::-1]:
|
|
self.dependant.dependencies.insert(
|
|
0,
|
|
get_parameterless_sub_dependant(depends=depends, path=self.path_format),
|
|
)
|
|
self.body_field = get_body_field(dependant=self.dependant, name=self.unique_id)
|
|
self.app = request_response(self.get_route_handler())
|
|
|
|
def get_route_handler(self) -> Callable[[Request], Coroutine[Any, Any, Response]]:
|
|
return get_request_handler(
|
|
dependant=self.dependant,
|
|
body_field=self.body_field,
|
|
status_code=self.status_code,
|
|
response_class=self.response_class,
|
|
response_field=self.secure_cloned_response_field,
|
|
response_model_include=self.response_model_include,
|
|
response_model_exclude=self.response_model_exclude,
|
|
response_model_by_alias=self.response_model_by_alias,
|
|
response_model_exclude_unset=self.response_model_exclude_unset,
|
|
response_model_exclude_defaults=self.response_model_exclude_defaults,
|
|
response_model_exclude_none=self.response_model_exclude_none,
|
|
dependency_overrides_provider=self.dependency_overrides_provider,
|
|
)
|
|
|
|
def matches(self, scope: Scope) -> Tuple[Match, Scope]:
|
|
match, child_scope = super().matches(scope)
|
|
if match != Match.NONE:
|
|
child_scope["route"] = self
|
|
return match, child_scope
|
|
|
|
|
|
class APIRouter(routing.Router):
|
|
def __init__(
|
|
self,
|
|
*,
|
|
prefix: str = "",
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
default_response_class: Type[Response] = Default(JSONResponse),
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
routes: Optional[List[routing.BaseRoute]] = None,
|
|
redirect_slashes: bool = True,
|
|
default: Optional[ASGIApp] = None,
|
|
dependency_overrides_provider: Optional[Any] = None,
|
|
route_class: Type[APIRoute] = APIRoute,
|
|
on_startup: Optional[Sequence[Callable[[], Any]]] = None,
|
|
on_shutdown: Optional[Sequence[Callable[[], Any]]] = None,
|
|
# the generic to Lifespan[AppType] is the type of the top level application
|
|
# which the router cannot know statically, so we use typing.Any
|
|
lifespan: Optional[Lifespan[Any]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
include_in_schema: bool = True,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> None:
|
|
super().__init__(
|
|
routes=routes,
|
|
redirect_slashes=redirect_slashes,
|
|
default=default,
|
|
on_startup=on_startup,
|
|
on_shutdown=on_shutdown,
|
|
lifespan=lifespan,
|
|
)
|
|
if prefix:
|
|
assert prefix.startswith("/"), "A path prefix must start with '/'"
|
|
assert not prefix.endswith(
|
|
"/"
|
|
), "A path prefix must not end with '/', as the routes will start with '/'"
|
|
self.prefix = prefix
|
|
self.tags: List[Union[str, Enum]] = tags or []
|
|
self.dependencies = list(dependencies or [])
|
|
self.deprecated = deprecated
|
|
self.include_in_schema = include_in_schema
|
|
self.responses = responses or {}
|
|
self.callbacks = callbacks or []
|
|
self.dependency_overrides_provider = dependency_overrides_provider
|
|
self.route_class = route_class
|
|
self.default_response_class = default_response_class
|
|
self.generate_unique_id_function = generate_unique_id_function
|
|
|
|
def route(
|
|
self,
|
|
path: str,
|
|
methods: Optional[List[str]] = None,
|
|
name: Optional[str] = None,
|
|
include_in_schema: bool = True,
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
def decorator(func: DecoratedCallable) -> DecoratedCallable:
|
|
self.add_route(
|
|
path,
|
|
func,
|
|
methods=methods,
|
|
name=name,
|
|
include_in_schema=include_in_schema,
|
|
)
|
|
return func
|
|
|
|
return decorator
|
|
|
|
def add_api_route(
|
|
self,
|
|
path: str,
|
|
endpoint: Callable[..., Any],
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
methods: Optional[Union[Set[str], List[str]]] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Union[Type[Response], DefaultPlaceholder] = Default(
|
|
JSONResponse
|
|
),
|
|
name: Optional[str] = None,
|
|
route_class_override: Optional[Type[APIRoute]] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Union[
|
|
Callable[[APIRoute], str], DefaultPlaceholder
|
|
] = Default(generate_unique_id),
|
|
) -> None:
|
|
route_class = route_class_override or self.route_class
|
|
responses = responses or {}
|
|
combined_responses = {**self.responses, **responses}
|
|
current_response_class = get_value_or_default(
|
|
response_class, self.default_response_class
|
|
)
|
|
current_tags = self.tags.copy()
|
|
if tags:
|
|
current_tags.extend(tags)
|
|
current_dependencies = self.dependencies.copy()
|
|
if dependencies:
|
|
current_dependencies.extend(dependencies)
|
|
current_callbacks = self.callbacks.copy()
|
|
if callbacks:
|
|
current_callbacks.extend(callbacks)
|
|
current_generate_unique_id = get_value_or_default(
|
|
generate_unique_id_function, self.generate_unique_id_function
|
|
)
|
|
route = route_class(
|
|
self.prefix + path,
|
|
endpoint=endpoint,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=current_tags,
|
|
dependencies=current_dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=combined_responses,
|
|
deprecated=deprecated or self.deprecated,
|
|
methods=methods,
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema and self.include_in_schema,
|
|
response_class=current_response_class,
|
|
name=name,
|
|
dependency_overrides_provider=self.dependency_overrides_provider,
|
|
callbacks=current_callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=current_generate_unique_id,
|
|
)
|
|
self.routes.append(route)
|
|
|
|
def api_route(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
methods: Optional[List[str]] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
def decorator(func: DecoratedCallable) -> DecoratedCallable:
|
|
self.add_api_route(
|
|
path,
|
|
func,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=methods,
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
return func
|
|
|
|
return decorator
|
|
|
|
def add_api_websocket_route(
|
|
self,
|
|
path: str,
|
|
endpoint: Callable[..., Any],
|
|
name: Optional[str] = None,
|
|
*,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
) -> None:
|
|
current_dependencies = self.dependencies.copy()
|
|
if dependencies:
|
|
current_dependencies.extend(dependencies)
|
|
|
|
route = APIWebSocketRoute(
|
|
self.prefix + path,
|
|
endpoint=endpoint,
|
|
name=name,
|
|
dependencies=current_dependencies,
|
|
dependency_overrides_provider=self.dependency_overrides_provider,
|
|
)
|
|
self.routes.append(route)
|
|
|
|
def websocket(
|
|
self,
|
|
path: str,
|
|
name: Optional[str] = None,
|
|
*,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
def decorator(func: DecoratedCallable) -> DecoratedCallable:
|
|
self.add_api_websocket_route(
|
|
path, func, name=name, dependencies=dependencies
|
|
)
|
|
return func
|
|
|
|
return decorator
|
|
|
|
def websocket_route(
|
|
self, path: str, name: Union[str, None] = None
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
def decorator(func: DecoratedCallable) -> DecoratedCallable:
|
|
self.add_websocket_route(path, func, name=name)
|
|
return func
|
|
|
|
return decorator
|
|
|
|
def include_router(
|
|
self,
|
|
router: "APIRouter",
|
|
*,
|
|
prefix: str = "",
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
default_response_class: Type[Response] = Default(JSONResponse),
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
include_in_schema: bool = True,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> None:
|
|
if prefix:
|
|
assert prefix.startswith("/"), "A path prefix must start with '/'"
|
|
assert not prefix.endswith(
|
|
"/"
|
|
), "A path prefix must not end with '/', as the routes will start with '/'"
|
|
else:
|
|
for r in router.routes:
|
|
path = getattr(r, "path") # noqa: B009
|
|
name = getattr(r, "name", "unknown")
|
|
if path is not None and not path:
|
|
raise FastAPIError(
|
|
f"Prefix and path cannot be both empty (path operation: {name})"
|
|
)
|
|
if responses is None:
|
|
responses = {}
|
|
for route in router.routes:
|
|
if isinstance(route, APIRoute):
|
|
combined_responses = {**responses, **route.responses}
|
|
use_response_class = get_value_or_default(
|
|
route.response_class,
|
|
router.default_response_class,
|
|
default_response_class,
|
|
self.default_response_class,
|
|
)
|
|
current_tags = []
|
|
if tags:
|
|
current_tags.extend(tags)
|
|
if route.tags:
|
|
current_tags.extend(route.tags)
|
|
current_dependencies: List[params.Depends] = []
|
|
if dependencies:
|
|
current_dependencies.extend(dependencies)
|
|
if route.dependencies:
|
|
current_dependencies.extend(route.dependencies)
|
|
current_callbacks = []
|
|
if callbacks:
|
|
current_callbacks.extend(callbacks)
|
|
if route.callbacks:
|
|
current_callbacks.extend(route.callbacks)
|
|
current_generate_unique_id = get_value_or_default(
|
|
route.generate_unique_id_function,
|
|
router.generate_unique_id_function,
|
|
generate_unique_id_function,
|
|
self.generate_unique_id_function,
|
|
)
|
|
self.add_api_route(
|
|
prefix + route.path,
|
|
route.endpoint,
|
|
response_model=route.response_model,
|
|
status_code=route.status_code,
|
|
tags=current_tags,
|
|
dependencies=current_dependencies,
|
|
summary=route.summary,
|
|
description=route.description,
|
|
response_description=route.response_description,
|
|
responses=combined_responses,
|
|
deprecated=route.deprecated or deprecated or self.deprecated,
|
|
methods=route.methods,
|
|
operation_id=route.operation_id,
|
|
response_model_include=route.response_model_include,
|
|
response_model_exclude=route.response_model_exclude,
|
|
response_model_by_alias=route.response_model_by_alias,
|
|
response_model_exclude_unset=route.response_model_exclude_unset,
|
|
response_model_exclude_defaults=route.response_model_exclude_defaults,
|
|
response_model_exclude_none=route.response_model_exclude_none,
|
|
include_in_schema=route.include_in_schema
|
|
and self.include_in_schema
|
|
and include_in_schema,
|
|
response_class=use_response_class,
|
|
name=route.name,
|
|
route_class_override=type(route),
|
|
callbacks=current_callbacks,
|
|
openapi_extra=route.openapi_extra,
|
|
generate_unique_id_function=current_generate_unique_id,
|
|
)
|
|
elif isinstance(route, routing.Route):
|
|
methods = list(route.methods or [])
|
|
self.add_route(
|
|
prefix + route.path,
|
|
route.endpoint,
|
|
methods=methods,
|
|
include_in_schema=route.include_in_schema,
|
|
name=route.name,
|
|
)
|
|
elif isinstance(route, APIWebSocketRoute):
|
|
current_dependencies = []
|
|
if dependencies:
|
|
current_dependencies.extend(dependencies)
|
|
if route.dependencies:
|
|
current_dependencies.extend(route.dependencies)
|
|
self.add_api_websocket_route(
|
|
prefix + route.path,
|
|
route.endpoint,
|
|
dependencies=current_dependencies,
|
|
name=route.name,
|
|
)
|
|
elif isinstance(route, routing.WebSocketRoute):
|
|
self.add_websocket_route(
|
|
prefix + route.path, route.endpoint, name=route.name
|
|
)
|
|
for handler in router.on_startup:
|
|
self.add_event_handler("startup", handler)
|
|
for handler in router.on_shutdown:
|
|
self.add_event_handler("shutdown", handler)
|
|
|
|
def get(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["GET"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def put(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["PUT"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def post(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["POST"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def delete(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["DELETE"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def options(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["OPTIONS"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def head(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["HEAD"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def patch(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["PATCH"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def trace(
|
|
self,
|
|
path: str,
|
|
*,
|
|
response_model: Any = Default(None),
|
|
status_code: Optional[int] = None,
|
|
tags: Optional[List[Union[str, Enum]]] = None,
|
|
dependencies: Optional[Sequence[params.Depends]] = None,
|
|
summary: Optional[str] = None,
|
|
description: Optional[str] = None,
|
|
response_description: str = "Successful Response",
|
|
responses: Optional[Dict[Union[int, str], Dict[str, Any]]] = None,
|
|
deprecated: Optional[bool] = None,
|
|
operation_id: Optional[str] = None,
|
|
response_model_include: Optional[IncEx] = None,
|
|
response_model_exclude: Optional[IncEx] = None,
|
|
response_model_by_alias: bool = True,
|
|
response_model_exclude_unset: bool = False,
|
|
response_model_exclude_defaults: bool = False,
|
|
response_model_exclude_none: bool = False,
|
|
include_in_schema: bool = True,
|
|
response_class: Type[Response] = Default(JSONResponse),
|
|
name: Optional[str] = None,
|
|
callbacks: Optional[List[BaseRoute]] = None,
|
|
openapi_extra: Optional[Dict[str, Any]] = None,
|
|
generate_unique_id_function: Callable[[APIRoute], str] = Default(
|
|
generate_unique_id
|
|
),
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
return self.api_route(
|
|
path=path,
|
|
response_model=response_model,
|
|
status_code=status_code,
|
|
tags=tags,
|
|
dependencies=dependencies,
|
|
summary=summary,
|
|
description=description,
|
|
response_description=response_description,
|
|
responses=responses,
|
|
deprecated=deprecated,
|
|
methods=["TRACE"],
|
|
operation_id=operation_id,
|
|
response_model_include=response_model_include,
|
|
response_model_exclude=response_model_exclude,
|
|
response_model_by_alias=response_model_by_alias,
|
|
response_model_exclude_unset=response_model_exclude_unset,
|
|
response_model_exclude_defaults=response_model_exclude_defaults,
|
|
response_model_exclude_none=response_model_exclude_none,
|
|
include_in_schema=include_in_schema,
|
|
response_class=response_class,
|
|
name=name,
|
|
callbacks=callbacks,
|
|
openapi_extra=openapi_extra,
|
|
generate_unique_id_function=generate_unique_id_function,
|
|
)
|
|
|
|
def on_event(
|
|
self, event_type: str
|
|
) -> Callable[[DecoratedCallable], DecoratedCallable]:
|
|
def decorator(func: DecoratedCallable) -> DecoratedCallable:
|
|
self.add_event_handler(event_type, func)
|
|
return func
|
|
|
|
return decorator
|