?HAMA

Navigation

Breadcrumbs

/ / opt / cloudlinux / venv / lib64 / python3.11 / site-packages / pydantic / _internal

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 4.74 KB

/opt/cloudlinux/venv/lib64/python3.11/site-packages/pydantic/_internal/_schema_generation_shared.py

Text Base64

"""Types and utility functions used by various other internal tools."""
from __future__ import annotations

from typing import TYPE_CHECKING, Any, Callable

from pydantic_core import core_schema
from typing_extensions import Literal

from ..annotated_handlers import GetCoreSchemaHandler, GetJsonSchemaHandler

if TYPE_CHECKING:
    from ..json_schema import GenerateJsonSchema, JsonSchemaValue
    from ._core_utils import CoreSchemaOrField
    from ._generate_schema import GenerateSchema

GetJsonSchemaFunction = Callable[[CoreSchemaOrField, GetJsonSchemaHandler], JsonSchemaValue]
    HandlerOverride = Callable[[CoreSchemaOrField], JsonSchemaValue]

class GenerateJsonSchemaHandler(GetJsonSchemaHandler):
    """JsonSchemaHandler implementation that doesn't do ref unwrapping by default.

This is used for any Annotated metadata so that we don't end up with conflicting
    modifications to the definition schema.

Used internally by Pydantic, please do not rely on this implementation.
    See `GetJsonSchemaHandler` for the handler API.
    """

def __init__(self, generate_json_schema: GenerateJsonSchema, handler_override: HandlerOverride | None) -> None:
        self.generate_json_schema = generate_json_schema
        self.handler = handler_override or generate_json_schema.generate_inner
        self.mode = generate_json_schema.mode

def __call__(self, __core_schema: CoreSchemaOrField) -> JsonSchemaValue:
        return self.handler(__core_schema)

def resolve_ref_schema(self, maybe_ref_json_schema: JsonSchemaValue) -> JsonSchemaValue:
        """Resolves `$ref` in the json schema.

This returns the input json schema if there is no `$ref` in json schema.

Args:
            maybe_ref_json_schema: The input json schema that may contains `$ref`.

Returns:
            Resolved json schema.

Raises:
            LookupError: If it can't find the definition for `$ref`.
        """
        if '$ref' not in maybe_ref_json_schema:
            return maybe_ref_json_schema
        ref = maybe_ref_json_schema['$ref']
        json_schema = self.generate_json_schema.get_schema_from_definitions(ref)
        if json_schema is None:
            raise LookupError(
                f'Could not find a ref for {ref}.'
                ' Maybe you tried to call resolve_ref_schema from within a recursive model?'
            )
        return json_schema

class CallbackGetCoreSchemaHandler(GetCoreSchemaHandler):
    """Wrapper to use an arbitrary function as a `GetCoreSchemaHandler`.

Used internally by Pydantic, please do not rely on this implementation.
    See `GetCoreSchemaHandler` for the handler API.
    """

def __init__(
        self,
        handler: Callable[[Any], core_schema.CoreSchema],
        generate_schema: GenerateSchema,
        ref_mode: Literal['to-def', 'unpack'] = 'to-def',
    ) -> None:
        self._handler = handler
        self._generate_schema = generate_schema
        self._ref_mode = ref_mode

def __call__(self, __source_type: Any) -> core_schema.CoreSchema:
        schema = self._handler(__source_type)
        ref = schema.get('ref')
        if self._ref_mode == 'to-def':
            if ref is not None:
                self._generate_schema.defs.definitions[ref] = schema
                return core_schema.definition_reference_schema(ref)
            return schema
        else:  # ref_mode = 'unpack
            return self.resolve_ref_schema(schema)

def _get_types_namespace(self) -> dict[str, Any] | None:
        return self._generate_schema._types_namespace

def generate_schema(self, __source_type: Any) -> core_schema.CoreSchema:
        return self._generate_schema.generate_schema(__source_type)

@property
    def field_name(self) -> str | None:
        return self._generate_schema.field_name_stack.get()

def resolve_ref_schema(self, maybe_ref_schema: core_schema.CoreSchema) -> core_schema.CoreSchema:
        """Resolves reference in the core schema.

Args:
            maybe_ref_schema: The input core schema that may contains reference.

Returns:
            Resolved core schema.

Raises:
            LookupError: If it can't find the definition for reference.
        """
        if maybe_ref_schema['type'] == 'definition-ref':
            ref = maybe_ref_schema['schema_ref']
            if ref not in self._generate_schema.defs.definitions:
                raise LookupError(
                    f'Could not find a ref for {ref}.'
                    ' Maybe you tried to call resolve_ref_schema from within a recursive model?'
                )
            return self._generate_schema.defs.definitions[ref]
        elif maybe_ref_schema['type'] == 'definitions':
            return self.resolve_ref_schema(maybe_ref_schema['schema'])
        return maybe_ref_schema

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 1 × Files: 23

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2024-03-21 11:39:42	Edit Download recursive recursive
_config.py	11.04 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_core_metadata.py	3.44 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_core_utils.py	24.24 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_dataclasses.py	10.46 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_decorators.py	30.05 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_decorators_v1.py	6.12 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_discriminated_union.py	25.61 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_fields.py	11.62 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_forward_ref.py	425 B	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_generate_schema.py	87.89 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_generics.py	21.65 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_internal_dataclass.py	207 B	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_known_annotated_metadata.py	15.89 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_mock_val_ser.py	4.20 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_model_construction.py	26.41 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_repr.py	4.27 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_schema_generation_shared.py	4.74 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_std_types_schema.py	28.27 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_typing_extra.py	16.15 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_utils.py	11.42 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_validate_call.py	5.36 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
_validators.py	9.81 KB	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2024-03-21 11:39:42	Edit Download recursive recursive