12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488 |
- """Collection of base classes."""
- from __future__ import annotations
- import contextlib
- import copy
- import dataclasses
- import datetime
- import functools
- import inspect
- import json
- import random
- import re
- import string
- import uuid
- import warnings
- from collections.abc import Callable, Coroutine, Iterable, Mapping, Sequence
- from decimal import Decimal
- from types import CodeType, FunctionType
- from typing import ( # noqa: UP035
- TYPE_CHECKING,
- Any,
- ClassVar,
- Dict,
- FrozenSet,
- Generic,
- List,
- Literal,
- NoReturn,
- ParamSpec,
- Protocol,
- Set,
- Tuple,
- TypeGuard,
- TypeVar,
- cast,
- get_args,
- get_type_hints,
- overload,
- )
- from rich.markup import escape
- from typing_extensions import deprecated, override
- from reflex import constants
- from reflex.base import Base
- from reflex.constants.compiler import Hooks
- from reflex.utils import console, exceptions, imports, serializers, types
- from reflex.utils.exceptions import (
- ComputedVarSignatureError,
- UntypedComputedVarError,
- VarAttributeError,
- VarDependencyError,
- VarTypeError,
- )
- from reflex.utils.format import format_state_name
- from reflex.utils.imports import (
- ImmutableImportDict,
- ImmutableParsedImportDict,
- ImportDict,
- ImportVar,
- ParsedImportTuple,
- parse_imports,
- )
- from reflex.utils.types import (
- GenericType,
- Self,
- _isinstance,
- get_origin,
- has_args,
- safe_issubclass,
- unionize,
- )
- if TYPE_CHECKING:
- from reflex.components.component import BaseComponent
- from reflex.state import BaseState
- from .number import BooleanVar, LiteralBooleanVar, LiteralNumberVar, NumberVar
- from .object import LiteralObjectVar, ObjectVar
- from .sequence import ArrayVar, LiteralArrayVar, LiteralStringVar, StringVar
- VAR_TYPE = TypeVar("VAR_TYPE", covariant=True)
- OTHER_VAR_TYPE = TypeVar("OTHER_VAR_TYPE")
- STRING_T = TypeVar("STRING_T", bound=str)
- SEQUENCE_TYPE = TypeVar("SEQUENCE_TYPE", bound=Sequence)
- warnings.filterwarnings("ignore", message="fields may not start with an underscore")
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- )
- class VarSubclassEntry:
- """Entry for a Var subclass."""
- var_subclass: type[Var]
- to_var_subclass: type[ToOperation]
- python_types: tuple[GenericType, ...]
- _var_subclasses: list[VarSubclassEntry] = []
- _var_literal_subclasses: list[tuple[type[LiteralVar], VarSubclassEntry]] = []
- @dataclasses.dataclass(
- eq=True,
- frozen=True,
- )
- class VarData:
- """Metadata associated with a x."""
- # The name of the enclosing state.
- state: str = dataclasses.field(default="")
- # The name of the field in the state.
- field_name: str = dataclasses.field(default="")
- # Imports needed to render this var
- imports: ParsedImportTuple = dataclasses.field(default_factory=tuple)
- # Hooks that need to be present in the component to render this var
- hooks: tuple[str, ...] = dataclasses.field(default_factory=tuple)
- # Dependencies of the var
- deps: tuple[Var, ...] = dataclasses.field(default_factory=tuple)
- # Position of the hook in the component
- position: Hooks.HookPosition | None = None
- # Components that are part of this var
- components: tuple[BaseComponent, ...] = dataclasses.field(default_factory=tuple)
- def __init__(
- self,
- state: str = "",
- field_name: str = "",
- imports: ImmutableImportDict | ImmutableParsedImportDict | None = None,
- hooks: Mapping[str, VarData | None] | Sequence[str] | str | None = None,
- deps: list[Var] | None = None,
- position: Hooks.HookPosition | None = None,
- components: Iterable[BaseComponent] | None = None,
- ):
- """Initialize the var data.
- Args:
- state: The name of the enclosing state.
- field_name: The name of the field in the state.
- imports: Imports needed to render this var.
- hooks: Hooks that need to be present in the component to render this var.
- deps: Dependencies of the var for useCallback.
- position: Position of the hook in the component.
- components: Components that are part of this var.
- """
- if isinstance(hooks, str):
- hooks = [hooks]
- if not isinstance(hooks, dict):
- hooks = dict.fromkeys(hooks or [])
- immutable_imports: ParsedImportTuple = tuple(
- (k, tuple(v)) for k, v in parse_imports(imports or {}).items()
- )
- object.__setattr__(self, "state", state)
- object.__setattr__(self, "field_name", field_name)
- object.__setattr__(self, "imports", immutable_imports)
- object.__setattr__(self, "hooks", tuple(hooks or {}))
- object.__setattr__(self, "deps", tuple(deps or []))
- object.__setattr__(self, "position", position or None)
- object.__setattr__(self, "components", tuple(components or []))
- if hooks and any(hooks.values()):
- # Merge our dependencies first, so they can be referenced.
- merged_var_data = VarData.merge(*hooks.values(), self)
- if merged_var_data is not None:
- object.__setattr__(self, "state", merged_var_data.state)
- object.__setattr__(self, "field_name", merged_var_data.field_name)
- object.__setattr__(self, "imports", merged_var_data.imports)
- object.__setattr__(self, "hooks", merged_var_data.hooks)
- object.__setattr__(self, "deps", merged_var_data.deps)
- object.__setattr__(self, "position", merged_var_data.position)
- object.__setattr__(self, "components", merged_var_data.components)
- def old_school_imports(self) -> ImportDict:
- """Return the imports as a mutable dict.
- Returns:
- The imports as a mutable dict.
- """
- return {k: list(v) for k, v in self.imports}
- def merge(*all: VarData | None) -> VarData | None:
- """Merge multiple var data objects.
- Args:
- *all: The var data objects to merge.
- Raises:
- ReflexError: If trying to merge VarData with different positions.
- Returns:
- The merged var data object.
- # noqa: DAR102 *all
- """
- all_var_datas = list(filter(None, all))
- if not all_var_datas:
- return None
- if len(all_var_datas) == 1:
- return all_var_datas[0]
- # Get the first non-empty field name or default to empty string.
- field_name = next(
- (var_data.field_name for var_data in all_var_datas if var_data.field_name),
- "",
- )
- # Get the first non-empty state or default to empty string.
- state = next(
- (var_data.state for var_data in all_var_datas if var_data.state), ""
- )
- hooks: dict[str, VarData | None] = {
- hook: None for var_data in all_var_datas for hook in var_data.hooks
- }
- _imports = imports.merge_imports(
- *(var_data.imports for var_data in all_var_datas)
- )
- deps = [dep for var_data in all_var_datas for dep in var_data.deps]
- positions = list(
- {
- var_data.position
- for var_data in all_var_datas
- if var_data.position is not None
- }
- )
- if positions:
- if len(positions) > 1:
- raise exceptions.ReflexError(
- f"Cannot merge var data with different positions: {positions}"
- )
- position = positions[0]
- else:
- position = None
- components = tuple(
- component for var_data in all_var_datas for component in var_data.components
- )
- return VarData(
- state=state,
- field_name=field_name,
- imports=_imports,
- hooks=hooks,
- deps=deps,
- position=position,
- components=components,
- )
- def __bool__(self) -> bool:
- """Check if the var data is non-empty.
- Returns:
- True if any field is set to a non-default value.
- """
- return bool(
- self.state
- or self.imports
- or self.hooks
- or self.field_name
- or self.deps
- or self.position
- or self.components
- )
- @classmethod
- def from_state(cls, state: type[BaseState] | str, field_name: str = "") -> VarData:
- """Set the state of the var.
- Args:
- state: The state to set or the full name of the state.
- field_name: The name of the field in the state. Optional.
- Returns:
- The var with the set state.
- """
- from reflex.utils import format
- state_name = state if isinstance(state, str) else state.get_full_name()
- return VarData(
- state=state_name,
- field_name=field_name,
- hooks={
- "const {0} = useContext(StateContexts.{0})".format(
- format.format_state_name(state_name)
- ): None
- },
- imports={
- f"$/{constants.Dirs.CONTEXTS_PATH}": [ImportVar(tag="StateContexts")],
- "react": [ImportVar(tag="useContext")],
- },
- )
- def _decode_var_immutable(value: str) -> tuple[VarData | None, str]:
- """Decode the state name from a formatted var.
- Args:
- value: The value to extract the state name from.
- Returns:
- The extracted state name and the value without the state name.
- """
- var_datas = []
- if isinstance(value, str):
- # fast path if there is no encoded VarData
- if constants.REFLEX_VAR_OPENING_TAG not in value:
- return None, value
- offset = 0
- # Find all tags.
- while m := _decode_var_pattern.search(value):
- start, end = m.span()
- value = value[:start] + value[end:]
- serialized_data = m.group(1)
- if serialized_data.isnumeric() or (
- serialized_data[0] == "-" and serialized_data[1:].isnumeric()
- ):
- # This is a global immutable var.
- var = _global_vars[int(serialized_data)]
- var_data = var._get_all_var_data()
- if var_data is not None:
- var_datas.append(var_data)
- offset += end - start
- return VarData.merge(*var_datas) if var_datas else None, value
- def can_use_in_object_var(cls: GenericType) -> bool:
- """Check if the class can be used in an ObjectVar.
- Args:
- cls: The class to check.
- Returns:
- Whether the class can be used in an ObjectVar.
- """
- if types.is_union(cls):
- return all(can_use_in_object_var(t) for t in types.get_args(cls))
- return (
- inspect.isclass(cls)
- and not safe_issubclass(cls, Var)
- and serializers.can_serialize(cls, dict)
- )
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- )
- class Var(Generic[VAR_TYPE]):
- """Base class for immutable vars."""
- # The name of the var.
- _js_expr: str = dataclasses.field()
- # The type of the var.
- _var_type: types.GenericType = dataclasses.field(default=Any)
- # Extra metadata associated with the Var
- _var_data: VarData | None = dataclasses.field(default=None)
- def __str__(self) -> str:
- """String representation of the var. Guaranteed to be a valid Javascript expression.
- Returns:
- The name of the var.
- """
- return self._js_expr
- @property
- def _var_is_local(self) -> bool:
- """Whether this is a local javascript variable.
- Returns:
- False
- """
- return False
- @property
- @deprecated("Use `_js_expr` instead.")
- def _var_name(self) -> str:
- """The name of the var.
- Returns:
- The name of the var.
- """
- return self._js_expr
- @property
- def _var_field_name(self) -> str:
- """The name of the field.
- Returns:
- The name of the field.
- """
- var_data = self._get_all_var_data()
- field_name = var_data.field_name if var_data else None
- return field_name or self._js_expr
- @property
- @deprecated("Use `_js_expr` instead.")
- def _var_name_unwrapped(self) -> str:
- """The name of the var without extra curly braces.
- Returns:
- The name of the var.
- """
- return self._js_expr
- @property
- def _var_is_string(self) -> bool:
- """Whether the var is a string literal.
- Returns:
- False
- """
- return False
- def __init_subclass__(
- cls,
- python_types: tuple[GenericType, ...] | GenericType = types.Unset(),
- default_type: GenericType = types.Unset(),
- **kwargs,
- ):
- """Initialize the subclass.
- Args:
- python_types: The python types that the var represents.
- default_type: The default type of the var. Defaults to the first python type.
- **kwargs: Additional keyword arguments.
- """
- super().__init_subclass__(**kwargs)
- if python_types or default_type:
- python_types = (
- (python_types if isinstance(python_types, tuple) else (python_types,))
- if python_types
- else ()
- )
- default_type = default_type or (python_types[0] if python_types else Any)
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- slots=True,
- )
- class ToVarOperation(ToOperation, cls):
- """Base class of converting a var to another var type."""
- _original: Var = dataclasses.field(
- default=Var(_js_expr="null", _var_type=None),
- )
- _default_var_type: ClassVar[GenericType] = default_type
- new_to_var_operation_name = f"{cls.__name__.removesuffix('Var')}CastedVar"
- ToVarOperation.__qualname__ = (
- ToVarOperation.__qualname__.removesuffix(ToVarOperation.__name__)
- + new_to_var_operation_name
- )
- ToVarOperation.__name__ = new_to_var_operation_name
- _var_subclasses.append(VarSubclassEntry(cls, ToVarOperation, python_types))
- def __post_init__(self):
- """Post-initialize the var.
- Raises:
- TypeError: If _js_expr is not a string.
- """
- if not isinstance(self._js_expr, str):
- raise TypeError(
- f"Expected _js_expr to be a string, got value {self._js_expr!r} of type {type(self._js_expr).__name__}"
- )
- if self._var_data is not None and not isinstance(self._var_data, VarData):
- raise TypeError(
- f"Expected _var_data to be a VarData, got value {self._var_data!r} of type {type(self._var_data).__name__}"
- )
- # Decode any inline Var markup and apply it to the instance
- _var_data, _js_expr = _decode_var_immutable(self._js_expr)
- if _var_data or _js_expr != self._js_expr:
- self.__init__(
- _js_expr=_js_expr,
- _var_type=self._var_type,
- _var_data=VarData.merge(self._var_data, _var_data),
- )
- def __hash__(self) -> int:
- """Define a hash function for the var.
- Returns:
- The hash of the var.
- """
- return hash((self._js_expr, self._var_type, self._var_data))
- def _get_all_var_data(self) -> VarData | None:
- """Get all VarData associated with the Var.
- Returns:
- The VarData of the components and all of its children.
- """
- return self._var_data
- def equals(self, other: Var) -> bool:
- """Check if two vars are equal.
- Args:
- other: The other var to compare.
- Returns:
- Whether the vars are equal.
- """
- return (
- self._js_expr == other._js_expr
- and self._var_type == other._var_type
- and self._get_all_var_data() == other._get_all_var_data()
- )
- @overload
- def _replace(
- self,
- _var_type: type[OTHER_VAR_TYPE],
- merge_var_data: VarData | None = None,
- **kwargs: Any,
- ) -> Var[OTHER_VAR_TYPE]: ...
- @overload
- def _replace(
- self,
- _var_type: GenericType | None = None,
- merge_var_data: VarData | None = None,
- **kwargs: Any,
- ) -> Self: ...
- def _replace(
- self,
- _var_type: GenericType | None = None,
- merge_var_data: VarData | None = None,
- **kwargs: Any,
- ) -> Self | Var:
- """Make a copy of this Var with updated fields.
- Args:
- _var_type: The new type of the Var.
- merge_var_data: VarData to merge into the existing VarData.
- **kwargs: Var fields to update.
- Returns:
- A new Var with the updated fields overwriting the corresponding fields in this Var.
- Raises:
- TypeError: If _var_is_local, _var_is_string, or _var_full_name_needs_state_prefix is not None.
- """
- if kwargs.get("_var_is_local", False) is not False:
- raise TypeError("The _var_is_local argument is not supported for Var.")
- if kwargs.get("_var_is_string", False) is not False:
- raise TypeError("The _var_is_string argument is not supported for Var.")
- if kwargs.get("_var_full_name_needs_state_prefix", False) is not False:
- raise TypeError(
- "The _var_full_name_needs_state_prefix argument is not supported for Var."
- )
- value_with_replaced = dataclasses.replace(
- self,
- _var_type=_var_type or self._var_type,
- _var_data=VarData.merge(
- kwargs.get("_var_data", self._var_data), merge_var_data
- ),
- **kwargs,
- )
- if (js_expr := kwargs.get("_js_expr")) is not None:
- object.__setattr__(value_with_replaced, "_js_expr", js_expr)
- return value_with_replaced
- @overload
- @classmethod
- def create( # pyright: ignore[reportOverlappingOverload]
- cls,
- value: NoReturn,
- _var_data: VarData | None = None,
- ) -> Var[Any]: ...
- @overload
- @classmethod
- def create( # pyright: ignore[reportOverlappingOverload]
- cls,
- value: bool,
- _var_data: VarData | None = None,
- ) -> LiteralBooleanVar: ...
- @overload
- @classmethod
- def create(
- cls,
- value: int,
- _var_data: VarData | None = None,
- ) -> LiteralNumberVar[int]: ...
- @overload
- @classmethod
- def create(
- cls,
- value: float,
- _var_data: VarData | None = None,
- ) -> LiteralNumberVar[float]: ...
- @overload
- @classmethod
- def create(
- cls,
- value: Decimal,
- _var_data: VarData | None = None,
- ) -> LiteralNumberVar[Decimal]: ...
- @overload
- @classmethod
- def create( # pyright: ignore [reportOverlappingOverload]
- cls,
- value: str,
- _var_data: VarData | None = None,
- ) -> LiteralStringVar: ...
- @overload
- @classmethod
- def create( # pyright: ignore [reportOverlappingOverload]
- cls,
- value: STRING_T,
- _var_data: VarData | None = None,
- ) -> StringVar[STRING_T]: ...
- @overload
- @classmethod
- def create( # pyright: ignore[reportOverlappingOverload]
- cls,
- value: None,
- _var_data: VarData | None = None,
- ) -> LiteralNoneVar: ...
- @overload
- @classmethod
- def create(
- cls,
- value: MAPPING_TYPE,
- _var_data: VarData | None = None,
- ) -> LiteralObjectVar[MAPPING_TYPE]: ...
- @overload
- @classmethod
- def create(
- cls,
- value: SEQUENCE_TYPE,
- _var_data: VarData | None = None,
- ) -> LiteralArrayVar[SEQUENCE_TYPE]: ...
- @overload
- @classmethod
- def create(
- cls,
- value: OTHER_VAR_TYPE,
- _var_data: VarData | None = None,
- ) -> Var[OTHER_VAR_TYPE]: ...
- @classmethod
- def create(
- cls,
- value: OTHER_VAR_TYPE,
- _var_data: VarData | None = None,
- ) -> Var[OTHER_VAR_TYPE]:
- """Create a var from a value.
- Args:
- value: The value to create the var from.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The var.
- """
- # If the value is already a var, do nothing.
- if isinstance(value, Var):
- return value
- return LiteralVar.create(value, _var_data=_var_data)
- @classmethod
- @deprecated("Use `.create()` instead.")
- def create_safe(
- cls,
- *args: Any,
- **kwargs: Any,
- ) -> Var:
- """Create a var from a value.
- Args:
- *args: The arguments to create the var from.
- **kwargs: The keyword arguments to create the var from.
- Returns:
- The var.
- """
- return cls.create(*args, **kwargs)
- def __format__(self, format_spec: str) -> str:
- """Format the var into a Javascript equivalent to an f-string.
- Args:
- format_spec: The format specifier (Ignored for now).
- Returns:
- The formatted var.
- """
- hashed_var = hash(self)
- _global_vars[hashed_var] = self
- # Encode the _var_data into the formatted output for tracking purposes.
- return f"{constants.REFLEX_VAR_OPENING_TAG}{hashed_var}{constants.REFLEX_VAR_CLOSING_TAG}{self._js_expr}"
- @overload
- def to(self, output: type[str]) -> StringVar: ...
- @overload
- def to(self, output: type[bool]) -> BooleanVar: ...
- @overload
- def to(self, output: type[int]) -> NumberVar[int]: ...
- @overload
- def to(self, output: type[float]) -> NumberVar[float]: ...
- @overload
- def to(self, output: type[Decimal]) -> NumberVar[Decimal]: ...
- @overload
- def to(
- self,
- output: type[list] | type[tuple] | type[set],
- ) -> ArrayVar: ...
- @overload
- def to(
- self,
- output: type[MAPPING_TYPE],
- ) -> ObjectVar[MAPPING_TYPE]: ...
- @overload
- def to(
- self, output: type[ObjectVar], var_type: type[VAR_INSIDE]
- ) -> ObjectVar[VAR_INSIDE]: ...
- @overload
- def to(
- self, output: type[ObjectVar], var_type: None = None
- ) -> ObjectVar[VAR_TYPE]: ...
- @overload
- def to(self, output: VAR_SUBCLASS, var_type: None = None) -> VAR_SUBCLASS: ...
- @overload
- def to(
- self,
- output: type[OUTPUT] | types.GenericType,
- var_type: types.GenericType | None = None,
- ) -> OUTPUT: ...
- def to(
- self,
- output: type[OUTPUT] | types.GenericType,
- var_type: types.GenericType | None = None,
- ) -> Var:
- """Convert the var to a different type.
- Args:
- output: The output type.
- var_type: The type of the var.
- Returns:
- The converted var.
- """
- from .object import ObjectVar
- fixed_output_type = get_origin(output) or output
- # If the first argument is a python type, we map it to the corresponding Var type.
- for var_subclass in _var_subclasses[::-1]:
- if fixed_output_type in var_subclass.python_types or safe_issubclass(
- fixed_output_type, var_subclass.python_types
- ):
- return self.to(var_subclass.var_subclass, output)
- if fixed_output_type is None:
- return get_to_operation(NoneVar).create(self) # pyright: ignore [reportReturnType]
- # Handle fixed_output_type being Base or a dataclass.
- if can_use_in_object_var(output):
- return self.to(ObjectVar, output)
- if inspect.isclass(output):
- for var_subclass in _var_subclasses[::-1]:
- if safe_issubclass(output, var_subclass.var_subclass):
- current_var_type = self._var_type
- if current_var_type is Any:
- new_var_type = var_type
- else:
- new_var_type = var_type or current_var_type
- to_operation_return = var_subclass.to_var_subclass.create(
- value=self, _var_type=new_var_type
- )
- return to_operation_return # pyright: ignore [reportReturnType]
- # If we can't determine the first argument, we just replace the _var_type.
- if not safe_issubclass(output, Var) or var_type is None:
- return dataclasses.replace(
- self,
- _var_type=output,
- )
- # We couldn't determine the output type to be any other Var type, so we replace the _var_type.
- if var_type is not None:
- return dataclasses.replace(
- self,
- _var_type=var_type,
- )
- return self
- @overload
- def guess_type(self: Var[NoReturn]) -> Var[Any]: ... # pyright: ignore [reportOverlappingOverload]
- @overload
- def guess_type(self: Var[str]) -> StringVar: ...
- @overload
- def guess_type(self: Var[bool]) -> BooleanVar: ...
- @overload
- def guess_type(self: Var[int] | Var[float] | Var[int | float]) -> NumberVar: ...
- @overload
- def guess_type(self: Var[BASE_TYPE]) -> ObjectVar[BASE_TYPE]: ...
- @overload
- def guess_type(self) -> Self: ...
- def guess_type(self) -> Var:
- """Guesses the type of the variable based on its `_var_type` attribute.
- Returns:
- Var: The guessed type of the variable.
- Raises:
- TypeError: If the type is not supported for guessing.
- """
- from .object import ObjectVar
- var_type = self._var_type
- if var_type is None:
- return self.to(None)
- if var_type is NoReturn:
- return self.to(Any)
- var_type = types.value_inside_optional(var_type)
- if var_type is Any:
- return self
- fixed_type = get_origin(var_type) or var_type
- if fixed_type in types.UnionTypes:
- inner_types = get_args(var_type)
- non_optional_inner_types = [
- types.value_inside_optional(inner_type) for inner_type in inner_types
- ]
- fixed_inner_types = [
- get_origin(inner_type) or inner_type
- for inner_type in non_optional_inner_types
- ]
- for var_subclass in _var_subclasses[::-1]:
- if all(
- safe_issubclass(t, var_subclass.python_types)
- for t in fixed_inner_types
- ):
- return self.to(var_subclass.var_subclass, self._var_type)
- if can_use_in_object_var(var_type):
- return self.to(ObjectVar, self._var_type)
- return self
- if fixed_type is Literal:
- args = get_args(var_type)
- fixed_type = unionize(*(type(arg) for arg in args))
- if not inspect.isclass(fixed_type):
- raise TypeError(f"Unsupported type {var_type} for guess_type.")
- if fixed_type is None:
- return self.to(None)
- for var_subclass in _var_subclasses[::-1]:
- if safe_issubclass(fixed_type, var_subclass.python_types):
- return self.to(var_subclass.var_subclass, self._var_type)
- if can_use_in_object_var(fixed_type):
- return self.to(ObjectVar, self._var_type)
- return self
- def _get_default_value(self) -> Any:
- """Get the default value of the var.
- Returns:
- The default value of the var.
- Raises:
- ImportError: If the var is a dataframe and pandas is not installed.
- """
- if types.is_optional(self._var_type):
- return None
- type_ = (
- get_origin(self._var_type)
- if types.is_generic_alias(self._var_type)
- else self._var_type
- )
- if type_ is Literal:
- args = get_args(self._var_type)
- return args[0] if args else None
- if safe_issubclass(type_, str):
- return ""
- if safe_issubclass(type_, types.get_args(int | float)):
- return 0
- if safe_issubclass(type_, bool):
- return False
- if safe_issubclass(type_, list):
- return []
- if safe_issubclass(type_, Mapping):
- return {}
- if safe_issubclass(type_, tuple):
- return ()
- if types.is_dataframe(type_):
- try:
- import pandas as pd
- return pd.DataFrame()
- except ImportError as e:
- raise ImportError(
- "Please install pandas to use dataframes in your app."
- ) from e
- return set() if safe_issubclass(type_, set) else None
- def _get_setter_name(self, include_state: bool = True) -> str:
- """Get the name of the var's generated setter function.
- Args:
- include_state: Whether to include the state name in the setter name.
- Returns:
- The name of the setter function.
- """
- setter = constants.SETTER_PREFIX + self._var_field_name
- var_data = self._get_all_var_data()
- if var_data is None:
- return setter
- if not include_state or var_data.state == "":
- return setter
- return ".".join((var_data.state, setter))
- def _get_setter(self) -> Callable[[BaseState, Any], None]:
- """Get the var's setter function.
- Returns:
- A function that that creates a setter for the var.
- """
- actual_name = self._var_field_name
- def setter(state: Any, value: Any):
- """Get the setter for the var.
- Args:
- state: The state within which we add the setter function.
- value: The value to set.
- """
- if self._var_type in [int, float]:
- try:
- value = self._var_type(value)
- setattr(state, actual_name, value)
- except ValueError:
- console.debug(
- f"{type(state).__name__}.{self._js_expr}: Failed conversion of {value!s} to '{self._var_type.__name__}'. Value not set.",
- )
- else:
- setattr(state, actual_name, value)
- setter.__annotations__["value"] = self._var_type
- setter.__qualname__ = self._get_setter_name()
- return setter
- def _var_set_state(self, state: type[BaseState] | str) -> Self:
- """Set the state of the var.
- Args:
- state: The state to set.
- Returns:
- The var with the state set.
- """
- formatted_state_name = (
- state
- if isinstance(state, str)
- else format_state_name(state.get_full_name())
- )
- return StateOperation.create( # pyright: ignore [reportReturnType]
- formatted_state_name,
- self,
- _var_data=VarData.merge(
- VarData.from_state(state, self._js_expr), self._var_data
- ),
- ).guess_type()
- def __eq__(self, other: Var | Any) -> BooleanVar:
- """Check if the current variable is equal to the given variable.
- Args:
- other (Var | Any): The variable to compare with.
- Returns:
- BooleanVar: A BooleanVar object representing the result of the equality check.
- """
- from .number import equal_operation
- return equal_operation(self, other)
- def __ne__(self, other: Var | Any) -> BooleanVar:
- """Check if the current object is not equal to the given object.
- Parameters:
- other (Var | Any): The object to compare with.
- Returns:
- BooleanVar: A BooleanVar object representing the result of the comparison.
- """
- from .number import equal_operation
- return ~equal_operation(self, other)
- def bool(self) -> BooleanVar:
- """Convert the var to a boolean.
- Returns:
- The boolean var.
- """
- from .number import boolify
- return boolify(self)
- def is_none(self) -> BooleanVar:
- """Check if the var is None.
- Returns:
- A BooleanVar object representing the result of the check.
- """
- from .number import is_not_none_operation
- return ~is_not_none_operation(self)
- def is_not_none(self) -> BooleanVar:
- """Check if the var is not None.
- Returns:
- A BooleanVar object representing the result of the check.
- """
- from .number import is_not_none_operation
- return is_not_none_operation(self)
- def __and__(
- self, other: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical AND operation on the current instance and another variable.
- Args:
- other: The variable to perform the logical AND operation with.
- Returns:
- A `BooleanVar` object representing the result of the logical AND operation.
- """
- return and_operation(self, other)
- def __rand__(
- self, other: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical AND operation on the current instance and another variable.
- Args:
- other: The variable to perform the logical AND operation with.
- Returns:
- A `BooleanVar` object representing the result of the logical AND operation.
- """
- return and_operation(other, self)
- def __or__(
- self, other: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical OR operation on the current instance and another variable.
- Args:
- other: The variable to perform the logical OR operation with.
- Returns:
- A `BooleanVar` object representing the result of the logical OR operation.
- """
- return or_operation(self, other)
- def __ror__(
- self, other: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical OR operation on the current instance and another variable.
- Args:
- other: The variable to perform the logical OR operation with.
- Returns:
- A `BooleanVar` object representing the result of the logical OR operation.
- """
- return or_operation(other, self)
- def __invert__(self) -> BooleanVar:
- """Perform a logical NOT operation on the current instance.
- Returns:
- A `BooleanVar` object representing the result of the logical NOT operation.
- """
- return ~self.bool()
- def to_string(self, use_json: bool = True) -> StringVar:
- """Convert the var to a string.
- Args:
- use_json: Whether to use JSON stringify. If False, uses Object.prototype.toString.
- Returns:
- The string var.
- """
- from .function import JSON_STRINGIFY, PROTOTYPE_TO_STRING
- from .sequence import StringVar
- return (
- JSON_STRINGIFY.call(self).to(StringVar)
- if use_json
- else PROTOTYPE_TO_STRING.call(self).to(StringVar)
- )
- def _as_ref(self) -> Var:
- """Get a reference to the var.
- Returns:
- The reference to the var.
- """
- from .object import ObjectVar
- refs = Var(
- _js_expr="refs",
- _var_data=VarData(
- imports={
- f"$/{constants.Dirs.STATE_PATH}": [imports.ImportVar(tag="refs")]
- }
- ),
- ).to(ObjectVar, Mapping[str, str])
- return refs[LiteralVar.create(str(self))]
- @deprecated("Use `.js_type()` instead.")
- def _type(self) -> StringVar:
- """Returns the type of the object.
- This method uses the `typeof` function from the `FunctionStringVar` class
- to determine the type of the object.
- Returns:
- StringVar: A string variable representing the type of the object.
- """
- return self.js_type()
- def js_type(self) -> StringVar:
- """Returns the javascript type of the object.
- This method uses the `typeof` function from the `FunctionStringVar` class
- to determine the type of the object.
- Returns:
- StringVar: A string variable representing the type of the object.
- """
- from .function import FunctionStringVar
- from .sequence import StringVar
- type_of = FunctionStringVar("typeof")
- return type_of.call(self).to(StringVar)
- def _without_data(self):
- """Create a copy of the var without the data.
- Returns:
- The var without the data.
- """
- return dataclasses.replace(self, _var_data=None)
- def __get__(self, instance: Any, owner: Any):
- """Get the var.
- Args:
- instance: The instance to get the var from.
- owner: The owner of the var.
- Returns:
- The var.
- """
- return self
- def _decode(self) -> Any:
- """Decode Var as a python value.
- Note that Var with state set cannot be decoded python-side and will be
- returned as full_name.
- Returns:
- The decoded value or the Var name.
- """
- if isinstance(self, LiteralVar):
- return self._var_value
- try:
- return json.loads(str(self))
- except ValueError:
- return str(self)
- @property
- def _var_state(self) -> str:
- """Compat method for getting the state.
- Returns:
- The state name associated with the var.
- """
- var_data = self._get_all_var_data()
- return var_data.state if var_data else ""
- @overload
- @classmethod
- def range(cls, stop: int | NumberVar, /) -> ArrayVar[Sequence[int]]: ...
- @overload
- @classmethod
- def range(
- cls,
- start: int | NumberVar,
- end: int | NumberVar,
- step: int | NumberVar = 1,
- /,
- ) -> ArrayVar[Sequence[int]]: ...
- @classmethod
- def range(
- cls,
- first_endpoint: int | NumberVar,
- second_endpoint: int | NumberVar | None = None,
- step: int | NumberVar | None = None,
- ) -> ArrayVar[Sequence[int]]:
- """Create a range of numbers.
- Args:
- first_endpoint: The end of the range if second_endpoint is not provided, otherwise the start of the range.
- second_endpoint: The end of the range.
- step: The step of the range.
- Returns:
- The range of numbers.
- """
- from .sequence import ArrayVar
- return ArrayVar.range(first_endpoint, second_endpoint, step)
- if not TYPE_CHECKING:
- def __getitem__(self, key: Any) -> Var:
- """Get the item from the var.
- Args:
- key: The key to get.
- Raises:
- UntypedVarError: If the var type is Any.
- TypeError: If the var type is Any.
- # noqa: DAR101 self
- """
- if self._var_type is Any:
- raise exceptions.UntypedVarError(
- self,
- f"access the item '{key}'",
- )
- raise TypeError(
- f"Var of type {self._var_type} does not support item access."
- )
- def __getattr__(self, name: str):
- """Get an attribute of the var.
- Args:
- name: The name of the attribute.
- Raises:
- VarAttributeError: If the attribute does not exist.
- UntypedVarError: If the var type is Any.
- TypeError: If the var type is Any.
- # noqa: DAR101 self
- """
- if name.startswith("_"):
- raise VarAttributeError(f"Attribute {name} not found.")
- if name == "contains":
- raise TypeError(
- f"Var of type {self._var_type} does not support contains check."
- )
- if name == "reverse":
- raise TypeError("Cannot reverse non-list var.")
- if self._var_type is Any:
- raise exceptions.UntypedVarError(
- self,
- f"access the attribute '{name}'",
- )
- raise VarAttributeError(
- f"The State var {escape(self._js_expr)} of type {escape(str(self._var_type))} has no attribute '{name}' or may have been annotated wrongly.",
- )
- def __bool__(self) -> bool:
- """Raise exception if using Var in a boolean context.
- Raises:
- VarTypeError: when attempting to bool-ify the Var.
- # noqa: DAR101 self
- """
- raise VarTypeError(
- f"Cannot convert Var {str(self)!r} to bool for use with `if`, `and`, `or`, and `not`. "
- "Instead use `rx.cond` and bitwise operators `&` (and), `|` (or), `~` (invert)."
- )
- def __iter__(self) -> Any:
- """Raise exception if using Var in an iterable context.
- Raises:
- VarTypeError: when attempting to iterate over the Var.
- # noqa: DAR101 self
- """
- raise VarTypeError(
- f"Cannot iterate over Var {str(self)!r}. Instead use `rx.foreach`."
- )
- def __contains__(self, _: Any) -> Var:
- """Override the 'in' operator to alert the user that it is not supported.
- Raises:
- VarTypeError: the operation is not supported
- # noqa: DAR101 self
- """
- raise VarTypeError(
- "'in' operator not supported for Var types, use Var.contains() instead."
- )
- OUTPUT = TypeVar("OUTPUT", bound=Var)
- VAR_SUBCLASS = TypeVar("VAR_SUBCLASS", bound=Var)
- VAR_INSIDE = TypeVar("VAR_INSIDE")
- class ToOperation:
- """A var operation that converts a var to another type."""
- def __getattr__(self, name: str) -> Any:
- """Get an attribute of the var.
- Args:
- name: The name of the attribute.
- Returns:
- The attribute of the var.
- """
- from .object import ObjectVar
- if isinstance(self, ObjectVar) and name != "_js_expr":
- return ObjectVar.__getattr__(self, name)
- return getattr(self._original, name)
- def __post_init__(self):
- """Post initialization."""
- object.__delattr__(self, "_js_expr")
- def __hash__(self) -> int:
- """Calculate the hash value of the object.
- Returns:
- int: The hash value of the object.
- """
- return hash(self._original)
- def _get_all_var_data(self) -> VarData | None:
- """Get all the var data.
- Returns:
- The var data.
- """
- return VarData.merge(
- self._original._get_all_var_data(),
- self._var_data,
- )
- @classmethod
- def create(
- cls,
- value: Var,
- _var_type: GenericType | None = None,
- _var_data: VarData | None = None,
- ):
- """Create a ToOperation.
- Args:
- value: The value of the var.
- _var_type: The type of the Var.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The ToOperation.
- """
- return cls(
- _js_expr="", # pyright: ignore [reportCallIssue]
- _var_data=_var_data, # pyright: ignore [reportCallIssue]
- _var_type=_var_type or cls._default_var_type, # pyright: ignore [reportCallIssue, reportAttributeAccessIssue]
- _original=value, # pyright: ignore [reportCallIssue]
- )
- class LiteralVar(Var):
- """Base class for immutable literal vars."""
- def __init_subclass__(cls, **kwargs):
- """Initialize the subclass.
- Args:
- **kwargs: Additional keyword arguments.
- Raises:
- TypeError: If the LiteralVar subclass does not have a corresponding Var subclass.
- """
- super().__init_subclass__(**kwargs)
- bases = cls.__bases__
- bases_normalized = [
- base if inspect.isclass(base) else get_origin(base) for base in bases
- ]
- possible_bases = [
- base
- for base in bases_normalized
- if safe_issubclass(base, Var) and base != LiteralVar
- ]
- if not possible_bases:
- raise TypeError(
- f"LiteralVar subclass {cls} must have a base class that is a subclass of Var and not LiteralVar."
- )
- var_subclasses = [
- var_subclass
- for var_subclass in _var_subclasses
- if var_subclass.var_subclass in possible_bases
- ]
- if not var_subclasses:
- raise TypeError(
- f"LiteralVar {cls} must have a base class annotated with `python_types`."
- )
- if len(var_subclasses) != 1:
- raise TypeError(
- f"LiteralVar {cls} must have exactly one base class annotated with `python_types`."
- )
- var_subclass = var_subclasses[0]
- # Remove the old subclass, happens because __init_subclass__ is called twice
- # for each subclass. This is because of __slots__ in dataclasses.
- for var_literal_subclass in list(_var_literal_subclasses):
- if var_literal_subclass[1] is var_subclass:
- _var_literal_subclasses.remove(var_literal_subclass)
- _var_literal_subclasses.append((cls, var_subclass))
- @classmethod
- def _create_literal_var(
- cls,
- value: Any,
- _var_data: VarData | None = None,
- ) -> Var:
- """Create a var from a value.
- Args:
- value: The value to create the var from.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The var.
- Raises:
- TypeError: If the value is not a supported type for LiteralVar.
- """
- from .object import LiteralObjectVar
- from .sequence import ArrayVar, LiteralStringVar
- if isinstance(value, Var):
- if _var_data is None:
- return value
- return value._replace(merge_var_data=_var_data)
- for literal_subclass, var_subclass in _var_literal_subclasses[::-1]:
- if isinstance(value, var_subclass.python_types):
- return literal_subclass.create(value, _var_data=_var_data)
- from reflex.event import EventHandler
- from reflex.utils.format import get_event_handler_parts
- if isinstance(value, EventHandler):
- return Var(_js_expr=".".join(filter(None, get_event_handler_parts(value))))
- serialized_value = serializers.serialize(value)
- if serialized_value is not None:
- if isinstance(serialized_value, Mapping):
- return LiteralObjectVar.create(
- serialized_value,
- _var_type=type(value),
- _var_data=_var_data,
- )
- if isinstance(serialized_value, str):
- return LiteralStringVar.create(
- serialized_value, _var_type=type(value), _var_data=_var_data
- )
- return LiteralVar.create(serialized_value, _var_data=_var_data)
- if isinstance(value, Base):
- # get the fields of the pydantic class
- fields = value.__fields__.keys()
- one_level_dict = {field: getattr(value, field) for field in fields}
- return LiteralObjectVar.create(
- {
- field: value
- for field, value in one_level_dict.items()
- if not callable(value)
- },
- _var_type=type(value),
- _var_data=_var_data,
- )
- if dataclasses.is_dataclass(value) and not isinstance(value, type):
- return LiteralObjectVar.create(
- {
- k: (None if callable(v) else v)
- for k, v in dataclasses.asdict(value).items()
- },
- _var_type=type(value),
- _var_data=_var_data,
- )
- if isinstance(value, range):
- return ArrayVar.range(value.start, value.stop, value.step)
- raise TypeError(
- f"Unsupported type {type(value)} for LiteralVar. Tried to create a LiteralVar from {value}."
- )
- if not TYPE_CHECKING:
- create = _create_literal_var
- def __post_init__(self):
- """Post-initialize the var."""
- @property
- def _var_value(self) -> Any:
- raise NotImplementedError(
- "LiteralVar subclasses must implement the _var_value property."
- )
- def json(self) -> str:
- """Serialize the var to a JSON string.
- Raises:
- NotImplementedError: If the method is not implemented.
- """
- raise NotImplementedError(
- "LiteralVar subclasses must implement the json method."
- )
- @serializers.serializer
- def serialize_literal(value: LiteralVar):
- """Serialize a Literal type.
- Args:
- value: The Literal to serialize.
- Returns:
- The serialized Literal.
- """
- return value._var_value
- def get_python_literal(value: LiteralVar | Any) -> Any | None:
- """Get the Python literal value.
- Args:
- value: The value to get the Python literal value of.
- Returns:
- The Python literal value.
- """
- if isinstance(value, LiteralVar):
- return value._var_value
- if isinstance(value, Var):
- return None
- return value
- P = ParamSpec("P")
- T = TypeVar("T")
- # NoReturn is used to match CustomVarOperationReturn with no type hint.
- @overload
- def var_operation( # pyright: ignore [reportOverlappingOverload]
- func: Callable[P, CustomVarOperationReturn[NoReturn]],
- ) -> Callable[P, Var]: ...
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[None]],
- ) -> Callable[P, NoneVar]: ...
- @overload
- def var_operation( # pyright: ignore [reportOverlappingOverload]
- func: Callable[P, CustomVarOperationReturn[bool]]
- | Callable[P, CustomVarOperationReturn[bool | None]],
- ) -> Callable[P, BooleanVar]: ...
- NUMBER_T = TypeVar("NUMBER_T", int, float, int | float)
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[NUMBER_T]]
- | Callable[P, CustomVarOperationReturn[NUMBER_T | None]],
- ) -> Callable[P, NumberVar[NUMBER_T]]: ...
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[str]]
- | Callable[P, CustomVarOperationReturn[str | None]],
- ) -> Callable[P, StringVar]: ...
- LIST_T = TypeVar("LIST_T", bound=Sequence)
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[LIST_T]]
- | Callable[P, CustomVarOperationReturn[LIST_T | None]],
- ) -> Callable[P, ArrayVar[LIST_T]]: ...
- OBJECT_TYPE = TypeVar("OBJECT_TYPE", bound=Mapping)
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[OBJECT_TYPE]]
- | Callable[P, CustomVarOperationReturn[OBJECT_TYPE | None]],
- ) -> Callable[P, ObjectVar[OBJECT_TYPE]]: ...
- @overload
- def var_operation(
- func: Callable[P, CustomVarOperationReturn[T]]
- | Callable[P, CustomVarOperationReturn[T | None]],
- ) -> Callable[P, Var[T]]: ...
- def var_operation( # pyright: ignore [reportInconsistentOverload]
- func: Callable[P, CustomVarOperationReturn[T]],
- ) -> Callable[P, Var[T]]:
- """Decorator for creating a var operation.
- Example:
- ```python
- @var_operation
- def add(a: NumberVar, b: NumberVar):
- return custom_var_operation(f"{a} + {b}")
- ```
- Args:
- func: The function to decorate.
- Returns:
- The decorated function.
- """
- @functools.wraps(func)
- def wrapper(*args: P.args, **kwargs: P.kwargs) -> Var[T]:
- func_args = list(inspect.signature(func).parameters)
- args_vars = {
- func_args[i]: (LiteralVar.create(arg) if not isinstance(arg, Var) else arg)
- for i, arg in enumerate(args)
- }
- kwargs_vars = {
- key: LiteralVar.create(value) if not isinstance(value, Var) else value
- for key, value in kwargs.items()
- }
- return CustomVarOperation.create(
- name=func.__name__,
- args=tuple(list(args_vars.items()) + list(kwargs_vars.items())),
- return_var=func(*args_vars.values(), **kwargs_vars), # pyright: ignore [reportCallIssue, reportReturnType]
- ).guess_type()
- return wrapper
- def figure_out_type(value: Any) -> types.GenericType:
- """Figure out the type of the value.
- Args:
- value: The value to figure out the type of.
- Returns:
- The type of the value.
- """
- if isinstance(value, Var):
- return value._var_type
- type_ = type(value)
- if has_args(type_):
- return type_
- if isinstance(value, list):
- if not value:
- return Sequence[NoReturn]
- return Sequence[unionize(*(figure_out_type(v) for v in value))]
- if isinstance(value, set):
- return set[unionize(*(figure_out_type(v) for v in value))]
- if isinstance(value, tuple):
- if not value:
- return tuple[NoReturn, ...]
- if len(value) <= 5:
- return tuple[tuple(figure_out_type(v) for v in value)]
- return tuple[unionize(*(figure_out_type(v) for v in value)), ...]
- if isinstance(value, Mapping):
- if not value:
- return Mapping[NoReturn, NoReturn]
- return Mapping[
- unionize(*(figure_out_type(k) for k in value)),
- unionize(*(figure_out_type(v) for v in value.values())),
- ]
- return type(value)
- GLOBAL_CACHE = {}
- class cached_property: # noqa: N801
- """A cached property that caches the result of the function."""
- def __init__(self, func: Callable):
- """Initialize the cached_property.
- Args:
- func: The function to cache.
- """
- self._func = func
- self._attrname = None
- def __set_name__(self, owner: Any, name: str):
- """Set the name of the cached property.
- Args:
- owner: The owner of the cached property.
- name: The name of the cached property.
- Raises:
- TypeError: If the cached property is assigned to two different names.
- """
- if self._attrname is None:
- self._attrname = name
- original_del = getattr(owner, "__del__", None)
- def delete_property(this: Any):
- """Delete the cached property.
- Args:
- this: The object to delete the cached property from.
- """
- cached_field_name = "_reflex_cache_" + name
- try:
- unique_id = object.__getattribute__(this, cached_field_name)
- except AttributeError:
- if original_del is not None:
- original_del(this)
- return
- GLOBAL_CACHE.pop(unique_id, None)
- if original_del is not None:
- original_del(this)
- owner.__del__ = delete_property
- elif name != self._attrname:
- raise TypeError(
- "Cannot assign the same cached_property to two different names "
- f"({self._attrname!r} and {name!r})."
- )
- def __get__(self, instance: Any, owner: type | None = None):
- """Get the cached property.
- Args:
- instance: The instance to get the cached property from.
- owner: The owner of the cached property.
- Returns:
- The cached property.
- Raises:
- TypeError: If the class does not have __set_name__.
- """
- if self._attrname is None:
- raise TypeError(
- "Cannot use cached_property on a class without __set_name__."
- )
- cached_field_name = "_reflex_cache_" + self._attrname
- try:
- unique_id = object.__getattribute__(instance, cached_field_name)
- except AttributeError:
- unique_id = uuid.uuid4().int
- object.__setattr__(instance, cached_field_name, unique_id)
- if unique_id not in GLOBAL_CACHE:
- GLOBAL_CACHE[unique_id] = self._func(instance)
- return GLOBAL_CACHE[unique_id]
- cached_property_no_lock = cached_property
- class VarProtocol(Protocol):
- """A protocol for Var."""
- __dataclass_fields__: ClassVar[dict[str, dataclasses.Field[Any]]]
- @property
- def _js_expr(self) -> str: ...
- @property
- def _var_type(self) -> types.GenericType: ...
- @property
- def _var_data(self) -> VarData: ...
- class CachedVarOperation:
- """Base class for cached var operations to lower boilerplate code."""
- def __post_init__(self):
- """Post-initialize the CachedVarOperation."""
- object.__delattr__(self, "_js_expr")
- def __getattr__(self, name: str) -> Any:
- """Get an attribute of the var.
- Args:
- name: The name of the attribute.
- Returns:
- The attribute.
- """
- if name == "_js_expr":
- return self._cached_var_name
- parent_classes = inspect.getmro(type(self))
- next_class = parent_classes[parent_classes.index(CachedVarOperation) + 1]
- return next_class.__getattr__(self, name)
- def _get_all_var_data(self) -> VarData | None:
- """Get all VarData associated with the Var.
- Returns:
- The VarData of the components and all of its children.
- """
- return self._cached_get_all_var_data
- @cached_property_no_lock
- def _cached_get_all_var_data(self: VarProtocol) -> VarData | None:
- """Get the cached VarData.
- Returns:
- The cached VarData.
- """
- return VarData.merge(
- *(
- value._get_all_var_data() if isinstance(value, Var) else None
- for value in (
- getattr(self, field.name) for field in dataclasses.fields(self)
- )
- ),
- self._var_data,
- )
- def __hash__(self: DataclassInstance) -> int:
- """Calculate the hash of the object.
- Returns:
- The hash of the object.
- """
- return hash(
- (
- type(self).__name__,
- *[
- getattr(self, field.name)
- for field in dataclasses.fields(self)
- if field.name not in ["_js_expr", "_var_data", "_var_type"]
- ],
- )
- )
- def and_operation(
- a: Var[VAR_TYPE] | Any, b: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical AND operation on two variables.
- Args:
- a: The first variable.
- b: The second variable.
- Returns:
- The result of the logical AND operation.
- """
- return _and_operation(a, b)
- @var_operation
- def _and_operation(a: Var, b: Var):
- """Perform a logical AND operation on two variables.
- Args:
- a: The first variable.
- b: The second variable.
- Returns:
- The result of the logical AND operation.
- """
- return var_operation_return(
- js_expression=f"({a} && {b})",
- var_type=unionize(a._var_type, b._var_type),
- )
- def or_operation(
- a: Var[VAR_TYPE] | Any, b: Var[OTHER_VAR_TYPE] | Any
- ) -> Var[VAR_TYPE | OTHER_VAR_TYPE]:
- """Perform a logical OR operation on two variables.
- Args:
- a: The first variable.
- b: The second variable.
- Returns:
- The result of the logical OR operation.
- """
- return _or_operation(a, b)
- @var_operation
- def _or_operation(a: Var, b: Var):
- """Perform a logical OR operation on two variables.
- Args:
- a: The first variable.
- b: The second variable.
- Returns:
- The result of the logical OR operation.
- """
- return var_operation_return(
- js_expression=f"({a} || {b})",
- var_type=unionize(a._var_type, b._var_type),
- )
- RETURN_TYPE = TypeVar("RETURN_TYPE")
- DICT_KEY = TypeVar("DICT_KEY")
- DICT_VAL = TypeVar("DICT_VAL")
- LIST_INSIDE = TypeVar("LIST_INSIDE")
- class FakeComputedVarBaseClass(property):
- """A fake base class for ComputedVar to avoid inheriting from property."""
- __pydantic_run_validation__ = False
- def is_computed_var(obj: Any) -> TypeGuard[ComputedVar]:
- """Check if the object is a ComputedVar.
- Args:
- obj: The object to check.
- Returns:
- Whether the object is a ComputedVar.
- """
- return isinstance(obj, FakeComputedVarBaseClass)
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- slots=True,
- )
- class ComputedVar(Var[RETURN_TYPE]):
- """A field with computed getters."""
- # Whether to track dependencies and cache computed values
- _cache: bool = dataclasses.field(default=False)
- # Whether the computed var is a backend var
- _backend: bool = dataclasses.field(default=False)
- # The initial value of the computed var
- _initial_value: RETURN_TYPE | types.Unset = dataclasses.field(default=types.Unset())
- # Explicit var dependencies to track
- _static_deps: dict[str | None, set[str]] = dataclasses.field(default_factory=dict)
- # Whether var dependencies should be auto-determined
- _auto_deps: bool = dataclasses.field(default=True)
- # Interval at which the computed var should be updated
- _update_interval: datetime.timedelta | None = dataclasses.field(default=None)
- _fget: Callable[[BaseState], RETURN_TYPE] = dataclasses.field(
- default_factory=lambda: lambda _: None
- ) # pyright: ignore [reportAssignmentType]
- def __init__(
- self,
- fget: Callable[[BASE_STATE], RETURN_TYPE],
- initial_value: RETURN_TYPE | types.Unset = types.Unset(),
- cache: bool = True,
- deps: list[str | Var] | None = None,
- auto_deps: bool = True,
- interval: int | datetime.timedelta | None = None,
- backend: bool | None = None,
- **kwargs,
- ):
- """Initialize a ComputedVar.
- Args:
- fget: The getter function.
- initial_value: The initial value of the computed var.
- cache: Whether to cache the computed value.
- deps: Explicit var dependencies to track.
- auto_deps: Whether var dependencies should be auto-determined.
- interval: Interval at which the computed var should be updated.
- backend: Whether the computed var is a backend var.
- **kwargs: additional attributes to set on the instance
- Raises:
- TypeError: If the computed var dependencies are not Var instances or var names.
- UntypedComputedVarError: If the computed var is untyped.
- """
- hint = kwargs.pop("return_type", None) or get_type_hints(fget).get(
- "return", Any
- )
- if hint is Any:
- raise UntypedComputedVarError(var_name=fget.__name__)
- kwargs.setdefault("_js_expr", fget.__name__)
- kwargs.setdefault("_var_type", hint)
- Var.__init__(
- self,
- _js_expr=kwargs.pop("_js_expr"),
- _var_type=kwargs.pop("_var_type"),
- _var_data=kwargs.pop("_var_data", None),
- )
- if kwargs:
- raise TypeError(f"Unexpected keyword arguments: {tuple(kwargs)}")
- if backend is None:
- backend = fget.__name__.startswith("_")
- object.__setattr__(self, "_backend", backend)
- object.__setattr__(self, "_initial_value", initial_value)
- object.__setattr__(self, "_cache", cache)
- if isinstance(interval, int):
- interval = datetime.timedelta(seconds=interval)
- object.__setattr__(self, "_update_interval", interval)
- object.__setattr__(
- self,
- "_static_deps",
- self._calculate_static_deps(deps),
- )
- object.__setattr__(self, "_auto_deps", auto_deps)
- object.__setattr__(self, "_fget", fget)
- def _calculate_static_deps(
- self,
- deps: list[str | Var] | dict[str | None, set[str]] | None = None,
- ) -> dict[str | None, set[str]]:
- """Calculate the static dependencies of the computed var from user input or existing dependencies.
- Args:
- deps: The user input dependencies or existing dependencies.
- Returns:
- The static dependencies.
- """
- if isinstance(deps, dict):
- # Assume a dict is coming from _replace, so no special processing.
- return deps
- _static_deps = {}
- if deps is not None:
- for dep in deps:
- _static_deps = self._add_static_dep(dep, _static_deps)
- return _static_deps
- def _add_static_dep(
- self, dep: str | Var, deps: dict[str | None, set[str]] | None = None
- ) -> dict[str | None, set[str]]:
- """Add a static dependency to the computed var or existing dependency set.
- Args:
- dep: The dependency to add.
- deps: The existing dependency set.
- Returns:
- The updated dependency set.
- Raises:
- TypeError: If the computed var dependencies are not Var instances or var names.
- """
- if deps is None:
- deps = self._static_deps
- if isinstance(dep, Var):
- state_name = (
- all_var_data.state
- if (all_var_data := dep._get_all_var_data()) and all_var_data.state
- else None
- )
- if all_var_data is not None:
- var_name = all_var_data.field_name
- else:
- var_name = dep._js_expr
- deps.setdefault(state_name, set()).add(var_name)
- elif isinstance(dep, str) and dep != "":
- deps.setdefault(None, set()).add(dep)
- else:
- raise TypeError(
- "ComputedVar dependencies must be Var instances or var names (non-empty strings)."
- )
- return deps
- @override
- def _replace(
- self,
- merge_var_data: VarData | None = None,
- **kwargs: Any,
- ) -> Self:
- """Replace the attributes of the ComputedVar.
- Args:
- merge_var_data: VarData to merge into the existing VarData.
- **kwargs: Var fields to update.
- Returns:
- The new ComputedVar instance.
- Raises:
- TypeError: If kwargs contains keys that are not allowed.
- """
- if "deps" in kwargs:
- kwargs["deps"] = self._calculate_static_deps(kwargs["deps"])
- field_values = {
- "fget": kwargs.pop("fget", self._fget),
- "initial_value": kwargs.pop("initial_value", self._initial_value),
- "cache": kwargs.pop("cache", self._cache),
- "deps": kwargs.pop("deps", copy.copy(self._static_deps)),
- "auto_deps": kwargs.pop("auto_deps", self._auto_deps),
- "interval": kwargs.pop("interval", self._update_interval),
- "backend": kwargs.pop("backend", self._backend),
- "_js_expr": kwargs.pop("_js_expr", self._js_expr),
- "_var_type": kwargs.pop("_var_type", self._var_type),
- "_var_data": kwargs.pop(
- "_var_data", VarData.merge(self._var_data, merge_var_data)
- ),
- "return_type": kwargs.pop("return_type", self._var_type),
- }
- if kwargs:
- unexpected_kwargs = ", ".join(kwargs.keys())
- raise TypeError(f"Unexpected keyword arguments: {unexpected_kwargs}")
- return type(self)(**field_values)
- @property
- def _cache_attr(self) -> str:
- """Get the attribute used to cache the value on the instance.
- Returns:
- An attribute name.
- """
- return f"__cached_{self._js_expr}"
- @property
- def _last_updated_attr(self) -> str:
- """Get the attribute used to store the last updated timestamp.
- Returns:
- An attribute name.
- """
- return f"__last_updated_{self._js_expr}"
- def needs_update(self, instance: BaseState) -> bool:
- """Check if the computed var needs to be updated.
- Args:
- instance: The state instance that the computed var is attached to.
- Returns:
- True if the computed var needs to be updated, False otherwise.
- """
- if self._update_interval is None:
- return False
- last_updated = getattr(instance, self._last_updated_attr, None)
- if last_updated is None:
- return True
- return datetime.datetime.now() - last_updated > self._update_interval
- @overload
- def __get__(
- self: ComputedVar[bool],
- instance: None,
- owner: type,
- ) -> BooleanVar: ...
- @overload
- def __get__(
- self: ComputedVar[int] | ComputedVar[float],
- instance: None,
- owner: type,
- ) -> NumberVar: ...
- @overload
- def __get__(
- self: ComputedVar[str],
- instance: None,
- owner: type,
- ) -> StringVar: ...
- @overload
- def __get__(
- self: ComputedVar[MAPPING_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[MAPPING_TYPE]: ...
- @overload
- def __get__(
- self: ComputedVar[list[LIST_INSIDE]],
- instance: None,
- owner: type,
- ) -> ArrayVar[list[LIST_INSIDE]]: ...
- @overload
- def __get__(
- self: ComputedVar[tuple[LIST_INSIDE, ...]],
- instance: None,
- owner: type,
- ) -> ArrayVar[tuple[LIST_INSIDE, ...]]: ...
- @overload
- def __get__(
- self: ComputedVar[BASE_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[BASE_TYPE]: ...
- @overload
- def __get__(
- self: ComputedVar[SQLA_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[SQLA_TYPE]: ...
- if TYPE_CHECKING:
- @overload
- def __get__(
- self: ComputedVar[DATACLASS_TYPE], instance: None, owner: Any
- ) -> ObjectVar[DATACLASS_TYPE]: ...
- @overload
- def __get__(self, instance: None, owner: type) -> ComputedVar[RETURN_TYPE]: ...
- @overload
- def __get__(self, instance: BaseState, owner: type) -> RETURN_TYPE: ...
- def __get__(self, instance: BaseState | None, owner: type):
- """Get the ComputedVar value.
- If the value is already cached on the instance, return the cached value.
- Args:
- instance: the instance of the class accessing this computed var.
- owner: the class that this descriptor is attached to.
- Returns:
- The value of the var for the given instance.
- """
- if instance is None:
- state_where_defined = owner
- while self._js_expr in state_where_defined.inherited_vars:
- state_where_defined = state_where_defined.get_parent_state()
- field_name = (
- format_state_name(state_where_defined.get_full_name())
- + "."
- + self._js_expr
- )
- return dispatch(
- field_name,
- var_data=VarData.from_state(state_where_defined, self._js_expr),
- result_var_type=self._var_type,
- existing_var=self,
- )
- if not self._cache:
- value = self.fget(instance)
- else:
- # handle caching
- if not hasattr(instance, self._cache_attr) or self.needs_update(instance):
- # Set cache attr on state instance.
- setattr(instance, self._cache_attr, self.fget(instance))
- # Ensure the computed var gets serialized to redis.
- instance._was_touched = True
- # Set the last updated timestamp on the state instance.
- setattr(instance, self._last_updated_attr, datetime.datetime.now())
- value = getattr(instance, self._cache_attr)
- self._check_deprecated_return_type(instance, value)
- return value
- def _check_deprecated_return_type(self, instance: BaseState, value: Any) -> None:
- if not _isinstance(value, self._var_type, nested=1, treat_var_as_type=False):
- console.error(
- f"Computed var '{type(instance).__name__}.{self._js_expr}' must return"
- f" a value of type '{escape(str(self._var_type))}', got '{value!s}' of type {type(value)}."
- )
- def _deps(
- self,
- objclass: type[BaseState],
- obj: FunctionType | CodeType | None = None,
- ) -> dict[str, set[str]]:
- """Determine var dependencies of this ComputedVar.
- Save references to attributes accessed on "self" or other fetched states.
- Recursively called when the function makes a method call on "self" or
- define comprehensions or nested functions that may reference "self".
- Args:
- objclass: the class obj this ComputedVar is attached to.
- obj: the object to disassemble (defaults to the fget function).
- Returns:
- A dictionary mapping state names to the set of variable names
- accessed by the given obj.
- """
- from .dep_tracking import DependencyTracker
- d = {}
- if self._static_deps:
- d.update(self._static_deps)
- # None is a placeholder for the current state class.
- if None in d:
- d[objclass.get_full_name()] = d.pop(None)
- if not self._auto_deps:
- return d
- if obj is None:
- fget = self._fget
- if fget is not None:
- obj = cast(FunctionType, fget)
- else:
- return d
- try:
- return DependencyTracker(
- func=obj, state_cls=objclass, dependencies=d
- ).dependencies
- except Exception as e:
- console.warn(
- "Failed to automatically determine dependencies for computed var "
- f"{objclass.__name__}.{self._js_expr}: {e}. "
- "Provide static_deps and set auto_deps=False to suppress this warning."
- )
- return d
- def mark_dirty(self, instance: BaseState) -> None:
- """Mark this ComputedVar as dirty.
- Args:
- instance: the state instance that needs to recompute the value.
- """
- with contextlib.suppress(AttributeError):
- delattr(instance, self._cache_attr)
- def add_dependency(self, objclass: type[BaseState], dep: Var):
- """Explicitly add a dependency to the ComputedVar.
- After adding the dependency, when the `dep` changes, this computed var
- will be marked dirty.
- Args:
- objclass: The class obj this ComputedVar is attached to.
- dep: The dependency to add.
- Raises:
- VarDependencyError: If the dependency is not a Var instance with a
- state and field name
- """
- if all_var_data := dep._get_all_var_data():
- state_name = all_var_data.state
- if state_name:
- var_name = all_var_data.field_name
- if var_name:
- self._static_deps.setdefault(state_name, set()).add(var_name)
- objclass.get_root_state().get_class_substate(
- state_name
- )._var_dependencies.setdefault(var_name, set()).add(
- (objclass.get_full_name(), self._js_expr)
- )
- return
- raise VarDependencyError(
- "ComputedVar dependencies must be Var instances with a state and "
- f"field name, got {dep!r}."
- )
- def _determine_var_type(self) -> type:
- """Get the type of the var.
- Returns:
- The type of the var.
- """
- hints = get_type_hints(self._fget)
- if "return" in hints:
- return hints["return"]
- return Any # pyright: ignore [reportReturnType]
- @property
- def __class__(self) -> type:
- """Get the class of the var.
- Returns:
- The class of the var.
- """
- return FakeComputedVarBaseClass
- @property
- def fget(self) -> Callable[[BaseState], RETURN_TYPE]:
- """Get the getter function.
- Returns:
- The getter function.
- """
- return self._fget
- class DynamicRouteVar(ComputedVar[str | list[str]]):
- """A ComputedVar that represents a dynamic route."""
- pass
- async def _default_async_computed_var(_self: BaseState) -> Any:
- return None
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- init=False,
- slots=True,
- )
- class AsyncComputedVar(ComputedVar[RETURN_TYPE]):
- """A computed var that wraps a coroutinefunction."""
- _fget: Callable[[BaseState], Coroutine[None, None, RETURN_TYPE]] = (
- dataclasses.field(default=_default_async_computed_var)
- )
- @overload
- def __get__(
- self: AsyncComputedVar[bool],
- instance: None,
- owner: type,
- ) -> BooleanVar: ...
- @overload
- def __get__(
- self: AsyncComputedVar[int] | ComputedVar[float],
- instance: None,
- owner: type,
- ) -> NumberVar: ...
- @overload
- def __get__(
- self: AsyncComputedVar[str],
- instance: None,
- owner: type,
- ) -> StringVar: ...
- @overload
- def __get__(
- self: AsyncComputedVar[MAPPING_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[MAPPING_TYPE]: ...
- @overload
- def __get__(
- self: AsyncComputedVar[list[LIST_INSIDE]],
- instance: None,
- owner: type,
- ) -> ArrayVar[list[LIST_INSIDE]]: ...
- @overload
- def __get__(
- self: AsyncComputedVar[tuple[LIST_INSIDE, ...]],
- instance: None,
- owner: type,
- ) -> ArrayVar[tuple[LIST_INSIDE, ...]]: ...
- @overload
- def __get__(
- self: AsyncComputedVar[BASE_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[BASE_TYPE]: ...
- @overload
- def __get__(
- self: AsyncComputedVar[SQLA_TYPE],
- instance: None,
- owner: type,
- ) -> ObjectVar[SQLA_TYPE]: ...
- if TYPE_CHECKING:
- @overload
- def __get__(
- self: AsyncComputedVar[DATACLASS_TYPE], instance: None, owner: Any
- ) -> ObjectVar[DATACLASS_TYPE]: ...
- @overload
- def __get__(self, instance: None, owner: type) -> AsyncComputedVar[RETURN_TYPE]: ...
- @overload
- def __get__(
- self, instance: BaseState, owner: type
- ) -> Coroutine[None, None, RETURN_TYPE]: ...
- def __get__(
- self, instance: BaseState | None, owner
- ) -> Var | Coroutine[None, None, RETURN_TYPE]:
- """Get the ComputedVar value.
- If the value is already cached on the instance, return the cached value.
- Args:
- instance: the instance of the class accessing this computed var.
- owner: the class that this descriptor is attached to.
- Returns:
- The value of the var for the given instance.
- """
- if instance is None:
- return super(AsyncComputedVar, self).__get__(instance, owner)
- if not self._cache:
- async def _awaitable_result(instance: BaseState = instance) -> RETURN_TYPE:
- value = await self.fget(instance)
- self._check_deprecated_return_type(instance, value)
- return value
- return _awaitable_result()
- else:
- # handle caching
- async def _awaitable_result(instance: BaseState = instance) -> RETURN_TYPE:
- if not hasattr(instance, self._cache_attr) or self.needs_update(
- instance
- ):
- # Set cache attr on state instance.
- setattr(instance, self._cache_attr, await self.fget(instance))
- # Ensure the computed var gets serialized to redis.
- instance._was_touched = True
- # Set the last updated timestamp on the state instance.
- setattr(instance, self._last_updated_attr, datetime.datetime.now())
- value = getattr(instance, self._cache_attr)
- self._check_deprecated_return_type(instance, value)
- return value
- return _awaitable_result()
- @property
- def fget(self) -> Callable[[BaseState], Coroutine[None, None, RETURN_TYPE]]:
- """Get the getter function.
- Returns:
- The getter function.
- """
- return self._fget
- if TYPE_CHECKING:
- BASE_STATE = TypeVar("BASE_STATE", bound=BaseState)
- class _ComputedVarDecorator(Protocol):
- """A protocol for the ComputedVar decorator."""
- @overload
- def __call__(
- self,
- fget: Callable[[BASE_STATE], Coroutine[Any, Any, RETURN_TYPE]],
- ) -> AsyncComputedVar[RETURN_TYPE]: ...
- @overload
- def __call__(
- self,
- fget: Callable[[BASE_STATE], RETURN_TYPE],
- ) -> ComputedVar[RETURN_TYPE]: ...
- def __call__(
- self,
- fget: Callable[[BASE_STATE], Any],
- ) -> ComputedVar[Any]: ...
- @overload
- def computed_var(
- fget: None = None,
- initial_value: Any | types.Unset = types.Unset(),
- cache: bool = True,
- deps: list[str | Var] | None = None,
- auto_deps: bool = True,
- interval: datetime.timedelta | int | None = None,
- backend: bool | None = None,
- **kwargs,
- ) -> _ComputedVarDecorator: ...
- @overload
- def computed_var(
- fget: Callable[[BASE_STATE], Coroutine[Any, Any, RETURN_TYPE]],
- initial_value: RETURN_TYPE | types.Unset = types.Unset(),
- cache: bool = True,
- deps: list[str | Var] | None = None,
- auto_deps: bool = True,
- interval: datetime.timedelta | int | None = None,
- backend: bool | None = None,
- **kwargs,
- ) -> AsyncComputedVar[RETURN_TYPE]: ...
- @overload
- def computed_var(
- fget: Callable[[BASE_STATE], RETURN_TYPE],
- initial_value: RETURN_TYPE | types.Unset = types.Unset(),
- cache: bool = True,
- deps: list[str | Var] | None = None,
- auto_deps: bool = True,
- interval: datetime.timedelta | int | None = None,
- backend: bool | None = None,
- **kwargs,
- ) -> ComputedVar[RETURN_TYPE]: ...
- def computed_var(
- fget: Callable[[BASE_STATE], Any] | None = None,
- initial_value: Any | types.Unset = types.Unset(),
- cache: bool = True,
- deps: list[str | Var] | None = None,
- auto_deps: bool = True,
- interval: datetime.timedelta | int | None = None,
- backend: bool | None = None,
- **kwargs,
- ) -> ComputedVar | Callable[[Callable[[BASE_STATE], Any]], ComputedVar]:
- """A ComputedVar decorator with or without kwargs.
- Args:
- fget: The getter function.
- initial_value: The initial value of the computed var.
- cache: Whether to cache the computed value.
- deps: Explicit var dependencies to track.
- auto_deps: Whether var dependencies should be auto-determined.
- interval: Interval at which the computed var should be updated.
- backend: Whether the computed var is a backend var.
- **kwargs: additional attributes to set on the instance
- Returns:
- A ComputedVar instance.
- Raises:
- ValueError: If caching is disabled and an update interval is set.
- VarDependencyError: If user supplies dependencies without caching.
- ComputedVarSignatureError: If the getter function has more than one argument.
- """
- if cache is False and interval is not None:
- raise ValueError("Cannot set update interval without caching.")
- if cache is False and (deps is not None or auto_deps is False):
- raise VarDependencyError("Cannot track dependencies without caching.")
- if fget is not None:
- sign = inspect.signature(fget)
- if len(sign.parameters) != 1:
- raise ComputedVarSignatureError(fget.__name__, signature=str(sign))
- if inspect.iscoroutinefunction(fget):
- computed_var_cls = AsyncComputedVar
- else:
- computed_var_cls = ComputedVar
- return computed_var_cls(
- fget,
- initial_value=initial_value,
- cache=cache,
- deps=deps,
- auto_deps=auto_deps,
- interval=interval,
- backend=backend,
- **kwargs,
- )
- def wrapper(fget: Callable[[BASE_STATE], Any]) -> ComputedVar:
- if inspect.iscoroutinefunction(fget):
- computed_var_cls = AsyncComputedVar
- else:
- computed_var_cls = ComputedVar
- return computed_var_cls(
- fget,
- initial_value=initial_value,
- cache=cache,
- deps=deps,
- auto_deps=auto_deps,
- interval=interval,
- backend=backend,
- **kwargs,
- )
- return wrapper
- RETURN = TypeVar("RETURN")
- class CustomVarOperationReturn(Var[RETURN]):
- """Base class for custom var operations."""
- @classmethod
- def create(
- cls,
- js_expression: str,
- _var_type: type[RETURN] | None = None,
- _var_data: VarData | None = None,
- ) -> CustomVarOperationReturn[RETURN]:
- """Create a CustomVarOperation.
- Args:
- js_expression: The JavaScript expression to evaluate.
- _var_type: The type of the var.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The CustomVarOperation.
- """
- return CustomVarOperationReturn(
- _js_expr=js_expression,
- _var_type=_var_type or Any,
- _var_data=_var_data,
- )
- def var_operation_return(
- js_expression: str,
- var_type: type[RETURN] | GenericType | None = None,
- var_data: VarData | None = None,
- ) -> CustomVarOperationReturn[RETURN]:
- """Shortcut for creating a CustomVarOperationReturn.
- Args:
- js_expression: The JavaScript expression to evaluate.
- var_type: The type of the var.
- var_data: Additional hooks and imports associated with the Var.
- Returns:
- The CustomVarOperationReturn.
- """
- return CustomVarOperationReturn.create(
- js_expression,
- var_type,
- var_data,
- )
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- slots=True,
- )
- class CustomVarOperation(CachedVarOperation, Var[T]):
- """Base class for custom var operations."""
- _name: str = dataclasses.field(default="")
- _args: tuple[tuple[str, Var], ...] = dataclasses.field(default_factory=tuple)
- _return: CustomVarOperationReturn[T] = dataclasses.field(
- default_factory=lambda: CustomVarOperationReturn.create("")
- )
- @cached_property_no_lock
- def _cached_var_name(self) -> str:
- """Get the cached var name.
- Returns:
- The cached var name.
- """
- return str(self._return)
- @cached_property_no_lock
- def _cached_get_all_var_data(self) -> VarData | None:
- """Get the cached VarData.
- Returns:
- The cached VarData.
- """
- return VarData.merge(
- *(arg[1]._get_all_var_data() for arg in self._args),
- self._return._get_all_var_data(),
- self._var_data,
- )
- @classmethod
- def create(
- cls,
- name: str,
- args: tuple[tuple[str, Var], ...],
- return_var: CustomVarOperationReturn[T],
- _var_data: VarData | None = None,
- ) -> CustomVarOperation[T]:
- """Create a CustomVarOperation.
- Args:
- name: The name of the operation.
- args: The arguments to the operation.
- return_var: The return var.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The CustomVarOperation.
- """
- return CustomVarOperation(
- _js_expr="",
- _var_type=return_var._var_type,
- _var_data=_var_data,
- _name=name,
- _args=args,
- _return=return_var,
- )
- class NoneVar(Var[None], python_types=type(None)):
- """A var representing None."""
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- slots=True,
- )
- class LiteralNoneVar(LiteralVar, NoneVar):
- """A var representing None."""
- _var_value: None = None
- def json(self) -> str:
- """Serialize the var to a JSON string.
- Returns:
- The JSON string.
- """
- return "null"
- @classmethod
- def create(
- cls,
- value: None = None,
- _var_data: VarData | None = None,
- ) -> LiteralNoneVar:
- """Create a var from a value.
- Args:
- value: The value of the var. Must be None. Existed for compatibility with LiteralVar.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The var.
- """
- return LiteralNoneVar(
- _js_expr="null",
- _var_type=None,
- _var_data=_var_data,
- )
- def get_to_operation(var_subclass: type[Var]) -> type[ToOperation]:
- """Get the ToOperation class for a given Var subclass.
- Args:
- var_subclass: The Var subclass.
- Returns:
- The ToOperation class.
- Raises:
- ValueError: If the ToOperation class cannot be found.
- """
- possible_classes = [
- saved_var_subclass.to_var_subclass
- for saved_var_subclass in _var_subclasses
- if saved_var_subclass.var_subclass is var_subclass
- ]
- if not possible_classes:
- raise ValueError(f"Could not find ToOperation for {var_subclass}.")
- return possible_classes[0]
- @dataclasses.dataclass(
- eq=False,
- frozen=True,
- slots=True,
- )
- class StateOperation(CachedVarOperation, Var):
- """A var operation that accesses a field on an object."""
- _state_name: str = dataclasses.field(default="")
- _field: Var = dataclasses.field(default_factory=lambda: LiteralNoneVar.create())
- @cached_property_no_lock
- def _cached_var_name(self) -> str:
- """Get the cached var name.
- Returns:
- The cached var name.
- """
- return f"{self._state_name!s}.{self._field!s}"
- def __getattr__(self, name: str) -> Any:
- """Get an attribute of the var.
- Args:
- name: The name of the attribute.
- Returns:
- The attribute.
- """
- if name == "_js_expr":
- return self._cached_var_name
- return getattr(self._field, name)
- @classmethod
- def create(
- cls,
- state_name: str,
- field: Var,
- _var_data: VarData | None = None,
- ) -> StateOperation:
- """Create a DotOperation.
- Args:
- state_name: The name of the state.
- field: The field of the state.
- _var_data: Additional hooks and imports associated with the Var.
- Returns:
- The DotOperation.
- """
- return StateOperation(
- _js_expr="",
- _var_type=field._var_type,
- _var_data=_var_data,
- _state_name=state_name,
- _field=field,
- )
- def get_uuid_string_var() -> Var:
- """Return a Var that generates a single memoized UUID via .web/utils/state.js.
- useMemo with an empty dependency array ensures that the generated UUID is
- consistent across re-renders of the component.
- Returns:
- A Var that generates a UUID at runtime.
- """
- from reflex.utils.imports import ImportVar
- from reflex.vars import Var
- unique_uuid_var = get_unique_variable_name()
- unique_uuid_var_data = VarData(
- imports={
- f"$/{constants.Dirs.STATE_PATH}": ImportVar(tag="generateUUID"),
- "react": "useMemo",
- },
- hooks={f"const {unique_uuid_var} = useMemo(generateUUID, [])": None},
- )
- return Var(
- _js_expr=unique_uuid_var,
- _var_type=str,
- _var_data=unique_uuid_var_data,
- )
- # Set of unique variable names.
- USED_VARIABLES = set()
- def get_unique_variable_name() -> str:
- """Get a unique variable name.
- Returns:
- The unique variable name.
- """
- name = "".join([random.choice(string.ascii_lowercase) for _ in range(8)])
- if name not in USED_VARIABLES:
- USED_VARIABLES.add(name)
- return name
- return get_unique_variable_name()
- # Compile regex for finding reflex var tags.
- _decode_var_pattern_re = (
- rf"{constants.REFLEX_VAR_OPENING_TAG}(.*?){constants.REFLEX_VAR_CLOSING_TAG}"
- )
- _decode_var_pattern = re.compile(_decode_var_pattern_re, flags=re.DOTALL)
- # Defined global immutable vars.
- _global_vars: dict[int, Var] = {}
- dispatchers: dict[GenericType, Callable[[Var], Var]] = {}
- def transform(fn: Callable[[Var], Var]) -> Callable[[Var], Var]:
- """Register a function to transform a Var.
- Args:
- fn: The function to register.
- Returns:
- The decorator.
- Raises:
- TypeError: If the return type of the function is not a Var.
- TypeError: If the Var return type does not have a generic type.
- ValueError: If a function for the generic type is already registered.
- """
- return_type = fn.__annotations__["return"]
- origin = get_origin(return_type)
- if origin is not Var:
- raise TypeError(
- f"Expected return type of {fn.__name__} to be a Var, got {origin}."
- )
- generic_args = get_args(return_type)
- if not generic_args:
- raise TypeError(
- f"Expected Var return type of {fn.__name__} to have a generic type."
- )
- generic_type = get_origin(generic_args[0]) or generic_args[0]
- if generic_type in dispatchers:
- raise ValueError(f"Function for {generic_type} already registered.")
- dispatchers[generic_type] = fn
- return fn
- def generic_type_to_actual_type_map(
- generic_type: GenericType, actual_type: GenericType
- ) -> dict[TypeVar, GenericType]:
- """Map the generic type to the actual type.
- Args:
- generic_type: The generic type.
- actual_type: The actual type.
- Returns:
- The mapping of type variables to actual types.
- Raises:
- TypeError: If the generic type and actual type do not match.
- TypeError: If the number of generic arguments and actual arguments do not match.
- """
- generic_origin = get_origin(generic_type) or generic_type
- actual_origin = get_origin(actual_type) or actual_type
- if generic_origin is not actual_origin:
- if isinstance(generic_origin, TypeVar):
- return {generic_origin: actual_origin}
- raise TypeError(
- f"Type mismatch: expected {generic_origin}, got {actual_origin}."
- )
- generic_args = get_args(generic_type)
- actual_args = get_args(actual_type)
- if len(generic_args) != len(actual_args):
- raise TypeError(
- f"Number of generic arguments mismatch: expected {len(generic_args)}, got {len(actual_args)}."
- )
- # call recursively for nested generic types and merge the results
- return {
- k: v
- for generic_arg, actual_arg in zip(generic_args, actual_args, strict=True)
- for k, v in generic_type_to_actual_type_map(generic_arg, actual_arg).items()
- }
- def resolve_generic_type_with_mapping(
- generic_type: GenericType, type_mapping: dict[TypeVar, GenericType]
- ):
- """Resolve a generic type with a type mapping.
- Args:
- generic_type: The generic type.
- type_mapping: The type mapping.
- Returns:
- The resolved generic type.
- """
- if isinstance(generic_type, TypeVar):
- return type_mapping.get(generic_type, generic_type)
- generic_origin = get_origin(generic_type) or generic_type
- generic_args = get_args(generic_type)
- if not generic_args:
- return generic_type
- mapping_for_older_python = {
- list: List, # noqa: UP006
- set: Set, # noqa: UP006
- dict: Dict, # noqa: UP006
- tuple: Tuple, # noqa: UP006
- frozenset: FrozenSet, # noqa: UP006
- }
- return mapping_for_older_python.get(generic_origin, generic_origin)[
- tuple(
- resolve_generic_type_with_mapping(arg, type_mapping) for arg in generic_args
- )
- ]
- def resolve_arg_type_from_return_type(
- arg_type: GenericType, return_type: GenericType, actual_return_type: GenericType
- ) -> GenericType:
- """Resolve the argument type from the return type.
- Args:
- arg_type: The argument type.
- return_type: The return type.
- actual_return_type: The requested return type.
- Returns:
- The argument type without the generics that are resolved.
- """
- return resolve_generic_type_with_mapping(
- arg_type, generic_type_to_actual_type_map(return_type, actual_return_type)
- )
- def dispatch(
- field_name: str,
- var_data: VarData,
- result_var_type: GenericType,
- existing_var: Var | None = None,
- ) -> Var:
- """Dispatch a Var to the appropriate transformation function.
- Args:
- field_name: The name of the field.
- var_data: The VarData associated with the Var.
- result_var_type: The type of the Var.
- existing_var: The existing Var to transform. Optional.
- Returns:
- The transformed Var.
- Raises:
- TypeError: If the return type of the function is not a Var.
- TypeError: If the Var return type does not have a generic type.
- TypeError: If the first argument of the function is not a Var.
- TypeError: If the first argument of the function does not have a generic type
- """
- result_origin_var_type = get_origin(result_var_type) or result_var_type
- if result_origin_var_type in dispatchers:
- fn = dispatchers[result_origin_var_type]
- fn_first_arg_type = next(
- iter(inspect.signature(fn).parameters.values())
- ).annotation
- fn_return = inspect.signature(fn).return_annotation
- fn_return_origin = get_origin(fn_return) or fn_return
- if fn_return_origin is not Var:
- raise TypeError(
- f"Expected return type of {fn.__name__} to be a Var, got {fn_return}."
- )
- fn_return_generic_args = get_args(fn_return)
- if not fn_return_generic_args:
- raise TypeError(f"Expected generic type of {fn_return} to be a type.")
- arg_origin = get_origin(fn_first_arg_type) or fn_first_arg_type
- if arg_origin is not Var:
- raise TypeError(
- f"Expected first argument of {fn.__name__} to be a Var, got {fn_first_arg_type}."
- )
- arg_generic_args = get_args(fn_first_arg_type)
- if not arg_generic_args:
- raise TypeError(
- f"Expected generic type of {fn_first_arg_type} to be a type."
- )
- arg_type = arg_generic_args[0]
- fn_return_type = fn_return_generic_args[0]
- var = (
- Var(
- field_name,
- _var_data=var_data,
- _var_type=resolve_arg_type_from_return_type(
- arg_type, fn_return_type, result_var_type
- ),
- ).guess_type()
- if existing_var is None
- else existing_var._replace(
- _var_type=resolve_arg_type_from_return_type(
- arg_type, fn_return_type, result_var_type
- ),
- _var_data=var_data,
- _js_expr=field_name,
- ).guess_type()
- )
- return fn(var)
- if existing_var is not None:
- return existing_var._replace(
- _js_expr=field_name,
- _var_data=var_data,
- _var_type=result_var_type,
- ).guess_type()
- return Var(
- field_name,
- _var_data=var_data,
- _var_type=result_var_type,
- ).guess_type()
- if TYPE_CHECKING:
- from _typeshed import DataclassInstance
- from sqlalchemy.orm import DeclarativeBase
- SQLA_TYPE = TypeVar("SQLA_TYPE", bound=DeclarativeBase | None)
- BASE_TYPE = TypeVar("BASE_TYPE", bound=Base | None)
- DATACLASS_TYPE = TypeVar("DATACLASS_TYPE", bound=DataclassInstance | None)
- MAPPING_TYPE = TypeVar("MAPPING_TYPE", bound=Mapping | None)
- V = TypeVar("V")
- FIELD_TYPE = TypeVar("FIELD_TYPE")
- class Field(Generic[FIELD_TYPE]):
- """Shadow class for Var to allow for type hinting in the IDE."""
- def __set__(self, instance: Any, value: FIELD_TYPE):
- """Set the Var.
- Args:
- instance: The instance of the class setting the Var.
- value: The value to set the Var to.
- """
- @overload
- def __get__(self: Field[None], instance: None, owner: Any) -> NoneVar: ...
- @overload
- def __get__(
- self: Field[bool] | Field[bool | None], instance: None, owner: Any
- ) -> BooleanVar: ...
- @overload
- def __get__(
- self: Field[int] | Field[int | None],
- instance: None,
- owner: Any,
- ) -> NumberVar[int]: ...
- @overload
- def __get__(
- self: Field[float]
- | Field[int | float]
- | Field[float | None]
- | Field[int | float | None],
- instance: None,
- owner: Any,
- ) -> NumberVar: ...
- @overload
- def __get__(
- self: Field[str] | Field[str | None], instance: None, owner: Any
- ) -> StringVar: ...
- @overload
- def __get__(
- self: Field[list[V]]
- | Field[set[V]]
- | Field[list[V] | None]
- | Field[set[V] | None],
- instance: None,
- owner: Any,
- ) -> ArrayVar[Sequence[V]]: ...
- @overload
- def __get__(
- self: Field[SEQUENCE_TYPE] | Field[SEQUENCE_TYPE | None],
- instance: None,
- owner: Any,
- ) -> ArrayVar[SEQUENCE_TYPE]: ...
- @overload
- def __get__(
- self: Field[MAPPING_TYPE] | Field[MAPPING_TYPE | None],
- instance: None,
- owner: Any,
- ) -> ObjectVar[MAPPING_TYPE]: ...
- @overload
- def __get__(
- self: Field[BASE_TYPE] | Field[BASE_TYPE | None], instance: None, owner: Any
- ) -> ObjectVar[BASE_TYPE]: ...
- @overload
- def __get__(
- self: Field[SQLA_TYPE] | Field[SQLA_TYPE | None], instance: None, owner: Any
- ) -> ObjectVar[SQLA_TYPE]: ...
- if TYPE_CHECKING:
- @overload
- def __get__(
- self: Field[DATACLASS_TYPE] | Field[DATACLASS_TYPE | None],
- instance: None,
- owner: Any,
- ) -> ObjectVar[DATACLASS_TYPE]: ...
- @overload
- def __get__(self, instance: None, owner: Any) -> Var[FIELD_TYPE]: ...
- @overload
- def __get__(self, instance: Any, owner: Any) -> FIELD_TYPE: ...
- def __get__(self, instance: Any, owner: Any): # pyright: ignore [reportInconsistentOverload]
- """Get the Var.
- Args:
- instance: The instance of the class accessing the Var.
- owner: The class that the Var is attached to.
- """
- def field(value: FIELD_TYPE) -> Field[FIELD_TYPE]:
- """Create a Field with a value.
- Args:
- value: The value of the Field.
- Returns:
- The Field.
- """
- return value # pyright: ignore [reportReturnType]
|