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

use a constant

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-10-31 15:13:32 -07:00
parent 77fe5770bb
commit dc0c76c7af
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
reflex/constants/state.py
View File

@ -9,3 +9,7 @@ class StateManagerMode(str, Enum):
DISK = "disk" DISK = "disk"
MEMORY = "memory" MEMORY = "memory"
REDIS = "redis" REDIS = "redis"
# Used for things like console_log, etc.
FRONTEND_EVENT_STATE = "__reflex_internal_frontend_event_state"

5
reflex/event.py
View File

@ -28,6 +28,7 @@ from typing import (
from typing_extensions import ParamSpec, Protocol, get_args, get_origin from typing_extensions import ParamSpec, Protocol, get_args, get_origin
from reflex import constants from reflex import constants
from reflex.constants.state import FRONTEND_EVENT_STATE
from reflex.utils import console, format from reflex.utils import console, format
from reflex.utils.exceptions import ( from reflex.utils.exceptions import (
EventFnArgMismatch, EventFnArgMismatch,
@ -688,7 +689,7 @@ def server_side(name: str, sig: inspect.Signature, **kwargs) -> EventSpec:
fn.__qualname__ = name fn.__qualname__ = name
fn.__signature__ = sig fn.__signature__ = sig
return EventSpec( return EventSpec(
handler=EventHandler(fn=fn, state_full_name="state"), handler=EventHandler(fn=fn, state_full_name=FRONTEND_EVENT_STATE),
args=tuple( args=tuple(
( (
Var(_js_expr=k), Var(_js_expr=k),
@ -1455,7 +1456,7 @@ def get_fn_signature(fn: Callable) -> inspect.Signature:
""" """
signature = inspect.signature(fn) signature = inspect.signature(fn)
new_param = inspect.Parameter( new_param = inspect.Parameter(
"state", inspect.Parameter.POSITIONAL_OR_KEYWORD, annotation=Any FRONTEND_EVENT_STATE, inspect.Parameter.POSITIONAL_OR_KEYWORD, annotation=Any
) )
return signature.replace(parameters=(new_param, *signature.parameters.values())) return signature.replace(parameters=(new_param, *signature.parameters.values()))

3
reflex/utils/format.py
View File

@ -9,6 +9,7 @@ import re
from typing import TYPE_CHECKING, Any, Callable, List, Optional, Union from typing import TYPE_CHECKING, Any, Callable, List, Optional, Union
from reflex import constants from reflex import constants
from reflex.constants.state import FRONTEND_EVENT_STATE
from reflex.utils import exceptions from reflex.utils import exceptions
from reflex.utils.console import deprecate from reflex.utils.console import deprecate
@ -439,7 +440,7 @@ def get_event_handler_parts(handler: EventHandler) -> tuple[str, str]:
from reflex.state import State from reflex.state import State
if state_full_name == "state" and name not in State.__dict__: if state_full_name == FRONTEND_EVENT_STATE and name not in State.__dict__:
return ("", to_snake_case(handler.fn.__qualname__)) return ("", to_snake_case(handler.fn.__qualname__))
return (state_full_name, name) return (state_full_name, name)