diff options
author | cyfraeviolae <cyfraeviolae> | 2024-04-03 03:17:55 -0400 |
---|---|---|
committer | cyfraeviolae <cyfraeviolae> | 2024-04-03 03:17:55 -0400 |
commit | 12cf076118570eebbff08c6b3090e0d4798447a1 (patch) | |
tree | 3ba25e17e3c3a5e82316558ba3864b955919ff72 /venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation | |
parent | c45662ff3923b34614ddcc8feb9195541166dcc5 (diff) |
no venv
Diffstat (limited to 'venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation')
20 files changed, 0 insertions, 1092 deletions
diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__init__.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__init__.py deleted file mode 100644 index 8b9183e..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__init__.py +++ /dev/null @@ -1,7 +0,0 @@ -from .plugins import openapi_schema_plugins -from .schema import SchemaCreator - -__all__ = ( - "SchemaCreator", - "openapi_schema_plugins", -) diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/__init__.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/__init__.cpython-311.pyc Binary files differdeleted file mode 100644 index 60cdb7e..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/__init__.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/constrained_fields.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/constrained_fields.cpython-311.pyc Binary files differdeleted file mode 100644 index 156b683..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/constrained_fields.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/examples.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/examples.cpython-311.pyc Binary files differdeleted file mode 100644 index 1d1a327..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/examples.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/schema.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/schema.cpython-311.pyc Binary files differdeleted file mode 100644 index c382161..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/schema.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/utils.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/utils.cpython-311.pyc Binary files differdeleted file mode 100644 index 8f1ed6b..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/__pycache__/utils.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/constrained_fields.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/constrained_fields.py deleted file mode 100644 index 80f355d..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/constrained_fields.py +++ /dev/null @@ -1,91 +0,0 @@ -from __future__ import annotations - -from datetime import date, datetime, timezone -from re import Pattern -from typing import TYPE_CHECKING - -from litestar.openapi.spec.enums import OpenAPIFormat, OpenAPIType -from litestar.openapi.spec.schema import Schema - -if TYPE_CHECKING: - from decimal import Decimal - - from litestar.params import KwargDefinition - -__all__ = ( - "create_date_constrained_field_schema", - "create_numerical_constrained_field_schema", - "create_string_constrained_field_schema", -) - - -def create_numerical_constrained_field_schema( - field_type: type[int] | type[float] | type[Decimal], - kwarg_definition: KwargDefinition, -) -> Schema: - """Create Schema from Constrained Int/Float/Decimal field.""" - schema = Schema(type=OpenAPIType.INTEGER if issubclass(field_type, int) else OpenAPIType.NUMBER) - if kwarg_definition.le is not None: - schema.maximum = float(kwarg_definition.le) - if kwarg_definition.lt is not None: - schema.exclusive_maximum = float(kwarg_definition.lt) - if kwarg_definition.ge is not None: - schema.minimum = float(kwarg_definition.ge) - if kwarg_definition.gt is not None: - schema.exclusive_minimum = float(kwarg_definition.gt) - if kwarg_definition.multiple_of is not None: - schema.multiple_of = float(kwarg_definition.multiple_of) - return schema - - -def create_date_constrained_field_schema( - field_type: type[date] | type[datetime], - kwarg_definition: KwargDefinition, -) -> Schema: - """Create Schema from Constrained Date Field.""" - schema = Schema( - type=OpenAPIType.STRING, format=OpenAPIFormat.DATE if issubclass(field_type, date) else OpenAPIFormat.DATE_TIME - ) - for kwarg_definition_attr, schema_attr in [ - ("le", "maximum"), - ("lt", "exclusive_maximum"), - ("ge", "minimum"), - ("gt", "exclusive_minimum"), - ]: - if attr := getattr(kwarg_definition, kwarg_definition_attr): - setattr( - schema, - schema_attr, - datetime.combine( - datetime.fromtimestamp(attr, tz=timezone.utc) if isinstance(attr, (float, int)) else attr, - datetime.min.time(), - tzinfo=timezone.utc, - ).timestamp(), - ) - - return schema - - -def create_string_constrained_field_schema( - field_type: type[str] | type[bytes], - kwarg_definition: KwargDefinition, -) -> Schema: - """Create Schema from Constrained Str/Bytes field.""" - schema = Schema(type=OpenAPIType.STRING) - if issubclass(field_type, bytes): - schema.content_encoding = "utf-8" - if kwarg_definition.min_length: - schema.min_length = kwarg_definition.min_length - if kwarg_definition.max_length: - schema.max_length = kwarg_definition.max_length - if kwarg_definition.pattern: - schema.pattern = ( - kwarg_definition.pattern.pattern # type: ignore[attr-defined] - if isinstance(kwarg_definition.pattern, Pattern) # type: ignore[unreachable] - else kwarg_definition.pattern - ) - if kwarg_definition.lower_case: - schema.description = "must be in lower case" - if kwarg_definition.upper_case: - schema.description = "must be in upper case" - return schema diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/examples.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/examples.py deleted file mode 100644 index 49edf72..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/examples.py +++ /dev/null @@ -1,84 +0,0 @@ -from __future__ import annotations - -import typing -from dataclasses import replace -from decimal import Decimal -from enum import Enum -from typing import TYPE_CHECKING, Any - -import msgspec -from polyfactory.exceptions import ParameterException -from polyfactory.factories import DataclassFactory -from polyfactory.field_meta import FieldMeta, Null -from polyfactory.utils.helpers import unwrap_annotation -from polyfactory.utils.predicates import is_union -from typing_extensions import get_args - -from litestar.contrib.pydantic.utils import is_pydantic_model_instance -from litestar.openapi.spec import Example -from litestar.types import Empty - -if TYPE_CHECKING: - from litestar.typing import FieldDefinition - - -class ExampleFactory(DataclassFactory[Example]): - __model__ = Example - __random_seed__ = 10 - - -def _normalize_example_value(value: Any) -> Any: - """Normalize the example value to make it look a bit prettier.""" - # if UnsetType is part of the union, then it might get chosen as the value - # but that will not be properly serialized by msgspec unless it is for a field - # in a msgspec Struct - if is_union(value): - args = list(get_args(value)) - try: - args.remove(msgspec.UnsetType) - value = typing.Union[tuple(args)] # pyright: ignore - except ValueError: - # UnsetType not part of the Union - pass - - value = unwrap_annotation(annotation=value, random=ExampleFactory.__random__) - if isinstance(value, (Decimal, float)): - value = round(float(value), 2) - if isinstance(value, Enum): - value = value.value - if is_pydantic_model_instance(value): - from litestar.contrib.pydantic import _model_dump - - value = _model_dump(value) - if isinstance(value, (list, set)): - value = [_normalize_example_value(v) for v in value] - if isinstance(value, dict): - for k, v in value.items(): - value[k] = _normalize_example_value(v) - return value - - -def _create_field_meta(field: FieldDefinition) -> FieldMeta: - return FieldMeta.from_type( - annotation=field.annotation, - default=field.default if field.default is not Empty else Null, - name=field.name, - random=ExampleFactory.__random__, - ) - - -def create_examples_for_field(field: FieldDefinition) -> list[Example]: - """Create an OpenAPI Example instance. - - Args: - field: A signature field. - - Returns: - A list including a single example. - """ - try: - field_meta = _create_field_meta(replace(field, annotation=_normalize_example_value(field.annotation))) - value = ExampleFactory.get_field_value(field_meta) - return [Example(description=f"Example {field.name} value", value=value)] - except ParameterException: - return [] diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__init__.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__init__.py deleted file mode 100644 index 1b12b1e..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__init__.py +++ /dev/null @@ -1,16 +0,0 @@ -from .dataclass import DataclassSchemaPlugin -from .pagination import PaginationSchemaPlugin -from .struct import StructSchemaPlugin -from .typed_dict import TypedDictSchemaPlugin - -__all__ = ("openapi_schema_plugins",) - -# NOTE: The Pagination type plugin has to come before the Dataclass plugin since the Pagination -# classes are dataclasses, but we want to handle them differently from how dataclasses are normally -# handled. -openapi_schema_plugins = [ - PaginationSchemaPlugin(), - StructSchemaPlugin(), - DataclassSchemaPlugin(), - TypedDictSchemaPlugin(), -] diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/__init__.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/__init__.cpython-311.pyc Binary files differdeleted file mode 100644 index f022bdd..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/__init__.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/dataclass.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/dataclass.cpython-311.pyc Binary files differdeleted file mode 100644 index 7cbfe6f..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/dataclass.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/pagination.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/pagination.cpython-311.pyc Binary files differdeleted file mode 100644 index 0bc4add..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/pagination.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/struct.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/struct.cpython-311.pyc Binary files differdeleted file mode 100644 index cef9d12..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/struct.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/typed_dict.cpython-311.pyc b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/typed_dict.cpython-311.pyc Binary files differdeleted file mode 100644 index 3d534ba..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/__pycache__/typed_dict.cpython-311.pyc +++ /dev/null diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/dataclass.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/dataclass.py deleted file mode 100644 index fb5da35..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/dataclass.py +++ /dev/null @@ -1,36 +0,0 @@ -from __future__ import annotations - -from dataclasses import MISSING, fields -from typing import TYPE_CHECKING - -from litestar.plugins import OpenAPISchemaPlugin -from litestar.typing import FieldDefinition -from litestar.utils.predicates import is_optional_union - -if TYPE_CHECKING: - from litestar._openapi.schema_generation import SchemaCreator - from litestar.openapi.spec import Schema - - -class DataclassSchemaPlugin(OpenAPISchemaPlugin): - def is_plugin_supported_field(self, field_definition: FieldDefinition) -> bool: - return field_definition.is_dataclass_type - - def to_openapi_schema(self, field_definition: FieldDefinition, schema_creator: SchemaCreator) -> Schema: - type_hints = field_definition.get_type_hints(include_extras=True, resolve_generics=True) - dataclass_fields = fields(field_definition.type_) - return schema_creator.create_component_schema( - field_definition, - required=sorted( - field.name - for field in dataclass_fields - if ( - field.default is MISSING - and field.default_factory is MISSING - and not is_optional_union(type_hints[field.name]) - ) - ), - property_fields={ - field.name: FieldDefinition.from_kwarg(type_hints[field.name], field.name) for field in dataclass_fields - }, - ) diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/pagination.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/pagination.py deleted file mode 100644 index 9b4f6c6..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/pagination.py +++ /dev/null @@ -1,60 +0,0 @@ -from __future__ import annotations - -from typing import TYPE_CHECKING - -from litestar.openapi.spec import OpenAPIType, Schema -from litestar.pagination import ClassicPagination, CursorPagination, OffsetPagination -from litestar.plugins import OpenAPISchemaPlugin - -if TYPE_CHECKING: - from litestar._openapi.schema_generation import SchemaCreator - from litestar.typing import FieldDefinition - - -class PaginationSchemaPlugin(OpenAPISchemaPlugin): - def is_plugin_supported_field(self, field_definition: FieldDefinition) -> bool: - return field_definition.origin in (ClassicPagination, CursorPagination, OffsetPagination) - - def to_openapi_schema(self, field_definition: FieldDefinition, schema_creator: SchemaCreator) -> Schema: - if field_definition.origin is ClassicPagination: - return Schema( - type=OpenAPIType.OBJECT, - properties={ - "items": Schema( - type=OpenAPIType.ARRAY, - items=schema_creator.for_field_definition(field_definition.inner_types[0]), - ), - "page_size": Schema(type=OpenAPIType.INTEGER, description="Number of items per page."), - "current_page": Schema(type=OpenAPIType.INTEGER, description="Current page number."), - "total_pages": Schema(type=OpenAPIType.INTEGER, description="Total number of pages."), - }, - ) - - if field_definition.origin is OffsetPagination: - return Schema( - type=OpenAPIType.OBJECT, - properties={ - "items": Schema( - type=OpenAPIType.ARRAY, - items=schema_creator.for_field_definition(field_definition.inner_types[0]), - ), - "limit": Schema(type=OpenAPIType.INTEGER, description="Maximal number of items to send."), - "offset": Schema(type=OpenAPIType.INTEGER, description="Offset from the beginning of the query."), - "total": Schema(type=OpenAPIType.INTEGER, description="Total number of items."), - }, - ) - - cursor_schema = schema_creator.not_generating_examples.for_field_definition(field_definition.inner_types[0]) - cursor_schema.description = "Unique ID, designating the last identifier in the given data set. This value can be used to request the 'next' batch of records." - - return Schema( - type=OpenAPIType.OBJECT, - properties={ - "items": Schema( - type=OpenAPIType.ARRAY, - items=schema_creator.for_field_definition(field_definition=field_definition.inner_types[1]), - ), - "cursor": cursor_schema, - "results_per_page": Schema(type=OpenAPIType.INTEGER, description="Maximal number of items to send."), - }, - ) diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/struct.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/struct.py deleted file mode 100644 index aabfdb3..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/struct.py +++ /dev/null @@ -1,44 +0,0 @@ -from __future__ import annotations - -from typing import TYPE_CHECKING - -from msgspec import Struct -from msgspec.structs import fields - -from litestar.plugins import OpenAPISchemaPlugin -from litestar.types.empty import Empty -from litestar.typing import FieldDefinition -from litestar.utils.predicates import is_optional_union - -if TYPE_CHECKING: - from msgspec.structs import FieldInfo - - from litestar._openapi.schema_generation import SchemaCreator - from litestar.openapi.spec import Schema - - -class StructSchemaPlugin(OpenAPISchemaPlugin): - def is_plugin_supported_field(self, field_definition: FieldDefinition) -> bool: - return not field_definition.is_union and field_definition.is_subclass_of(Struct) - - def to_openapi_schema(self, field_definition: FieldDefinition, schema_creator: SchemaCreator) -> Schema: - def is_field_required(field: FieldInfo) -> bool: - return field.required or field.default_factory is Empty - - type_hints = field_definition.get_type_hints(include_extras=True, resolve_generics=True) - struct_fields = fields(field_definition.type_) - - return schema_creator.create_component_schema( - field_definition, - required=sorted( - [ - field.encode_name - for field in struct_fields - if is_field_required(field=field) and not is_optional_union(type_hints[field.name]) - ] - ), - property_fields={ - field.encode_name: FieldDefinition.from_kwarg(type_hints[field.name], field.encode_name) - for field in struct_fields - }, - ) diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/typed_dict.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/typed_dict.py deleted file mode 100644 index ef34e2b..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/plugins/typed_dict.py +++ /dev/null @@ -1,24 +0,0 @@ -from __future__ import annotations - -from typing import TYPE_CHECKING - -from litestar.plugins import OpenAPISchemaPlugin -from litestar.typing import FieldDefinition - -if TYPE_CHECKING: - from litestar._openapi.schema_generation import SchemaCreator - from litestar.openapi.spec import Schema - - -class TypedDictSchemaPlugin(OpenAPISchemaPlugin): - def is_plugin_supported_field(self, field_definition: FieldDefinition) -> bool: - return field_definition.is_typeddict_type - - def to_openapi_schema(self, field_definition: FieldDefinition, schema_creator: SchemaCreator) -> Schema: - type_hints = field_definition.get_type_hints(include_extras=True, resolve_generics=True) - - return schema_creator.create_component_schema( - field_definition, - required=sorted(getattr(field_definition.type_, "__required_keys__", [])), - property_fields={k: FieldDefinition.from_kwarg(v, k) for k, v in type_hints.items()}, - ) diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/schema.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/schema.py deleted file mode 100644 index 0b7d6c6..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/schema.py +++ /dev/null @@ -1,616 +0,0 @@ -from __future__ import annotations - -from collections import deque -from copy import copy -from datetime import date, datetime, time, timedelta -from decimal import Decimal -from enum import Enum, EnumMeta -from ipaddress import IPv4Address, IPv4Interface, IPv4Network, IPv6Address, IPv6Interface, IPv6Network -from pathlib import Path -from typing import ( - TYPE_CHECKING, - Any, - DefaultDict, - Deque, - Dict, - FrozenSet, - Hashable, - Iterable, - List, - Literal, - Mapping, - MutableMapping, - MutableSequence, - OrderedDict, - Pattern, - Sequence, - Set, - Tuple, - Union, - cast, -) -from uuid import UUID - -from typing_extensions import Self, get_args - -from litestar._openapi.datastructures import SchemaRegistry -from litestar._openapi.schema_generation.constrained_fields import ( - create_date_constrained_field_schema, - create_numerical_constrained_field_schema, - create_string_constrained_field_schema, -) -from litestar._openapi.schema_generation.utils import ( - _get_normalized_schema_key, - _should_create_enum_schema, - _should_create_literal_schema, - _type_or_first_not_none_inner_type, - get_json_schema_formatted_examples, -) -from litestar.datastructures import UploadFile -from litestar.exceptions import ImproperlyConfiguredException -from litestar.openapi.spec.enums import OpenAPIFormat, OpenAPIType -from litestar.openapi.spec.schema import Schema, SchemaDataContainer -from litestar.params import BodyKwarg, ParameterKwarg -from litestar.plugins import OpenAPISchemaPlugin -from litestar.types import Empty -from litestar.types.builtin_types import NoneType -from litestar.typing import FieldDefinition -from litestar.utils.helpers import get_name -from litestar.utils.predicates import ( - is_class_and_subclass, - is_undefined_sentinel, -) -from litestar.utils.typing import ( - get_origin_or_inner_type, - make_non_optional_union, -) - -if TYPE_CHECKING: - from litestar._openapi.datastructures import OpenAPIContext - from litestar.openapi.spec import Example, Reference - from litestar.plugins import OpenAPISchemaPluginProtocol - -KWARG_DEFINITION_ATTRIBUTE_TO_OPENAPI_PROPERTY_MAP: dict[str, str] = { - "content_encoding": "content_encoding", - "default": "default", - "description": "description", - "enum": "enum", - "examples": "examples", - "external_docs": "external_docs", - "format": "format", - "ge": "minimum", - "gt": "exclusive_minimum", - "le": "maximum", - "lt": "exclusive_maximum", - "max_items": "max_items", - "max_length": "max_length", - "min_items": "min_items", - "min_length": "min_length", - "multiple_of": "multiple_of", - "pattern": "pattern", - "title": "title", - "read_only": "read_only", -} - -TYPE_MAP: dict[type[Any] | None | Any, Schema] = { - Decimal: Schema(type=OpenAPIType.NUMBER), - DefaultDict: Schema(type=OpenAPIType.OBJECT), - Deque: Schema(type=OpenAPIType.ARRAY), - Dict: Schema(type=OpenAPIType.OBJECT), - FrozenSet: Schema(type=OpenAPIType.ARRAY), - IPv4Address: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV4), - IPv4Interface: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV4), - IPv4Network: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV4), - IPv6Address: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV6), - IPv6Interface: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV6), - IPv6Network: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.IPV6), - Iterable: Schema(type=OpenAPIType.ARRAY), - List: Schema(type=OpenAPIType.ARRAY), - Mapping: Schema(type=OpenAPIType.OBJECT), - MutableMapping: Schema(type=OpenAPIType.OBJECT), - MutableSequence: Schema(type=OpenAPIType.ARRAY), - None: Schema(type=OpenAPIType.NULL), - NoneType: Schema(type=OpenAPIType.NULL), - OrderedDict: Schema(type=OpenAPIType.OBJECT), - Path: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.URI), - Pattern: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.REGEX), - Sequence: Schema(type=OpenAPIType.ARRAY), - Set: Schema(type=OpenAPIType.ARRAY), - Tuple: Schema(type=OpenAPIType.ARRAY), - UUID: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.UUID), - bool: Schema(type=OpenAPIType.BOOLEAN), - bytearray: Schema(type=OpenAPIType.STRING), - bytes: Schema(type=OpenAPIType.STRING), - date: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.DATE), - datetime: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.DATE_TIME), - deque: Schema(type=OpenAPIType.ARRAY), - dict: Schema(type=OpenAPIType.OBJECT), - float: Schema(type=OpenAPIType.NUMBER), - frozenset: Schema(type=OpenAPIType.ARRAY), - int: Schema(type=OpenAPIType.INTEGER), - list: Schema(type=OpenAPIType.ARRAY), - set: Schema(type=OpenAPIType.ARRAY), - str: Schema(type=OpenAPIType.STRING), - time: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.DURATION), - timedelta: Schema(type=OpenAPIType.STRING, format=OpenAPIFormat.DURATION), - tuple: Schema(type=OpenAPIType.ARRAY), -} - - -def _types_in_list(lst: list[Any]) -> list[OpenAPIType] | OpenAPIType: - """Extract unique OpenAPITypes present in the values of a list. - - Args: - lst: A list of values - - Returns: - OpenAPIType in the given list. If more then one exists, return - a list of OpenAPITypes. - """ - schema_types: list[OpenAPIType] = [] - for item in lst: - schema_type = TYPE_MAP[type(item)].type - if isinstance(schema_type, OpenAPIType): - schema_types.append(schema_type) - else: - raise RuntimeError("Unexpected type for schema item") # pragma: no cover - schema_types = list(set(schema_types)) - return schema_types[0] if len(schema_types) == 1 else schema_types - - -def _get_type_schema_name(field_definition: FieldDefinition) -> str: - """Extract the schema name from a data container. - - Args: - field_definition: A field definition instance. - - Returns: - A string - """ - - if name := getattr(field_definition.annotation, "__schema_name__", None): - return cast("str", name) - - name = get_name(field_definition.annotation) - if field_definition.inner_types: - inner_parts = ", ".join(_get_type_schema_name(t) for t in field_definition.inner_types) - return f"{name}[{inner_parts}]" - - return name - - -def create_enum_schema(annotation: EnumMeta, include_null: bool = False) -> Schema: - """Create a schema instance for an enum. - - Args: - annotation: An enum. - include_null: Whether to include null as a possible value. - - Returns: - A schema instance. - """ - enum_values: list[str | int | None] = [v.value for v in annotation] # type: ignore[var-annotated] - if include_null and None not in enum_values: - enum_values.append(None) - return Schema(type=_types_in_list(enum_values), enum=enum_values) - - -def _iter_flat_literal_args(annotation: Any) -> Iterable[Any]: - """Iterate over the flattened arguments of a Literal. - - Args: - annotation: An Literal annotation. - - Yields: - The flattened arguments of the Literal. - """ - for arg in get_args(annotation): - if get_origin_or_inner_type(arg) is Literal: - yield from _iter_flat_literal_args(arg) - else: - yield arg.value if isinstance(arg, Enum) else arg - - -def create_literal_schema(annotation: Any, include_null: bool = False) -> Schema: - """Create a schema instance for a Literal. - - Args: - annotation: An Literal annotation. - include_null: Whether to include null as a possible value. - - Returns: - A schema instance. - """ - args = list(_iter_flat_literal_args(annotation)) - if include_null and None not in args: - args.append(None) - schema = Schema(type=_types_in_list(args)) - if len(args) > 1: - schema.enum = args - else: - schema.const = args[0] - return schema - - -def create_schema_for_annotation(annotation: Any) -> Schema: - """Get a schema from the type mapping - if possible. - - Args: - annotation: A type annotation. - - Returns: - A schema instance or None. - """ - - return copy(TYPE_MAP[annotation]) if annotation in TYPE_MAP else Schema() - - -class SchemaCreator: - __slots__ = ("generate_examples", "plugins", "prefer_alias", "schema_registry") - - def __init__( - self, - generate_examples: bool = False, - plugins: Iterable[OpenAPISchemaPluginProtocol] | None = None, - prefer_alias: bool = True, - schema_registry: SchemaRegistry | None = None, - ) -> None: - """Instantiate a SchemaCreator. - - Args: - generate_examples: Whether to generate examples if none are given. - plugins: A list of plugins. - prefer_alias: Whether to prefer the alias name for the schema. - schema_registry: A SchemaRegistry instance. - """ - self.generate_examples = generate_examples - self.plugins = plugins if plugins is not None else [] - self.prefer_alias = prefer_alias - self.schema_registry = schema_registry or SchemaRegistry() - - @classmethod - def from_openapi_context(cls, context: OpenAPIContext, prefer_alias: bool = True, **kwargs: Any) -> Self: - kwargs.setdefault("generate_examples", context.openapi_config.create_examples) - kwargs.setdefault("plugins", context.plugins) - kwargs.setdefault("schema_registry", context.schema_registry) - return cls(**kwargs, prefer_alias=prefer_alias) - - @property - def not_generating_examples(self) -> SchemaCreator: - """Return a SchemaCreator with generate_examples set to False.""" - if not self.generate_examples: - return self - return type(self)(generate_examples=False, plugins=self.plugins, prefer_alias=False) - - @staticmethod - def plugin_supports_field(plugin: OpenAPISchemaPluginProtocol, field: FieldDefinition) -> bool: - if predicate := getattr(plugin, "is_plugin_supported_field", None): - return predicate(field) # type: ignore[no-any-return] - return plugin.is_plugin_supported_type(field.annotation) - - def get_plugin_for(self, field_definition: FieldDefinition) -> OpenAPISchemaPluginProtocol | None: - return next( - (plugin for plugin in self.plugins if self.plugin_supports_field(plugin, field_definition)), - None, - ) - - def is_constrained_field(self, field_definition: FieldDefinition) -> bool: - """Return if the field is constrained, taking into account constraints defined by plugins""" - return ( - isinstance(field_definition.kwarg_definition, (ParameterKwarg, BodyKwarg)) - and field_definition.kwarg_definition.is_constrained - ) or any( - p.is_constrained_field(field_definition) - for p in self.plugins - if isinstance(p, OpenAPISchemaPlugin) and p.is_plugin_supported_field(field_definition) - ) - - def is_undefined(self, value: Any) -> bool: - """Return if the field is undefined, taking into account undefined types defined by plugins""" - return is_undefined_sentinel(value) or any( - p.is_undefined_sentinel(value) for p in self.plugins if isinstance(p, OpenAPISchemaPlugin) - ) - - def for_field_definition(self, field_definition: FieldDefinition) -> Schema | Reference: - """Create a Schema for a given FieldDefinition. - - Args: - field_definition: A signature field instance. - - Returns: - A schema instance. - """ - - result: Schema | Reference - - if plugin_for_annotation := self.get_plugin_for(field_definition): - result = self.for_plugin(field_definition, plugin_for_annotation) - elif _should_create_enum_schema(field_definition): - annotation = _type_or_first_not_none_inner_type(field_definition) - result = create_enum_schema(annotation, include_null=field_definition.is_optional) - elif _should_create_literal_schema(field_definition): - annotation = ( - make_non_optional_union(field_definition.annotation) - if field_definition.is_optional - else field_definition.annotation - ) - result = create_literal_schema(annotation, include_null=field_definition.is_optional) - elif field_definition.is_optional: - result = self.for_optional_field(field_definition) - elif field_definition.is_union: - result = self.for_union_field(field_definition) - elif field_definition.is_type_var: - result = self.for_typevar() - elif field_definition.inner_types and not field_definition.is_generic: - result = self.for_object_type(field_definition) - elif self.is_constrained_field(field_definition): - result = self.for_constrained_field(field_definition) - elif field_definition.is_subclass_of(UploadFile): - result = self.for_upload_file(field_definition) - else: - result = create_schema_for_annotation(field_definition.annotation) - - return self.process_schema_result(field_definition, result) if isinstance(result, Schema) else result - - @staticmethod - def for_upload_file(field_definition: FieldDefinition) -> Schema: - """Create schema for UploadFile. - - Args: - field_definition: A field definition instance. - - Returns: - A Schema instance. - """ - - property_key = "file" - schema = Schema( - type=OpenAPIType.STRING, - content_media_type="application/octet-stream", - format=OpenAPIFormat.BINARY, - ) - - # If the type is `dict[str, UploadFile]`, then it's the same as a `list[UploadFile]` - # but we will internally convert that into a `dict[str, UploadFile]`. - if field_definition.is_non_string_sequence or field_definition.is_mapping: - property_key = "files" - schema = Schema(type=OpenAPIType.ARRAY, items=schema) - - # If the uploadfile is annotated directly on the handler, then the - # 'properties' needs to be created. Else, the 'properties' will be - # created by the corresponding plugin. - is_defined_on_handler = field_definition.name == "data" and isinstance( - field_definition.kwarg_definition, BodyKwarg - ) - if is_defined_on_handler: - return Schema(type=OpenAPIType.OBJECT, properties={property_key: schema}) - - return schema - - @staticmethod - def for_typevar() -> Schema: - """Create a schema for a TypeVar. - - Returns: - A schema instance. - """ - - return Schema(type=OpenAPIType.OBJECT) - - def for_optional_field(self, field_definition: FieldDefinition) -> Schema: - """Create a Schema for an optional FieldDefinition. - - Args: - field_definition: A signature field instance. - - Returns: - A schema instance. - """ - schema_or_reference = self.for_field_definition( - FieldDefinition.from_kwarg( - annotation=make_non_optional_union(field_definition.annotation), - name=field_definition.name, - default=field_definition.default, - ) - ) - if isinstance(schema_or_reference, Schema) and isinstance(schema_or_reference.one_of, list): - result = schema_or_reference.one_of - else: - result = [schema_or_reference] - - return Schema(one_of=[Schema(type=OpenAPIType.NULL), *result]) - - def for_union_field(self, field_definition: FieldDefinition) -> Schema: - """Create a Schema for a union FieldDefinition. - - Args: - field_definition: A signature field instance. - - Returns: - A schema instance. - """ - inner_types = (f for f in (field_definition.inner_types or []) if not self.is_undefined(f.annotation)) - values = list(map(self.for_field_definition, inner_types)) - return Schema(one_of=values) - - def for_object_type(self, field_definition: FieldDefinition) -> Schema: - """Create schema for object types (dict, Mapping, list, Sequence etc.) types. - - Args: - field_definition: A signature field instance. - - Returns: - A schema instance. - """ - if field_definition.has_inner_subclass_of(UploadFile): - return self.for_upload_file(field_definition) - - if field_definition.is_mapping: - return Schema( - type=OpenAPIType.OBJECT, - additional_properties=( - self.for_field_definition(field_definition.inner_types[1]) - if field_definition.inner_types and len(field_definition.inner_types) == 2 - else None - ), - ) - - if field_definition.is_non_string_sequence or field_definition.is_non_string_iterable: - # filters out ellipsis from tuple[int, ...] type annotations - inner_types = (f for f in field_definition.inner_types if f.annotation is not Ellipsis) - items = list(map(self.for_field_definition, inner_types or ())) - - return Schema( - type=OpenAPIType.ARRAY, - items=Schema(one_of=items) if len(items) > 1 else items[0], - ) - - raise ImproperlyConfiguredException( # pragma: no cover - f"Parameter '{field_definition.name}' with type '{field_definition.annotation}' could not be mapped to an Open API type. " - f"This can occur if a user-defined generic type is resolved as a parameter. If '{field_definition.name}' should " - "not be documented as a parameter, annotate it using the `Dependency` function, e.g., " - f"`{field_definition.name}: ... = Dependency(...)`." - ) - - def for_plugin(self, field_definition: FieldDefinition, plugin: OpenAPISchemaPluginProtocol) -> Schema | Reference: - """Create a schema using a plugin. - - Args: - field_definition: A signature field instance. - plugin: A plugin for the field type. - - Returns: - A schema instance. - """ - key = _get_normalized_schema_key(field_definition.annotation) - if (ref := self.schema_registry.get_reference_for_key(key)) is not None: - return ref - - schema = plugin.to_openapi_schema(field_definition=field_definition, schema_creator=self) - if isinstance(schema, SchemaDataContainer): # pragma: no cover - return self.for_field_definition( - FieldDefinition.from_kwarg( - annotation=schema.data_container, - name=field_definition.name, - default=field_definition.default, - extra=field_definition.extra, - kwarg_definition=field_definition.kwarg_definition, - ) - ) - return schema - - def for_constrained_field(self, field: FieldDefinition) -> Schema: - """Create Schema for Pydantic Constrained fields (created using constr(), conint() and so forth, or by subclassing - Constrained*) - - Args: - field: A signature field instance. - - Returns: - A schema instance. - """ - kwarg_definition = cast(Union[ParameterKwarg, BodyKwarg], field.kwarg_definition) - if any(is_class_and_subclass(field.annotation, t) for t in (int, float, Decimal)): - return create_numerical_constrained_field_schema(field.annotation, kwarg_definition) - if any(is_class_and_subclass(field.annotation, t) for t in (str, bytes)): # type: ignore[arg-type] - return create_string_constrained_field_schema(field.annotation, kwarg_definition) - if any(is_class_and_subclass(field.annotation, t) for t in (date, datetime)): - return create_date_constrained_field_schema(field.annotation, kwarg_definition) - return self.for_collection_constrained_field(field) - - def for_collection_constrained_field(self, field_definition: FieldDefinition) -> Schema: - """Create Schema from Constrained List/Set field. - - Args: - field_definition: A signature field instance. - - Returns: - A schema instance. - """ - schema = Schema(type=OpenAPIType.ARRAY) - kwarg_definition = cast(Union[ParameterKwarg, BodyKwarg], field_definition.kwarg_definition) - if kwarg_definition.min_items: - schema.min_items = kwarg_definition.min_items - if kwarg_definition.max_items: - schema.max_items = kwarg_definition.max_items - if any(is_class_and_subclass(field_definition.annotation, t) for t in (set, frozenset)): # type: ignore[arg-type] - schema.unique_items = True - - item_creator = self.not_generating_examples - if field_definition.inner_types: - items = list(map(item_creator.for_field_definition, field_definition.inner_types)) - schema.items = Schema(one_of=items) if len(items) > 1 else items[0] - else: - schema.items = item_creator.for_field_definition( - FieldDefinition.from_kwarg( - field_definition.annotation.item_type, f"{field_definition.annotation.__name__}Field" - ) - ) - return schema - - def process_schema_result(self, field: FieldDefinition, schema: Schema) -> Schema | Reference: - if field.kwarg_definition and field.is_const and field.has_default and schema.const is None: - schema.const = field.default - - if field.kwarg_definition: - for kwarg_definition_key, schema_key in KWARG_DEFINITION_ATTRIBUTE_TO_OPENAPI_PROPERTY_MAP.items(): - if (value := getattr(field.kwarg_definition, kwarg_definition_key, Empty)) and ( - not isinstance(value, Hashable) or not self.is_undefined(value) - ): - if schema_key == "examples": - value = get_json_schema_formatted_examples(cast("list[Example]", value)) - - # we only want to transfer values from the `KwargDefinition` to `Schema` if the schema object - # doesn't already have a value for that property. For example, if a field is a constrained date, - # by this point, we have already set the `exclusive_minimum` and/or `exclusive_maximum` fields - # to floating point timestamp values on the schema object. However, the original `date` objects - # that define those constraints on `KwargDefinition` are still `date` objects. We don't want to - # overwrite them here. - if getattr(schema, schema_key, None) is None: - setattr(schema, schema_key, value) - - if not schema.examples and self.generate_examples: - from litestar._openapi.schema_generation.examples import create_examples_for_field - - schema.examples = get_json_schema_formatted_examples(create_examples_for_field(field)) - - if schema.title and schema.type == OpenAPIType.OBJECT: - key = _get_normalized_schema_key(field.annotation) - return self.schema_registry.get_reference_for_key(key) or schema - return schema - - def create_component_schema( - self, - type_: FieldDefinition, - /, - required: list[str], - property_fields: Mapping[str, FieldDefinition], - openapi_type: OpenAPIType = OpenAPIType.OBJECT, - title: str | None = None, - examples: list[Any] | None = None, - ) -> Schema: - """Create a schema for the components/schemas section of the OpenAPI spec. - - These are schemas that can be referenced by other schemas in the document, including self references. - - To support self referencing schemas, the schema is added to the registry before schemas for its properties - are created. This allows the schema to be referenced by its properties. - - Args: - type_: ``FieldDefinition`` instance of the type to create a schema for. - required: A list of required fields. - property_fields: Mapping of name to ``FieldDefinition`` instances for the properties of the schema. - openapi_type: The OpenAPI type, defaults to ``OpenAPIType.OBJECT``. - title: The schema title, generated if not provided. - examples: A mapping of example names to ``Example`` instances, not required. - - Returns: - A schema instance. - """ - schema = self.schema_registry.get_schema_for_key(_get_normalized_schema_key(type_.annotation)) - schema.title = title or _get_type_schema_name(type_) - schema.required = required - schema.type = openapi_type - schema.properties = {k: self.for_field_definition(v) for k, v in property_fields.items()} - schema.examples = examples - return schema diff --git a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/utils.py b/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/utils.py deleted file mode 100644 index 7ce27ca..0000000 --- a/venv/lib/python3.11/site-packages/litestar/_openapi/schema_generation/utils.py +++ /dev/null @@ -1,114 +0,0 @@ -from __future__ import annotations - -from enum import Enum -from typing import TYPE_CHECKING, Any, Mapping, _GenericAlias # type: ignore[attr-defined] - -from litestar.utils.helpers import get_name - -if TYPE_CHECKING: - from collections.abc import Sequence - - from litestar.openapi.spec import Example - from litestar.typing import FieldDefinition - -__all__ = ( - "_type_or_first_not_none_inner_type", - "_should_create_enum_schema", - "_should_create_literal_schema", - "_get_normalized_schema_key", -) - - -def _type_or_first_not_none_inner_type(field_definition: FieldDefinition) -> Any: - """Get the first inner type that is not None. - - This is a narrow focussed utility to be used when we know that a field definition either represents - a single type, or a single type in a union with `None`, and we want the single type. - - Args: - field_definition: A field definition instance. - - Returns: - A field definition instance. - """ - if not field_definition.is_optional: - return field_definition.annotation - inner = next((t for t in field_definition.inner_types if not t.is_none_type), None) - if inner is None: - raise ValueError("Field definition has no inner type that is not None") - return inner.annotation - - -def _should_create_enum_schema(field_definition: FieldDefinition) -> bool: - """Predicate to determine if we should create an enum schema for the field def, or not. - - This returns true if the field definition is an enum, or if the field definition is a union - of an enum and ``None``. - - When an annotation is ``SomeEnum | None`` we should create a schema for the enum that includes ``null`` - in the enum values. - - Args: - field_definition: A field definition instance. - - Returns: - A boolean - """ - return field_definition.is_subclass_of(Enum) or ( - field_definition.is_optional - and len(field_definition.args) == 2 - and field_definition.has_inner_subclass_of(Enum) - ) - - -def _should_create_literal_schema(field_definition: FieldDefinition) -> bool: - """Predicate to determine if we should create a literal schema for the field def, or not. - - This returns ``True`` if the field definition is an literal, or if the field definition is a union - of a literal and None. - - When an annotation is `Literal["anything"] | None` we should create a schema for the literal that includes `null` - in the enum values. - - Args: - field_definition: A field definition instance. - - Returns: - A boolean - """ - return ( - field_definition.is_literal - or field_definition.is_optional - and all(inner.is_literal for inner in field_definition.inner_types if not inner.is_none_type) - ) - - -def _get_normalized_schema_key(annotation: Any) -> tuple[str, ...]: - """Create a key for a type annotation. - - The key should be a tuple such as ``("path", "to", "type", "TypeName")``. - - Args: - annotation: a type annotation - - Returns: - A tuple of strings. - """ - module = getattr(annotation, "__module__", "") - name = str(annotation)[len(module) + 1 :] if isinstance(annotation, _GenericAlias) else annotation.__qualname__ - name = name.replace(".<locals>.", ".") - return *module.split("."), name - - -def get_formatted_examples(field_definition: FieldDefinition, examples: Sequence[Example]) -> Mapping[str, Example]: - """Format the examples into the OpenAPI schema format.""" - - name = field_definition.name or get_name(field_definition.type_) - name = name.lower() - - return {f"{name}-example-{i}": example for i, example in enumerate(examples, 1)} - - -def get_json_schema_formatted_examples(examples: Sequence[Example]) -> list[Any]: - """Format the examples into the JSON schema format.""" - return [example.value for example in examples] |