1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117 |
- """Contains custom types and methods to check types."""
- from __future__ import annotations
- import dataclasses
- import inspect
- import types
- from collections.abc import Callable, Iterable, Mapping, Sequence
- from functools import cached_property, lru_cache, wraps
- from types import GenericAlias
- from typing import ( # noqa: UP035
- TYPE_CHECKING,
- Any,
- Awaitable,
- ClassVar,
- Dict,
- ForwardRef,
- List,
- Literal,
- MutableMapping,
- NoReturn,
- Tuple,
- Union,
- _GenericAlias, # pyright: ignore [reportAttributeAccessIssue]
- _SpecialGenericAlias, # pyright: ignore [reportAttributeAccessIssue]
- get_args,
- is_typeddict,
- )
- from typing import get_origin as get_origin_og
- from typing import get_type_hints as get_type_hints_og
- from pydantic.v1.fields import ModelField
- from typing_extensions import Self as Self
- from typing_extensions import override as override
- import reflex
- from reflex import constants
- from reflex.base import Base
- from reflex.components.core.breakpoints import Breakpoints
- from reflex.utils import console
- # Potential GenericAlias types for isinstance checks.
- GenericAliasTypes = (_GenericAlias, GenericAlias, _SpecialGenericAlias)
- # Potential Union types for isinstance checks.
- UnionTypes = (Union, types.UnionType)
- # Union of generic types.
- GenericType = type | _GenericAlias
- # Valid state var types.
- JSONType = {str, int, float, bool}
- PrimitiveType = int | float | bool | str | list | dict | set | tuple
- PrimitiveTypes = (int, float, bool, str, list, dict, set, tuple)
- StateVar = PrimitiveType | Base | None
- StateIterVar = list | set | tuple
- if TYPE_CHECKING:
- from reflex.vars.base import Var
- ArgsSpec = (
- Callable[[], Sequence[Var]]
- | Callable[[Var], Sequence[Var]]
- | Callable[[Var, Var], Sequence[Var]]
- | Callable[[Var, Var, Var], Sequence[Var]]
- | Callable[[Var, Var, Var, Var], Sequence[Var]]
- | Callable[[Var, Var, Var, Var, Var], Sequence[Var]]
- | Callable[[Var, Var, Var, Var, Var, Var], Sequence[Var]]
- | Callable[[Var, Var, Var, Var, Var, Var, Var], Sequence[Var]]
- )
- else:
- ArgsSpec = Callable[..., list[Any]]
- Scope = MutableMapping[str, Any]
- Message = MutableMapping[str, Any]
- Receive = Callable[[], Awaitable[Message]]
- Send = Callable[[Message], Awaitable[None]]
- ASGIApp = Callable[[Scope, Receive, Send], Awaitable[None]]
- PrimitiveToAnnotation = {
- list: List, # noqa: UP006
- tuple: Tuple, # noqa: UP006
- dict: Dict, # noqa: UP006
- }
- RESERVED_BACKEND_VAR_NAMES = {
- "_abc_impl",
- "_backend_vars",
- "_was_touched",
- }
- class Unset:
- """A class to represent an unset value.
- This is used to differentiate between a value that is not set and a value that is set to None.
- """
- def __repr__(self) -> str:
- """Return the string representation of the class.
- Returns:
- The string representation of the class.
- """
- return "Unset"
- def __bool__(self) -> bool:
- """Return False when the class is used in a boolean context.
- Returns:
- False
- """
- return False
- @lru_cache
- def _get_origin_cached(tp: Any):
- return get_origin_og(tp)
- def get_origin(tp: Any):
- """Get the origin of a class.
- Args:
- tp: The class to get the origin of.
- Returns:
- The origin of the class.
- """
- return (
- origin
- if (origin := getattr(tp, "__origin__", None)) is not None
- else _get_origin_cached(tp)
- )
- @lru_cache
- def is_generic_alias(cls: GenericType) -> bool:
- """Check whether the class is a generic alias.
- Args:
- cls: The class to check.
- Returns:
- Whether the class is a generic alias.
- """
- return isinstance(cls, GenericAliasTypes)
- @lru_cache
- def get_type_hints(obj: Any) -> dict[str, Any]:
- """Get the type hints of a class.
- Args:
- obj: The class to get the type hints of.
- Returns:
- The type hints of the class.
- """
- return get_type_hints_og(obj)
- def _unionize(args: list[GenericType]) -> GenericType:
- if not args:
- return Any # pyright: ignore [reportReturnType]
- if len(args) == 1:
- return args[0]
- return Union[tuple(args)] # noqa: UP007
- def unionize(*args: GenericType) -> type:
- """Unionize the types.
- Args:
- args: The types to unionize.
- Returns:
- The unionized types.
- """
- return _unionize([arg for arg in args if arg is not NoReturn])
- def is_none(cls: GenericType) -> bool:
- """Check if a class is None.
- Args:
- cls: The class to check.
- Returns:
- Whether the class is None.
- """
- return cls is type(None) or cls is None
- def is_union(cls: GenericType) -> bool:
- """Check if a class is a Union.
- Args:
- cls: The class to check.
- Returns:
- Whether the class is a Union.
- """
- origin = getattr(cls, "__origin__", None)
- if origin is Union:
- return True
- return origin is None and isinstance(cls, types.UnionType)
- def is_literal(cls: GenericType) -> bool:
- """Check if a class is a Literal.
- Args:
- cls: The class to check.
- Returns:
- Whether the class is a literal.
- """
- return getattr(cls, "__origin__", None) is Literal
- def has_args(cls: type) -> bool:
- """Check if the class has generic parameters.
- Args:
- cls: The class to check.
- Returns:
- Whether the class has generic
- """
- if get_args(cls):
- return True
- # Check if the class inherits from a generic class (using __orig_bases__)
- if hasattr(cls, "__orig_bases__"):
- for base in cls.__orig_bases__:
- if get_args(base):
- return True
- return False
- def is_optional(cls: GenericType) -> bool:
- """Check if a class is an Optional.
- Args:
- cls: The class to check.
- Returns:
- Whether the class is an Optional.
- """
- return is_union(cls) and type(None) in get_args(cls)
- def is_classvar(a_type: Any) -> bool:
- """Check if a type is a ClassVar.
- Args:
- a_type: The type to check.
- Returns:
- Whether the type is a ClassVar.
- """
- return a_type is ClassVar or (
- type(a_type) is _GenericAlias and a_type.__origin__ is ClassVar
- )
- def true_type_for_pydantic_field(f: ModelField):
- """Get the type for a pydantic field.
- Args:
- f: The field to get the type for.
- Returns:
- The type for the field.
- """
- if not isinstance(f.annotation, (str, ForwardRef)):
- return f.annotation
- type_ = f.outer_type_
- if (
- f.field_info.default is None
- or (isinstance(f.annotation, str) and f.annotation.startswith("Optional"))
- or (
- isinstance(f.annotation, ForwardRef)
- and f.annotation.__forward_arg__.startswith("Optional")
- )
- ) and not is_optional(type_):
- return type_ | None
- return type_
- def value_inside_optional(cls: GenericType) -> GenericType:
- """Get the value inside an Optional type or the original type.
- Args:
- cls: The class to check.
- Returns:
- The value inside the Optional type or the original type.
- """
- if is_union(cls) and len(args := get_args(cls)) >= 2 and type(None) in args:
- if len(args) == 2:
- return args[0] if args[1] is type(None) else args[1]
- return unionize(*[arg for arg in args if arg is not type(None)])
- return cls
- def get_field_type(cls: GenericType, field_name: str) -> GenericType | None:
- """Get the type of a field in a class.
- Args:
- cls: The class to check.
- field_name: The name of the field to check.
- Returns:
- The type of the field, if it exists, else None.
- """
- if (
- hasattr(cls, "__fields__")
- and field_name in cls.__fields__
- and hasattr(cls.__fields__[field_name], "annotation")
- and not isinstance(cls.__fields__[field_name].annotation, (str, ForwardRef))
- ):
- return cls.__fields__[field_name].annotation
- type_hints = get_type_hints(cls)
- return type_hints.get(field_name, None)
- def get_property_hint(attr: Any | None) -> GenericType | None:
- """Check if an attribute is a property and return its type hint.
- Args:
- attr: The descriptor to check.
- Returns:
- The type hint of the property, if it is a property, else None.
- """
- from sqlalchemy.ext.hybrid import hybrid_property
- if not isinstance(attr, (property, hybrid_property)):
- return None
- hints = get_type_hints(attr.fget)
- return hints.get("return", None)
- def get_attribute_access_type(cls: GenericType, name: str) -> GenericType | None:
- """Check if an attribute can be accessed on the cls and return its type.
- Supports pydantic models, unions, and annotated attributes on rx.Model.
- Args:
- cls: The class to check.
- name: The name of the attribute to check.
- Returns:
- The type of the attribute, if accessible, or None
- """
- import sqlalchemy
- from sqlalchemy.ext.associationproxy import AssociationProxyInstance
- from sqlalchemy.orm import DeclarativeBase, Mapped, QueryableAttribute, Relationship
- from reflex.model import Model
- try:
- attr = getattr(cls, name, None)
- except NotImplementedError:
- attr = None
- if hint := get_property_hint(attr):
- return hint
- if hasattr(cls, "__fields__") and name in cls.__fields__:
- # pydantic models
- return get_field_type(cls, name)
- elif isinstance(cls, type) and issubclass(cls, DeclarativeBase):
- insp = sqlalchemy.inspect(cls)
- if name in insp.columns:
- # check for list types
- column = insp.columns[name]
- column_type = column.type
- try:
- type_ = insp.columns[name].type.python_type
- except NotImplementedError:
- type_ = None
- if type_ is not None:
- if hasattr(column_type, "item_type"):
- try:
- item_type = column_type.item_type.python_type # pyright: ignore [reportAttributeAccessIssue]
- except NotImplementedError:
- item_type = None
- if item_type is not None:
- if type_ in PrimitiveToAnnotation:
- type_ = PrimitiveToAnnotation[type_]
- type_ = type_[item_type] # pyright: ignore [reportIndexIssue]
- if column.nullable:
- type_ = type_ | None
- return type_
- if name in insp.all_orm_descriptors:
- descriptor = insp.all_orm_descriptors[name]
- if hint := get_property_hint(descriptor):
- return hint
- if isinstance(descriptor, QueryableAttribute):
- prop = descriptor.property
- if isinstance(prop, Relationship):
- type_ = prop.mapper.class_
- # TODO: check for nullable?
- type_ = list[type_] if prop.uselist else type_ | None
- return type_
- if isinstance(attr, AssociationProxyInstance):
- return list[
- get_attribute_access_type(
- attr.target_class,
- attr.remote_attr.key, # type: ignore[attr-defined]
- )
- ]
- elif isinstance(cls, type) and not is_generic_alias(cls) and issubclass(cls, Model):
- # Check in the annotations directly (for sqlmodel.Relationship)
- hints = get_type_hints(cls)
- if name in hints:
- type_ = hints[name]
- type_origin = get_origin(type_)
- if isinstance(type_origin, type) and issubclass(type_origin, Mapped):
- return get_args(type_)[0] # SQLAlchemy v2
- if isinstance(type_, ModelField):
- return type_.type_ # SQLAlchemy v1.4
- return type_
- elif is_union(cls):
- # Check in each arg of the annotation.
- return unionize(
- *(get_attribute_access_type(arg, name) for arg in get_args(cls))
- )
- elif isinstance(cls, type):
- # Bare class
- exceptions = NameError
- try:
- hints = get_type_hints(cls)
- if name in hints:
- return hints[name]
- except exceptions as e:
- console.warn(f"Failed to resolve ForwardRefs for {cls}.{name} due to {e}")
- pass
- return None # Attribute is not accessible.
- @lru_cache
- def get_base_class(cls: GenericType) -> type:
- """Get the base class of a class.
- Args:
- cls: The class.
- Returns:
- The base class of the class.
- Raises:
- TypeError: If a literal has multiple types.
- """
- if is_literal(cls):
- # only literals of the same type are supported.
- arg_type = type(get_args(cls)[0])
- if not all(type(arg) is arg_type for arg in get_args(cls)):
- raise TypeError("only literals of the same type are supported")
- return type(get_args(cls)[0])
- if is_union(cls):
- return tuple(get_base_class(arg) for arg in get_args(cls)) # pyright: ignore [reportReturnType]
- return get_base_class(cls.__origin__) if is_generic_alias(cls) else cls
- def _breakpoints_satisfies_typing(cls_check: GenericType, instance: Any) -> bool:
- """Check if the breakpoints instance satisfies the typing.
- Args:
- cls_check: The class to check against.
- instance: The instance to check.
- Returns:
- Whether the breakpoints instance satisfies the typing.
- """
- cls_check_base = get_base_class(cls_check)
- if cls_check_base == Breakpoints:
- _, expected_type = get_args(cls_check)
- if is_literal(expected_type):
- for value in instance.values():
- if not isinstance(value, str) or value not in get_args(expected_type):
- return False
- return True
- elif isinstance(cls_check_base, tuple):
- # union type, so check all types
- return any(
- _breakpoints_satisfies_typing(type_to_check, instance)
- for type_to_check in get_args(cls_check)
- )
- elif cls_check_base == reflex.vars.Var and "__args__" in cls_check.__dict__:
- return _breakpoints_satisfies_typing(get_args(cls_check)[0], instance)
- return False
- def _issubclass(cls: GenericType, cls_check: GenericType, instance: Any = None) -> bool:
- """Check if a class is a subclass of another class.
- Args:
- cls: The class to check.
- cls_check: The class to check against.
- instance: An instance of cls to aid in checking generics.
- Returns:
- Whether the class is a subclass of the other class.
- Raises:
- TypeError: If the base class is not valid for issubclass.
- """
- # Special check for Any.
- if cls_check == Any:
- return True
- if cls in [Any, Callable, None]:
- return False
- # Get the base classes.
- cls_base = get_base_class(cls)
- cls_check_base = get_base_class(cls_check)
- # The class we're checking should not be a union.
- if isinstance(cls_base, tuple):
- return False
- # Check that fields of breakpoints match the expected values.
- if isinstance(instance, Breakpoints):
- return _breakpoints_satisfies_typing(cls_check, instance)
- if isinstance(cls_check_base, tuple):
- cls_check_base = tuple(
- cls_check_one if not is_typeddict(cls_check_one) else dict
- for cls_check_one in cls_check_base
- )
- if is_typeddict(cls_check_base):
- cls_check_base = dict
- # Check if the types match.
- try:
- return cls_check_base == Any or issubclass(cls_base, cls_check_base)
- except TypeError as te:
- # These errors typically arise from bad annotations and are hard to
- # debug without knowing the type that we tried to compare.
- raise TypeError(f"Invalid type for issubclass: {cls_base}") from te
- def does_obj_satisfy_typed_dict(obj: Any, cls: GenericType) -> bool:
- """Check if an object satisfies a typed dict.
- Args:
- obj: The object to check.
- cls: The typed dict to check against.
- Returns:
- Whether the object satisfies the typed dict.
- """
- if not isinstance(obj, Mapping):
- return False
- key_names_to_values = get_type_hints(cls)
- required_keys: frozenset[str] = getattr(cls, "__required_keys__", frozenset())
- if not all(
- isinstance(key, str)
- and key in key_names_to_values
- and _isinstance(value, key_names_to_values[key])
- for key, value in obj.items()
- ):
- return False
- # TODO in 3.14: Implement https://peps.python.org/pep-0728/ if it's approved
- # required keys are all present
- return required_keys.issubset(required_keys)
- def _isinstance(
- obj: Any,
- cls: GenericType,
- *,
- nested: int = 0,
- treat_var_as_type: bool = True,
- treat_mutable_obj_as_immutable: bool = False,
- ) -> bool:
- """Check if an object is an instance of a class.
- Args:
- obj: The object to check.
- cls: The class to check against.
- nested: How many levels deep to check.
- treat_var_as_type: Whether to treat Var as the type it represents, i.e. _var_type.
- treat_mutable_obj_as_immutable: Whether to treat mutable objects as immutable. Useful if a component declares a mutable object as a prop, but the value is not expected to change.
- Returns:
- Whether the object is an instance of the class.
- """
- if cls is Any:
- return True
- from reflex.vars import LiteralVar, Var
- if cls is Var:
- return isinstance(obj, Var)
- if isinstance(obj, LiteralVar):
- return treat_var_as_type and _isinstance(
- obj._var_value, cls, nested=nested, treat_var_as_type=True
- )
- if isinstance(obj, Var):
- return treat_var_as_type and typehint_issubclass(
- obj._var_type,
- cls,
- treat_mutable_superclasss_as_immutable=treat_mutable_obj_as_immutable,
- treat_literals_as_union_of_types=True,
- treat_any_as_subtype_of_everything=True,
- )
- if cls is None or cls is type(None):
- return obj is None
- if cls is not None and is_union(cls):
- return any(
- _isinstance(obj, arg, nested=nested, treat_var_as_type=treat_var_as_type)
- for arg in get_args(cls)
- )
- if is_literal(cls):
- return obj in get_args(cls)
- origin = get_origin(cls)
- if origin is None:
- # cls is a typed dict
- if is_typeddict(cls):
- if nested:
- return does_obj_satisfy_typed_dict(obj, cls)
- return isinstance(obj, dict)
- # cls is a float
- if cls is float:
- return isinstance(obj, (float, int))
- # cls is a simple class
- return isinstance(obj, cls)
- args = get_args(cls)
- if not args:
- if treat_mutable_obj_as_immutable:
- if origin is dict:
- origin = Mapping
- elif origin is list or origin is set:
- origin = Sequence
- # cls is a simple generic class
- return isinstance(obj, origin)
- if origin is Var and args:
- # cls is a Var
- return _isinstance(
- obj,
- args[0],
- nested=nested,
- treat_var_as_type=treat_var_as_type,
- treat_mutable_obj_as_immutable=treat_mutable_obj_as_immutable,
- )
- if nested > 0 and args:
- if origin is list:
- expected_class = Sequence if treat_mutable_obj_as_immutable else list
- return isinstance(obj, expected_class) and all(
- _isinstance(
- item,
- args[0],
- nested=nested - 1,
- treat_var_as_type=treat_var_as_type,
- )
- for item in obj
- )
- if origin is tuple:
- if args[-1] is Ellipsis:
- return isinstance(obj, tuple) and all(
- _isinstance(
- item,
- args[0],
- nested=nested - 1,
- treat_var_as_type=treat_var_as_type,
- )
- for item in obj
- )
- return (
- isinstance(obj, tuple)
- and len(obj) == len(args)
- and all(
- _isinstance(
- item,
- arg,
- nested=nested - 1,
- treat_var_as_type=treat_var_as_type,
- )
- for item, arg in zip(obj, args, strict=True)
- )
- )
- if origin in (dict, Mapping, Breakpoints):
- expected_class = (
- dict
- if origin is dict and not treat_mutable_obj_as_immutable
- else Mapping
- )
- return isinstance(obj, expected_class) and all(
- _isinstance(
- key, args[0], nested=nested - 1, treat_var_as_type=treat_var_as_type
- )
- and _isinstance(
- value,
- args[1],
- nested=nested - 1,
- treat_var_as_type=treat_var_as_type,
- )
- for key, value in obj.items()
- )
- if origin is set:
- expected_class = Sequence if treat_mutable_obj_as_immutable else set
- return isinstance(obj, expected_class) and all(
- _isinstance(
- item,
- args[0],
- nested=nested - 1,
- treat_var_as_type=treat_var_as_type,
- )
- for item in obj
- )
- if args:
- from reflex.vars import Field
- if origin is Field:
- return _isinstance(
- obj, args[0], nested=nested, treat_var_as_type=treat_var_as_type
- )
- return isinstance(obj, get_base_class(cls))
- def is_dataframe(value: type) -> bool:
- """Check if the given value is a dataframe.
- Args:
- value: The value to check.
- Returns:
- Whether the value is a dataframe.
- """
- if is_generic_alias(value) or value == Any:
- return False
- return value.__name__ == "DataFrame"
- def is_valid_var_type(type_: type) -> bool:
- """Check if the given type is a valid prop type.
- Args:
- type_: The type to check.
- Returns:
- Whether the type is a valid prop type.
- """
- from reflex.utils import serializers
- if is_union(type_):
- return all(is_valid_var_type(arg) for arg in get_args(type_))
- return (
- _issubclass(type_, StateVar)
- or serializers.has_serializer(type_)
- or dataclasses.is_dataclass(type_)
- )
- def is_backend_base_variable(name: str, cls: type) -> bool:
- """Check if this variable name correspond to a backend variable.
- Args:
- name: The name of the variable to check
- cls: The class of the variable to check
- Returns:
- bool: The result of the check
- """
- if name in RESERVED_BACKEND_VAR_NAMES:
- return False
- if not name.startswith("_"):
- return False
- if name.startswith("__"):
- return False
- if name.startswith(f"_{cls.__name__}__"):
- return False
- # Extract the namespace of the original module if defined (dynamic substates).
- if callable(getattr(cls, "_get_type_hints", None)):
- hints = cls._get_type_hints()
- else:
- hints = get_type_hints(cls)
- if name in hints:
- hint = get_origin(hints[name])
- if hint == ClassVar:
- return False
- if name in cls.inherited_backend_vars:
- return False
- from reflex.vars.base import is_computed_var
- if name in cls.__dict__:
- value = cls.__dict__[name]
- if type(value) is classmethod:
- return False
- if callable(value):
- return False
- if isinstance(
- value,
- (
- types.FunctionType,
- property,
- cached_property,
- ),
- ) or is_computed_var(value):
- return False
- return True
- def check_type_in_allowed_types(value_type: type, allowed_types: Iterable) -> bool:
- """Check that a value type is found in a list of allowed types.
- Args:
- value_type: Type of value.
- allowed_types: Iterable of allowed types.
- Returns:
- If the type is found in the allowed types.
- """
- return get_base_class(value_type) in allowed_types
- def check_prop_in_allowed_types(prop: Any, allowed_types: Iterable) -> bool:
- """Check that a prop value is in a list of allowed types.
- Does the check in a way that works regardless if it's a raw value or a state Var.
- Args:
- prop: The prop to check.
- allowed_types: The list of allowed types.
- Returns:
- If the prop type match one of the allowed_types.
- """
- from reflex.vars import Var
- type_ = prop._var_type if isinstance(prop, Var) else type(prop)
- return type_ in allowed_types
- def is_encoded_fstring(value: Any) -> bool:
- """Check if a value is an encoded Var f-string.
- Args:
- value: The value string to check.
- Returns:
- Whether the value is an f-string
- """
- return isinstance(value, str) and constants.REFLEX_VAR_OPENING_TAG in value
- def validate_literal(key: str, value: Any, expected_type: type, comp_name: str):
- """Check that a value is a valid literal.
- Args:
- key: The prop name.
- value: The prop value to validate.
- expected_type: The expected type(literal type).
- comp_name: Name of the component.
- Raises:
- ValueError: When the value is not a valid literal.
- """
- from reflex.vars import Var
- if (
- is_literal(expected_type)
- and not isinstance(value, Var) # validating vars is not supported yet.
- and not is_encoded_fstring(value) # f-strings are not supported.
- and value not in expected_type.__args__
- ):
- allowed_values = expected_type.__args__
- if value not in allowed_values:
- allowed_value_str = ",".join(
- [str(v) if not isinstance(v, str) else f"'{v}'" for v in allowed_values]
- )
- value_str = f"'{value}'" if isinstance(value, str) else value
- raise ValueError(
- f"prop value for {key!s} of the `{comp_name}` component should be one of the following: {allowed_value_str}. Got {value_str} instead"
- )
- def validate_parameter_literals(func: Callable):
- """Decorator to check that the arguments passed to a function
- correspond to the correct function parameter if it (the parameter)
- is a literal type.
- Args:
- func: The function to validate.
- Returns:
- The wrapper function.
- """
- console.deprecate(
- "validate_parameter_literals",
- reason="Use manual validation instead.",
- deprecation_version="0.7.11",
- removal_version="0.8.0",
- dedupe=True,
- )
- func_params = list(inspect.signature(func).parameters.items())
- annotations = {param[0]: param[1].annotation for param in func_params}
- @wraps(func)
- def wrapper(*args, **kwargs):
- # validate args
- for param, arg in zip(annotations, args, strict=False):
- if annotations[param] is inspect.Parameter.empty:
- continue
- validate_literal(param, arg, annotations[param], func.__name__)
- # validate kwargs.
- for key, value in kwargs.items():
- annotation = annotations.get(key)
- if not annotation or annotation is inspect.Parameter.empty:
- continue
- validate_literal(key, value, annotation, func.__name__)
- return func(*args, **kwargs)
- return wrapper
- # Store this here for performance.
- StateBases = get_base_class(StateVar)
- StateIterBases = get_base_class(StateIterVar)
- def safe_issubclass(cls: Any, cls_check: Any | tuple[Any, ...]):
- """Check if a class is a subclass of another class. Returns False if internal error occurs.
- Args:
- cls: The class to check.
- cls_check: The class to check against.
- Returns:
- Whether the class is a subclass of the other class.
- """
- try:
- return issubclass(cls, cls_check)
- except TypeError:
- return False
- def typehint_issubclass(
- possible_subclass: Any,
- possible_superclass: Any,
- *,
- treat_mutable_superclasss_as_immutable: bool = False,
- treat_literals_as_union_of_types: bool = True,
- treat_any_as_subtype_of_everything: bool = False,
- ) -> bool:
- """Check if a type hint is a subclass of another type hint.
- Args:
- possible_subclass: The type hint to check.
- possible_superclass: The type hint to check against.
- treat_mutable_superclasss_as_immutable: Whether to treat target classes as immutable.
- treat_literals_as_union_of_types: Whether to treat literals as a union of their types.
- treat_any_as_subtype_of_everything: Whether to treat Any as a subtype of everything. This is the default behavior in Python.
- Returns:
- Whether the type hint is a subclass of the other type hint.
- """
- if possible_superclass is Any:
- return True
- if possible_subclass is Any:
- return treat_any_as_subtype_of_everything
- if possible_subclass is NoReturn:
- return True
- provided_type_origin = get_origin(possible_subclass)
- accepted_type_origin = get_origin(possible_superclass)
- if provided_type_origin is None and accepted_type_origin is None:
- # In this case, we are dealing with a non-generic type, so we can use issubclass
- return issubclass(possible_subclass, possible_superclass)
- if treat_literals_as_union_of_types and is_literal(possible_superclass):
- args = get_args(possible_superclass)
- return any(
- typehint_issubclass(
- possible_subclass,
- type(arg),
- treat_mutable_superclasss_as_immutable=treat_mutable_superclasss_as_immutable,
- treat_literals_as_union_of_types=treat_literals_as_union_of_types,
- treat_any_as_subtype_of_everything=treat_any_as_subtype_of_everything,
- )
- for arg in args
- )
- if is_literal(possible_subclass):
- args = get_args(possible_subclass)
- return all(
- _isinstance(
- arg,
- possible_superclass,
- treat_mutable_obj_as_immutable=treat_mutable_superclasss_as_immutable,
- nested=2,
- )
- for arg in args
- )
- provided_type_origin = (
- Union if provided_type_origin is types.UnionType else provided_type_origin
- )
- accepted_type_origin = (
- Union if accepted_type_origin is types.UnionType else accepted_type_origin
- )
- # Get type arguments (e.g., [float, int] for dict[float, int])
- provided_args = get_args(possible_subclass)
- accepted_args = get_args(possible_superclass)
- if accepted_type_origin is Union:
- if provided_type_origin is not Union:
- return any(
- typehint_issubclass(
- possible_subclass,
- accepted_arg,
- treat_mutable_superclasss_as_immutable=treat_mutable_superclasss_as_immutable,
- treat_literals_as_union_of_types=treat_literals_as_union_of_types,
- treat_any_as_subtype_of_everything=treat_any_as_subtype_of_everything,
- )
- for accepted_arg in accepted_args
- )
- return all(
- any(
- typehint_issubclass(
- provided_arg,
- accepted_arg,
- treat_mutable_superclasss_as_immutable=treat_mutable_superclasss_as_immutable,
- treat_literals_as_union_of_types=treat_literals_as_union_of_types,
- treat_any_as_subtype_of_everything=treat_any_as_subtype_of_everything,
- )
- for accepted_arg in accepted_args
- )
- for provided_arg in provided_args
- )
- if provided_type_origin is Union:
- return all(
- typehint_issubclass(
- provided_arg,
- possible_superclass,
- treat_mutable_superclasss_as_immutable=treat_mutable_superclasss_as_immutable,
- treat_literals_as_union_of_types=treat_literals_as_union_of_types,
- treat_any_as_subtype_of_everything=treat_any_as_subtype_of_everything,
- )
- for provided_arg in provided_args
- )
- provided_type_origin = provided_type_origin or possible_subclass
- accepted_type_origin = accepted_type_origin or possible_superclass
- if treat_mutable_superclasss_as_immutable:
- if accepted_type_origin is dict:
- accepted_type_origin = Mapping
- elif accepted_type_origin is list or accepted_type_origin is set:
- accepted_type_origin = Sequence
- # Check if the origin of both types is the same (e.g., list for list[int])
- if not safe_issubclass(
- provided_type_origin or possible_subclass,
- accepted_type_origin or possible_superclass,
- ):
- return False
- # Ensure all specific types are compatible with accepted types
- # Note this is not necessarily correct, as it doesn't check against contravariance and covariance
- # It also ignores when the length of the arguments is different
- return all(
- typehint_issubclass(
- provided_arg,
- accepted_arg,
- treat_mutable_superclasss_as_immutable=treat_mutable_superclasss_as_immutable,
- treat_literals_as_union_of_types=treat_literals_as_union_of_types,
- treat_any_as_subtype_of_everything=treat_any_as_subtype_of_everything,
- )
- for provided_arg, accepted_arg in zip(
- provided_args, accepted_args, strict=False
- )
- if accepted_arg is not Any
- )
|