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

have a shorter path for get_field_type if it's nice

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-12-13 07:14:33 +03:00
parent c2a39b46d7
commit d3fbf123b5
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
reflex/utils/types.py
View File

@ -308,6 +308,13 @@ def get_field_type(cls: GenericType, field_name: str) -> GenericType | None:
Returns:
The type of the field, if it exists, else None.
"""
if (
hasattr(cls, "__fields__")
and field_name in cls.__fields__
and hasattr(cls.__fields__[field_name], "annotation")
and not isinstance(cls.__fields__[field_name].annotation, (str, ForwardRef))
):
return cls.__fields__[field_name].annotation
type_hints = get_type_hints(cls)
if field_name in type_hints:
return type_hints[field_name]