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

improve typing for serializer decorator

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-11-07 01:01:22 +01:00
parent 4a6c16e9dc
commit c684444561
No known key found for this signature in database
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
reflex/utils/serializers.py
View File

@ -39,10 +39,24 @@ SERIALIZERS: dict[Type, Serializer] = {}
SERIALIZER_TYPES: dict[Type, Type] = {} SERIALIZER_TYPES: dict[Type, Type] = {}
@overload
def serializer(
fn: None = None,
to: Any = None,
) -> Serializer: ...
@overload
def serializer(
fn: Serializer,
to: None = None,
) -> functools.partial[Serializer]: ...
def serializer( def serializer(
fn: Serializer | None = None, fn: Serializer | None = None,
to: Type | None = None, to: Any = None,
) -> Serializer: ) -> Serializer | functools.partial[Serializer]:
"""Decorator to add a serializer for a given type. """Decorator to add a serializer for a given type.
Args: Args: