reflex 0.8.15a0__py3-none-any.whl → 0.8.15a1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of reflex might be problematic. Click here for more details.
- reflex/__init__.py +11 -6
- reflex/__init__.pyi +9 -2
- reflex/base.py +8 -11
- reflex/components/field.py +3 -1
- reflex/components/markdown/markdown.py +101 -27
- reflex/constants/base.py +5 -0
- reflex/constants/installer.py +2 -2
- reflex/event.py +3 -0
- reflex/istate/manager/__init__.py +120 -0
- reflex/istate/manager/disk.py +210 -0
- reflex/istate/manager/memory.py +76 -0
- reflex/istate/{manager.py → manager/redis.py} +5 -372
- reflex/model.py +5 -1
- reflex/state.py +14 -9
- reflex/testing.py +4 -8
- reflex/utils/compat.py +49 -1
- reflex/utils/misc.py +2 -1
- reflex/utils/monitoring.py +1 -2
- reflex/utils/prerequisites.py +17 -3
- reflex/utils/processes.py +3 -1
- reflex/utils/redir.py +21 -37
- reflex/utils/templates.py +4 -4
- reflex/utils/types.py +1 -0
- reflex/vars/base.py +106 -25
- reflex/vars/color.py +28 -8
- reflex/vars/datetime.py +6 -2
- reflex/vars/dep_tracking.py +2 -2
- reflex/vars/number.py +26 -0
- reflex/vars/object.py +23 -6
- reflex/vars/sequence.py +32 -1
- {reflex-0.8.15a0.dist-info → reflex-0.8.15a1.dist-info}/METADATA +4 -3
- {reflex-0.8.15a0.dist-info → reflex-0.8.15a1.dist-info}/RECORD +35 -32
- {reflex-0.8.15a0.dist-info → reflex-0.8.15a1.dist-info}/WHEEL +0 -0
- {reflex-0.8.15a0.dist-info → reflex-0.8.15a1.dist-info}/entry_points.txt +0 -0
- {reflex-0.8.15a0.dist-info → reflex-0.8.15a1.dist-info}/licenses/LICENSE +0 -0
reflex/__init__.py
CHANGED
|
@@ -84,13 +84,18 @@ In the example above, you will be able to do `rx.list`
|
|
|
84
84
|
|
|
85
85
|
from __future__ import annotations
|
|
86
86
|
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
87
|
+
import sys
|
|
88
|
+
|
|
89
|
+
from reflex.utils import lazy_loader
|
|
90
|
+
|
|
91
|
+
if sys.version_info < (3, 11):
|
|
92
|
+
from reflex.utils import console
|
|
91
93
|
|
|
92
|
-
|
|
93
|
-
|
|
94
|
+
console.warn(
|
|
95
|
+
"Reflex support for Python 3.10 is deprecated and will be removed in a future release. Please upgrade to Python 3.11 or higher for continued support."
|
|
96
|
+
)
|
|
97
|
+
del console
|
|
98
|
+
del sys
|
|
94
99
|
|
|
95
100
|
RADIX_THEMES_MAPPING: dict = {
|
|
96
101
|
"components.radix.themes.base": ["color_mode", "theme", "theme_panel"],
|
reflex/__init__.pyi
CHANGED
|
@@ -3,7 +3,7 @@
|
|
|
3
3
|
# This file was generated by `reflex/utils/pyi_generator.py`!
|
|
4
4
|
# ------------------------------------------------------
|
|
5
5
|
|
|
6
|
-
|
|
6
|
+
import sys
|
|
7
7
|
|
|
8
8
|
from . import (
|
|
9
9
|
admin,
|
|
@@ -160,7 +160,14 @@ from .utils.misc import run_in_thread
|
|
|
160
160
|
from .utils.serializers import serializer
|
|
161
161
|
from .vars import Field, Var, field
|
|
162
162
|
|
|
163
|
-
|
|
163
|
+
if sys.version_info < (3, 11):
|
|
164
|
+
from reflex.utils import console
|
|
165
|
+
|
|
166
|
+
console.warn(
|
|
167
|
+
"Reflex support for Python 3.10 is deprecated and will be removed in a future release. Please upgrade to Python 3.11 or higher for continued support."
|
|
168
|
+
)
|
|
169
|
+
del console
|
|
170
|
+
del sys
|
|
164
171
|
RADIX_THEMES_MAPPING: dict
|
|
165
172
|
RADIX_THEMES_COMPONENTS_MAPPING: dict
|
|
166
173
|
RADIX_THEMES_LAYOUT_MAPPING: dict
|
reflex/base.py
CHANGED
|
@@ -5,7 +5,9 @@ from importlib.util import find_spec
|
|
|
5
5
|
if find_spec("pydantic") and find_spec("pydantic.v1"):
|
|
6
6
|
from pydantic.v1 import BaseModel
|
|
7
7
|
|
|
8
|
-
|
|
8
|
+
from reflex.utils.compat import ModelMetaclassLazyAnnotations
|
|
9
|
+
|
|
10
|
+
class Base(BaseModel, metaclass=ModelMetaclassLazyAnnotations):
|
|
9
11
|
"""The base class subclassed by all Reflex classes.
|
|
10
12
|
|
|
11
13
|
This class wraps Pydantic and provides common methods such as
|
|
@@ -22,22 +24,17 @@ if find_spec("pydantic") and find_spec("pydantic.v1"):
|
|
|
22
24
|
use_enum_values = True
|
|
23
25
|
extra = "allow"
|
|
24
26
|
|
|
25
|
-
def
|
|
26
|
-
"""
|
|
27
|
-
|
|
28
|
-
Args:
|
|
29
|
-
*args: Positional arguments.
|
|
30
|
-
**kwargs: Keyword arguments.
|
|
31
|
-
"""
|
|
27
|
+
def __init_subclass__(cls):
|
|
28
|
+
"""Warn that rx.Base is deprecated."""
|
|
32
29
|
from reflex.utils import console
|
|
33
30
|
|
|
34
31
|
console.deprecate(
|
|
35
32
|
feature_name="rx.Base",
|
|
36
|
-
reason="You can subclass from `pydantic.BaseModel` directly instead or use dataclasses if possible.",
|
|
37
|
-
deprecation_version="0.8.
|
|
33
|
+
reason=f"{cls!r} is subclassing rx.Base. You can subclass from `pydantic.BaseModel` directly instead or use dataclasses if possible.",
|
|
34
|
+
deprecation_version="0.8.15",
|
|
38
35
|
removal_version="0.9.0",
|
|
39
36
|
)
|
|
40
|
-
super().
|
|
37
|
+
super().__init_subclass__()
|
|
41
38
|
|
|
42
39
|
def json(self) -> str:
|
|
43
40
|
"""Convert the object to a json string.
|
reflex/components/field.py
CHANGED
|
@@ -7,6 +7,7 @@ from dataclasses import _MISSING_TYPE, MISSING
|
|
|
7
7
|
from typing import Annotated, Any, Generic, TypeVar, get_origin
|
|
8
8
|
|
|
9
9
|
from reflex.utils import types
|
|
10
|
+
from reflex.utils.compat import annotations_from_namespace
|
|
10
11
|
|
|
11
12
|
FIELD_TYPE = TypeVar("FIELD_TYPE")
|
|
12
13
|
|
|
@@ -117,7 +118,8 @@ class FieldBasedMeta(type):
|
|
|
117
118
|
cls, namespace: dict[str, Any], name: str
|
|
118
119
|
) -> dict[str, Any]:
|
|
119
120
|
return types.resolve_annotations(
|
|
120
|
-
namespace
|
|
121
|
+
annotations_from_namespace(namespace),
|
|
122
|
+
namespace["__module__"],
|
|
121
123
|
)
|
|
122
124
|
|
|
123
125
|
@classmethod
|
|
@@ -38,6 +38,84 @@ _REHYPE_PLUGINS = LiteralVar.create([_REHYPE_KATEX, _REHYPE_RAW])
|
|
|
38
38
|
NO_PROPS_TAGS = ("ul", "ol", "li")
|
|
39
39
|
|
|
40
40
|
|
|
41
|
+
def _h1(value: object):
|
|
42
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
43
|
+
|
|
44
|
+
return Heading.create(value, as_="h1", size="6", margin_y="0.5em")
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
def _h2(value: object):
|
|
48
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
49
|
+
|
|
50
|
+
return Heading.create(value, as_="h2", size="5", margin_y="0.5em")
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
def _h3(value: object):
|
|
54
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
55
|
+
|
|
56
|
+
return Heading.create(value, as_="h3", size="4", margin_y="0.5em")
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
def _h4(value: object):
|
|
60
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
61
|
+
|
|
62
|
+
return Heading.create(value, as_="h4", size="3", margin_y="0.5em")
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
def _h5(value: object):
|
|
66
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
67
|
+
|
|
68
|
+
return Heading.create(value, as_="h5", size="2", margin_y="0.5em")
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
def _h6(value: object):
|
|
72
|
+
from reflex.components.radix.themes.typography.heading import Heading
|
|
73
|
+
|
|
74
|
+
return Heading.create(value, as_="h6", size="1", margin_y="0.5em")
|
|
75
|
+
|
|
76
|
+
|
|
77
|
+
def _p(value: object):
|
|
78
|
+
from reflex.components.radix.themes.typography.text import Text
|
|
79
|
+
|
|
80
|
+
return Text.create(value, margin_y="1em")
|
|
81
|
+
|
|
82
|
+
|
|
83
|
+
def _ul(value: object):
|
|
84
|
+
from reflex.components.radix.themes.layout.list import UnorderedList
|
|
85
|
+
|
|
86
|
+
return UnorderedList.create(value, margin_y="1em")
|
|
87
|
+
|
|
88
|
+
|
|
89
|
+
def _ol(value: object):
|
|
90
|
+
from reflex.components.radix.themes.layout.list import OrderedList
|
|
91
|
+
|
|
92
|
+
return OrderedList.create(value, margin_y="1em")
|
|
93
|
+
|
|
94
|
+
|
|
95
|
+
def _li(value: object):
|
|
96
|
+
from reflex.components.radix.themes.layout.list import ListItem
|
|
97
|
+
|
|
98
|
+
return ListItem.create(value, margin_y="0.5em")
|
|
99
|
+
|
|
100
|
+
|
|
101
|
+
def _a(value: object):
|
|
102
|
+
from reflex.components.radix.themes.typography.link import Link
|
|
103
|
+
|
|
104
|
+
return Link.create(value)
|
|
105
|
+
|
|
106
|
+
|
|
107
|
+
def _code(value: object):
|
|
108
|
+
from reflex.components.radix.themes.typography.code import Code
|
|
109
|
+
|
|
110
|
+
return Code.create(value)
|
|
111
|
+
|
|
112
|
+
|
|
113
|
+
def _codeblock(value: object, **props):
|
|
114
|
+
from reflex.components.datadisplay.code import CodeBlock
|
|
115
|
+
|
|
116
|
+
return CodeBlock.create(value, margin_y="1em", wrap_long_lines=True, **props)
|
|
117
|
+
|
|
118
|
+
|
|
41
119
|
# Component Mapping
|
|
42
120
|
@lru_cache
|
|
43
121
|
def get_base_component_map() -> dict[str, Callable]:
|
|
@@ -46,33 +124,20 @@ def get_base_component_map() -> dict[str, Callable]:
|
|
|
46
124
|
Returns:
|
|
47
125
|
The base component map.
|
|
48
126
|
"""
|
|
49
|
-
from reflex.components.datadisplay.code import CodeBlock
|
|
50
|
-
from reflex.components.radix.themes.layout.list import (
|
|
51
|
-
ListItem,
|
|
52
|
-
OrderedList,
|
|
53
|
-
UnorderedList,
|
|
54
|
-
)
|
|
55
|
-
from reflex.components.radix.themes.typography.code import Code
|
|
56
|
-
from reflex.components.radix.themes.typography.heading import Heading
|
|
57
|
-
from reflex.components.radix.themes.typography.link import Link
|
|
58
|
-
from reflex.components.radix.themes.typography.text import Text
|
|
59
|
-
|
|
60
127
|
return {
|
|
61
|
-
"h1":
|
|
62
|
-
"h2":
|
|
63
|
-
"h3":
|
|
64
|
-
"h4":
|
|
65
|
-
"h5":
|
|
66
|
-
"h6":
|
|
67
|
-
"p":
|
|
68
|
-
"ul":
|
|
69
|
-
"ol":
|
|
70
|
-
"li":
|
|
71
|
-
"a":
|
|
72
|
-
"code":
|
|
73
|
-
"codeblock":
|
|
74
|
-
value, margin_y="1em", wrap_long_lines=True, **props
|
|
75
|
-
),
|
|
128
|
+
"h1": _h1,
|
|
129
|
+
"h2": _h2,
|
|
130
|
+
"h3": _h3,
|
|
131
|
+
"h4": _h4,
|
|
132
|
+
"h5": _h5,
|
|
133
|
+
"h6": _h6,
|
|
134
|
+
"p": _p,
|
|
135
|
+
"ul": _ul,
|
|
136
|
+
"ol": _ol,
|
|
137
|
+
"li": _li,
|
|
138
|
+
"a": _a,
|
|
139
|
+
"code": _code,
|
|
140
|
+
"codeblock": _codeblock,
|
|
76
141
|
}
|
|
77
142
|
|
|
78
143
|
|
|
@@ -413,7 +478,16 @@ let {_LANGUAGE!s} = match ? match[1] : '';
|
|
|
413
478
|
@staticmethod
|
|
414
479
|
def _component_map_hash(component_map: dict) -> str:
|
|
415
480
|
inp = str(
|
|
416
|
-
{
|
|
481
|
+
{
|
|
482
|
+
tag: (
|
|
483
|
+
f"{component.__module__}.{component.__qualname__}"
|
|
484
|
+
if (
|
|
485
|
+
"<" not in component.__name__
|
|
486
|
+
) # simple way to check against lambdas
|
|
487
|
+
else component(_MOCK_ARG)
|
|
488
|
+
)
|
|
489
|
+
for tag, component in component_map.items()
|
|
490
|
+
}
|
|
417
491
|
).encode()
|
|
418
492
|
return md5(inp).hexdigest()
|
|
419
493
|
|
reflex/constants/base.py
CHANGED
|
@@ -134,6 +134,11 @@ class Templates(SimpleNamespace):
|
|
|
134
134
|
# The reflex.build frontend host
|
|
135
135
|
REFLEX_BUILD_FRONTEND = "https://build.reflex.dev"
|
|
136
136
|
|
|
137
|
+
# The reflex.build frontend with referrer
|
|
138
|
+
REFLEX_BUILD_FRONTEND_WITH_REFERRER = (
|
|
139
|
+
f"{REFLEX_BUILD_FRONTEND}/?utm_source=reflex_cli"
|
|
140
|
+
)
|
|
141
|
+
|
|
137
142
|
class Dirs(SimpleNamespace):
|
|
138
143
|
"""Folders used by the template system of Reflex."""
|
|
139
144
|
|
reflex/constants/installer.py
CHANGED
|
@@ -143,11 +143,11 @@ class PackageJson(SimpleNamespace):
|
|
|
143
143
|
"postcss-import": "16.1.1",
|
|
144
144
|
"@react-router/dev": _react_router_version,
|
|
145
145
|
"@react-router/fs-routes": _react_router_version,
|
|
146
|
-
"vite": "npm:rolldown-vite@7.1.
|
|
146
|
+
"vite": "npm:rolldown-vite@7.1.16",
|
|
147
147
|
}
|
|
148
148
|
OVERRIDES = {
|
|
149
149
|
# This should always match the `react` version in DEPENDENCIES for recharts compatibility.
|
|
150
150
|
"react-is": _react_version,
|
|
151
151
|
"cookie": "1.0.2",
|
|
152
|
-
"vite": "npm:rolldown-vite@7.1.
|
|
152
|
+
"vite": "npm:rolldown-vite@7.1.16",
|
|
153
153
|
}
|
reflex/event.py
CHANGED
|
@@ -1866,6 +1866,9 @@ def fix_events(
|
|
|
1866
1866
|
# Fix the events created by the handler.
|
|
1867
1867
|
out = []
|
|
1868
1868
|
for e in events:
|
|
1869
|
+
if callable(e) and getattr(e, "__name__", "") == "<lambda>":
|
|
1870
|
+
# A lambda was returned, assume the user wants to call it with no args.
|
|
1871
|
+
e = e()
|
|
1869
1872
|
if isinstance(e, Event):
|
|
1870
1873
|
# If the event is already an event, append it to the list.
|
|
1871
1874
|
out.append(e)
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"""State manager for managing client states."""
|
|
2
|
+
|
|
3
|
+
import contextlib
|
|
4
|
+
import dataclasses
|
|
5
|
+
from abc import ABC, abstractmethod
|
|
6
|
+
from collections.abc import AsyncIterator
|
|
7
|
+
|
|
8
|
+
from reflex import constants
|
|
9
|
+
from reflex.config import get_config
|
|
10
|
+
from reflex.state import BaseState
|
|
11
|
+
from reflex.utils import console, prerequisites
|
|
12
|
+
from reflex.utils.exceptions import InvalidStateManagerModeError
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
@dataclasses.dataclass
|
|
16
|
+
class StateManager(ABC):
|
|
17
|
+
"""A class to manage many client states."""
|
|
18
|
+
|
|
19
|
+
# The state class to use.
|
|
20
|
+
state: type[BaseState]
|
|
21
|
+
|
|
22
|
+
@classmethod
|
|
23
|
+
def create(cls, state: type[BaseState]):
|
|
24
|
+
"""Create a new state manager.
|
|
25
|
+
|
|
26
|
+
Args:
|
|
27
|
+
state: The state class to use.
|
|
28
|
+
|
|
29
|
+
Raises:
|
|
30
|
+
InvalidStateManagerModeError: If the state manager mode is invalid.
|
|
31
|
+
|
|
32
|
+
Returns:
|
|
33
|
+
The state manager (either disk, memory or redis).
|
|
34
|
+
"""
|
|
35
|
+
config = get_config()
|
|
36
|
+
if prerequisites.parse_redis_url() is not None:
|
|
37
|
+
config.state_manager_mode = constants.StateManagerMode.REDIS
|
|
38
|
+
if config.state_manager_mode == constants.StateManagerMode.MEMORY:
|
|
39
|
+
from reflex.istate.manager.memory import StateManagerMemory
|
|
40
|
+
|
|
41
|
+
return StateManagerMemory(state=state)
|
|
42
|
+
if config.state_manager_mode == constants.StateManagerMode.DISK:
|
|
43
|
+
from reflex.istate.manager.disk import StateManagerDisk
|
|
44
|
+
|
|
45
|
+
return StateManagerDisk(state=state)
|
|
46
|
+
if config.state_manager_mode == constants.StateManagerMode.REDIS:
|
|
47
|
+
redis = prerequisites.get_redis()
|
|
48
|
+
if redis is not None:
|
|
49
|
+
from reflex.istate.manager.redis import StateManagerRedis
|
|
50
|
+
|
|
51
|
+
# make sure expiration values are obtained only from the config object on creation
|
|
52
|
+
return StateManagerRedis(
|
|
53
|
+
state=state,
|
|
54
|
+
redis=redis,
|
|
55
|
+
token_expiration=config.redis_token_expiration,
|
|
56
|
+
lock_expiration=config.redis_lock_expiration,
|
|
57
|
+
lock_warning_threshold=config.redis_lock_warning_threshold,
|
|
58
|
+
)
|
|
59
|
+
msg = f"Expected one of: DISK, MEMORY, REDIS, got {config.state_manager_mode}"
|
|
60
|
+
raise InvalidStateManagerModeError(msg)
|
|
61
|
+
|
|
62
|
+
@abstractmethod
|
|
63
|
+
async def get_state(self, token: str) -> BaseState:
|
|
64
|
+
"""Get the state for a token.
|
|
65
|
+
|
|
66
|
+
Args:
|
|
67
|
+
token: The token to get the state for.
|
|
68
|
+
|
|
69
|
+
Returns:
|
|
70
|
+
The state for the token.
|
|
71
|
+
"""
|
|
72
|
+
|
|
73
|
+
@abstractmethod
|
|
74
|
+
async def set_state(self, token: str, state: BaseState):
|
|
75
|
+
"""Set the state for a token.
|
|
76
|
+
|
|
77
|
+
Args:
|
|
78
|
+
token: The token to set the state for.
|
|
79
|
+
state: The state to set.
|
|
80
|
+
"""
|
|
81
|
+
|
|
82
|
+
@abstractmethod
|
|
83
|
+
@contextlib.asynccontextmanager
|
|
84
|
+
async def modify_state(self, token: str) -> AsyncIterator[BaseState]:
|
|
85
|
+
"""Modify the state for a token while holding exclusive lock.
|
|
86
|
+
|
|
87
|
+
Args:
|
|
88
|
+
token: The token to modify the state for.
|
|
89
|
+
|
|
90
|
+
Yields:
|
|
91
|
+
The state for the token.
|
|
92
|
+
"""
|
|
93
|
+
yield self.state()
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
def _default_token_expiration() -> int:
|
|
97
|
+
"""Get the default token expiration time.
|
|
98
|
+
|
|
99
|
+
Returns:
|
|
100
|
+
The default token expiration time.
|
|
101
|
+
"""
|
|
102
|
+
return get_config().redis_token_expiration
|
|
103
|
+
|
|
104
|
+
|
|
105
|
+
def reset_disk_state_manager():
|
|
106
|
+
"""Reset the disk state manager."""
|
|
107
|
+
console.debug("Resetting disk state manager.")
|
|
108
|
+
states_directory = prerequisites.get_states_dir()
|
|
109
|
+
if states_directory.exists():
|
|
110
|
+
for path in states_directory.iterdir():
|
|
111
|
+
path.unlink()
|
|
112
|
+
|
|
113
|
+
|
|
114
|
+
def get_state_manager() -> StateManager:
|
|
115
|
+
"""Get the state manager for the app that is currently running.
|
|
116
|
+
|
|
117
|
+
Returns:
|
|
118
|
+
The state manager.
|
|
119
|
+
"""
|
|
120
|
+
return prerequisites.get_and_validate_app().app.state_manager
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
"""A state manager that stores states on disk."""
|
|
2
|
+
|
|
3
|
+
import asyncio
|
|
4
|
+
import contextlib
|
|
5
|
+
import dataclasses
|
|
6
|
+
import functools
|
|
7
|
+
from collections.abc import AsyncIterator
|
|
8
|
+
from hashlib import md5
|
|
9
|
+
from pathlib import Path
|
|
10
|
+
|
|
11
|
+
from typing_extensions import override
|
|
12
|
+
|
|
13
|
+
from reflex.istate.manager import StateManager, _default_token_expiration
|
|
14
|
+
from reflex.state import BaseState, _split_substate_key, _substate_key
|
|
15
|
+
from reflex.utils import path_ops, prerequisites
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
@dataclasses.dataclass
|
|
19
|
+
class StateManagerDisk(StateManager):
|
|
20
|
+
"""A state manager that stores states on disk."""
|
|
21
|
+
|
|
22
|
+
# The mapping of client ids to states.
|
|
23
|
+
states: dict[str, BaseState] = dataclasses.field(default_factory=dict)
|
|
24
|
+
|
|
25
|
+
# The mutex ensures the dict of mutexes is updated exclusively
|
|
26
|
+
_state_manager_lock: asyncio.Lock = dataclasses.field(default=asyncio.Lock())
|
|
27
|
+
|
|
28
|
+
# The dict of mutexes for each client
|
|
29
|
+
_states_locks: dict[str, asyncio.Lock] = dataclasses.field(
|
|
30
|
+
default_factory=dict,
|
|
31
|
+
init=False,
|
|
32
|
+
)
|
|
33
|
+
|
|
34
|
+
# The token expiration time (s).
|
|
35
|
+
token_expiration: int = dataclasses.field(default_factory=_default_token_expiration)
|
|
36
|
+
|
|
37
|
+
def __post_init__(self):
|
|
38
|
+
"""Create a new state manager."""
|
|
39
|
+
path_ops.mkdir(self.states_directory)
|
|
40
|
+
|
|
41
|
+
self._purge_expired_states()
|
|
42
|
+
|
|
43
|
+
@functools.cached_property
|
|
44
|
+
def states_directory(self) -> Path:
|
|
45
|
+
"""Get the states directory.
|
|
46
|
+
|
|
47
|
+
Returns:
|
|
48
|
+
The states directory.
|
|
49
|
+
"""
|
|
50
|
+
return prerequisites.get_states_dir()
|
|
51
|
+
|
|
52
|
+
def _purge_expired_states(self):
|
|
53
|
+
"""Purge expired states from the disk."""
|
|
54
|
+
import time
|
|
55
|
+
|
|
56
|
+
for path in path_ops.ls(self.states_directory):
|
|
57
|
+
# check path is a pickle file
|
|
58
|
+
if path.suffix != ".pkl":
|
|
59
|
+
continue
|
|
60
|
+
|
|
61
|
+
# load last edited field from file
|
|
62
|
+
last_edited = path.stat().st_mtime
|
|
63
|
+
|
|
64
|
+
# check if the file is older than the token expiration time
|
|
65
|
+
if time.time() - last_edited > self.token_expiration:
|
|
66
|
+
# remove the file
|
|
67
|
+
path.unlink()
|
|
68
|
+
|
|
69
|
+
def token_path(self, token: str) -> Path:
|
|
70
|
+
"""Get the path for a token.
|
|
71
|
+
|
|
72
|
+
Args:
|
|
73
|
+
token: The token to get the path for.
|
|
74
|
+
|
|
75
|
+
Returns:
|
|
76
|
+
The path for the token.
|
|
77
|
+
"""
|
|
78
|
+
return (
|
|
79
|
+
self.states_directory / f"{md5(token.encode()).hexdigest()}.pkl"
|
|
80
|
+
).absolute()
|
|
81
|
+
|
|
82
|
+
async def load_state(self, token: str) -> BaseState | None:
|
|
83
|
+
"""Load a state object based on the provided token.
|
|
84
|
+
|
|
85
|
+
Args:
|
|
86
|
+
token: The token used to identify the state object.
|
|
87
|
+
|
|
88
|
+
Returns:
|
|
89
|
+
The loaded state object or None.
|
|
90
|
+
"""
|
|
91
|
+
token_path = self.token_path(token)
|
|
92
|
+
|
|
93
|
+
if token_path.exists():
|
|
94
|
+
try:
|
|
95
|
+
with token_path.open(mode="rb") as file:
|
|
96
|
+
return BaseState._deserialize(fp=file)
|
|
97
|
+
except Exception:
|
|
98
|
+
pass
|
|
99
|
+
return None
|
|
100
|
+
|
|
101
|
+
async def populate_substates(
|
|
102
|
+
self, client_token: str, state: BaseState, root_state: BaseState
|
|
103
|
+
):
|
|
104
|
+
"""Populate the substates of a state object.
|
|
105
|
+
|
|
106
|
+
Args:
|
|
107
|
+
client_token: The client token.
|
|
108
|
+
state: The state object to populate.
|
|
109
|
+
root_state: The root state object.
|
|
110
|
+
"""
|
|
111
|
+
for substate in state.get_substates():
|
|
112
|
+
substate_token = _substate_key(client_token, substate)
|
|
113
|
+
|
|
114
|
+
fresh_instance = await root_state.get_state(substate)
|
|
115
|
+
instance = await self.load_state(substate_token)
|
|
116
|
+
if instance is not None:
|
|
117
|
+
# Ensure all substates exist, even if they weren't serialized previously.
|
|
118
|
+
instance.substates = fresh_instance.substates
|
|
119
|
+
else:
|
|
120
|
+
instance = fresh_instance
|
|
121
|
+
state.substates[substate.get_name()] = instance
|
|
122
|
+
instance.parent_state = state
|
|
123
|
+
|
|
124
|
+
await self.populate_substates(client_token, instance, root_state)
|
|
125
|
+
|
|
126
|
+
@override
|
|
127
|
+
async def get_state(
|
|
128
|
+
self,
|
|
129
|
+
token: str,
|
|
130
|
+
) -> BaseState:
|
|
131
|
+
"""Get the state for a token.
|
|
132
|
+
|
|
133
|
+
Args:
|
|
134
|
+
token: The token to get the state for.
|
|
135
|
+
|
|
136
|
+
Returns:
|
|
137
|
+
The state for the token.
|
|
138
|
+
"""
|
|
139
|
+
client_token = _split_substate_key(token)[0]
|
|
140
|
+
root_state = self.states.get(client_token)
|
|
141
|
+
if root_state is not None:
|
|
142
|
+
# Retrieved state from memory.
|
|
143
|
+
return root_state
|
|
144
|
+
|
|
145
|
+
# Deserialize root state from disk.
|
|
146
|
+
root_state = await self.load_state(_substate_key(client_token, self.state))
|
|
147
|
+
# Create a new root state tree with all substates instantiated.
|
|
148
|
+
fresh_root_state = self.state(_reflex_internal_init=True)
|
|
149
|
+
if root_state is None:
|
|
150
|
+
root_state = fresh_root_state
|
|
151
|
+
else:
|
|
152
|
+
# Ensure all substates exist, even if they were not serialized previously.
|
|
153
|
+
root_state.substates = fresh_root_state.substates
|
|
154
|
+
self.states[client_token] = root_state
|
|
155
|
+
await self.populate_substates(client_token, root_state, root_state)
|
|
156
|
+
return root_state
|
|
157
|
+
|
|
158
|
+
async def set_state_for_substate(self, client_token: str, substate: BaseState):
|
|
159
|
+
"""Set the state for a substate.
|
|
160
|
+
|
|
161
|
+
Args:
|
|
162
|
+
client_token: The client token.
|
|
163
|
+
substate: The substate to set.
|
|
164
|
+
"""
|
|
165
|
+
substate_token = _substate_key(client_token, substate)
|
|
166
|
+
|
|
167
|
+
if substate._get_was_touched():
|
|
168
|
+
substate._was_touched = False # Reset the touched flag after serializing.
|
|
169
|
+
pickle_state = substate._serialize()
|
|
170
|
+
if pickle_state:
|
|
171
|
+
if not self.states_directory.exists():
|
|
172
|
+
self.states_directory.mkdir(parents=True, exist_ok=True)
|
|
173
|
+
self.token_path(substate_token).write_bytes(pickle_state)
|
|
174
|
+
|
|
175
|
+
for substate_substate in substate.substates.values():
|
|
176
|
+
await self.set_state_for_substate(client_token, substate_substate)
|
|
177
|
+
|
|
178
|
+
@override
|
|
179
|
+
async def set_state(self, token: str, state: BaseState):
|
|
180
|
+
"""Set the state for a token.
|
|
181
|
+
|
|
182
|
+
Args:
|
|
183
|
+
token: The token to set the state for.
|
|
184
|
+
state: The state to set.
|
|
185
|
+
"""
|
|
186
|
+
client_token, _ = _split_substate_key(token)
|
|
187
|
+
await self.set_state_for_substate(client_token, state)
|
|
188
|
+
|
|
189
|
+
@override
|
|
190
|
+
@contextlib.asynccontextmanager
|
|
191
|
+
async def modify_state(self, token: str) -> AsyncIterator[BaseState]:
|
|
192
|
+
"""Modify the state for a token while holding exclusive lock.
|
|
193
|
+
|
|
194
|
+
Args:
|
|
195
|
+
token: The token to modify the state for.
|
|
196
|
+
|
|
197
|
+
Yields:
|
|
198
|
+
The state for the token.
|
|
199
|
+
"""
|
|
200
|
+
# Disk state manager ignores the substate suffix and always returns the top-level state.
|
|
201
|
+
client_token, _ = _split_substate_key(token)
|
|
202
|
+
if client_token not in self._states_locks:
|
|
203
|
+
async with self._state_manager_lock:
|
|
204
|
+
if client_token not in self._states_locks:
|
|
205
|
+
self._states_locks[client_token] = asyncio.Lock()
|
|
206
|
+
|
|
207
|
+
async with self._states_locks[client_token]:
|
|
208
|
+
state = await self.get_state(token)
|
|
209
|
+
yield state
|
|
210
|
+
await self.set_state(token, state)
|