laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ComputedVar overloads for BASE_TYPE, SQLA_TYPE, and DATACLASS_TYPE

Browse Source 
Allow typing to find __getattr__ for rx.var that returns an object-like model.
This commit is contained in:
Masen Furer 2025-02-07 17:12:46 -08:00
parent 70920a64be
commit 50d0aaaf3e
No known key found for this signature in database
GPG Key ID: B0008AD22B3B3A95
1 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
reflex/vars/base.py
View File

@ -2254,6 +2254,27 @@ class ComputedVar(Var[RETURN_TYPE]):
owner: Type, owner: Type,
) -> ArrayVar[tuple[LIST_INSIDE, ...]]: ... ) -> 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 @overload
def __get__(self, instance: None, owner: Type) -> ComputedVar[RETURN_TYPE]: ... def __get__(self, instance: None, owner: Type) -> ComputedVar[RETURN_TYPE]: ...
@ -2500,6 +2521,27 @@ class AsyncComputedVar(ComputedVar[RETURN_TYPE]):
owner: Type, owner: Type,
) -> ArrayVar[tuple[LIST_INSIDE, ...]]: ... ) -> 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 @overload
def __get__(self, instance: None, owner: Type) -> AsyncComputedVar[RETURN_TYPE]: ... def __get__(self, instance: None, owner: Type) -> AsyncComputedVar[RETURN_TYPE]: ...