reflex 0.8.6a0__py3-none-any.whl → 0.8.6a1__py3-none-any.whl

reflex/app.py

@@ -604,6 +604,11 @@ class App(MiddlewareMixin, LifespanMixin):
 
         self._compile(prerender_routes=is_prod_mode())
 
+        config = get_config()
+
+        for plugin in config.plugins:
+            plugin.post_compile(app=self)
+
         # We will not be making more vars, so we can clear the global cache to free up memory.
         GLOBAL_CACHE.clear()
 
@@ -1926,6 +1931,13 @@ def upload(app: App):
                 )
             )
 
+        for file in files:
+            if not isinstance(file, StarletteUploadFile):
+                raise UploadValueError(
+                    "Uploaded file is not an UploadFile." + str(file)
+                )
+            await file.close()
+
         event = Event(
             token=token,
             name=handler,

reflex/components/base/error_boundary.py

@@ -136,6 +136,8 @@ class ErrorBoundary(Component):
                         height="100%",
                         width="100%",
                         position="absolute",
+                        background_color="#fff",
+                        color="#000",
                         display="flex",
                         align_items="center",
                         justify_content="center",

reflex/istate/manager.py

@@ -143,6 +143,8 @@ class StateManagerMemory(StateManager):
             token: The token to set the state for.
             state: The state to set.
         """
+        token = _split_substate_key(token)[0]
+        self.states[token] = state
 
     @override
     @contextlib.asynccontextmanager
@@ -165,7 +167,6 @@ class StateManagerMemory(StateManager):
         async with self._states_locks[token]:
             state = await self.get_state(token)
             yield state
-            await self.set_state(token, state)
 
 
 def _default_token_expiration() -> int:

reflex/plugins/__init__.py

@@ -1,5 +1,6 @@
 """Reflex Plugin System."""
 
+from ._screenshot import ScreenshotPlugin as _ScreenshotPlugin
 from .base import CommonContext, Plugin, PreCompileContext
 from .sitemap import SitemapPlugin
 from .tailwind_v3 import TailwindV3Plugin
@@ -12,4 +13,5 @@ __all__ = [
     "SitemapPlugin",
     "TailwindV3Plugin",
     "TailwindV4Plugin",
+    "_ScreenshotPlugin",
 ]

reflex/plugins/_screenshot.py

@@ -0,0 +1,144 @@
+"""Plugin to enable screenshot functionality."""
+
+from typing import TYPE_CHECKING
+
+from reflex.plugins.base import Plugin as BasePlugin
+
+if TYPE_CHECKING:
+    from starlette.requests import Request
+    from starlette.responses import Response
+    from typing_extensions import Unpack
+
+    from reflex.app import App
+    from reflex.plugins.base import PostCompileContext
+    from reflex.state import BaseState
+
+ACTIVE_CONNECTIONS = "/_active_connections"
+CLONE_STATE = "/_clone_state"
+
+
+def _deep_copy(state: "BaseState") -> "BaseState":
+    """Create a deep copy of the state.
+
+    Args:
+        state: The state to copy.
+
+    Returns:
+        A deep copy of the state.
+    """
+    import copy
+
+    copy_of_state = copy.deepcopy(state)
+
+    def copy_substate(substate: "BaseState") -> "BaseState":
+        substate_copy = _deep_copy(substate)
+
+        substate_copy.parent_state = copy_of_state
+
+        return substate_copy
+
+    copy_of_state.substates = {
+        substate_name: copy_substate(substate)
+        for substate_name, substate in state.substates.items()
+    }
+
+    return copy_of_state
+
+
+class ScreenshotPlugin(BasePlugin):
+    """Plugin to handle screenshot functionality."""
+
+    def post_compile(self, **context: "Unpack[PostCompileContext]") -> None:
+        """Called after the compilation of the plugin.
+
+        Args:
+            context: The context for the plugin.
+        """
+        app = context["app"]
+        self._add_active_connections_endpoint(app)
+        self._add_clone_state_endpoint(app)
+
+    @staticmethod
+    def _add_active_connections_endpoint(app: "App") -> None:
+        """Add an endpoint to the app that returns the active connections.
+
+        Args:
+            app: The application instance to which the endpoint will be added.
+        """
+        if not app._api:
+            return
+
+        async def active_connections(_request: "Request") -> "Response":
+            from starlette.responses import JSONResponse
+
+            if not app.event_namespace:
+                return JSONResponse({})
+
+            return JSONResponse(app.event_namespace.token_to_sid)
+
+        app._api.add_route(
+            ACTIVE_CONNECTIONS,
+            active_connections,
+            methods=["GET"],
+        )
+
+    @staticmethod
+    def _add_clone_state_endpoint(app: "App") -> None:
+        """Add an endpoint to the app that clones the current state.
+
+        Args:
+            app: The application instance to which the endpoint will be added.
+        """
+        if not app._api:
+            return
+
+        async def clone_state(request: "Request") -> "Response":
+            import uuid
+
+            from starlette.responses import JSONResponse
+
+            from reflex.state import _substate_key
+
+            if not app.event_namespace:
+                return JSONResponse({})
+
+            token_to_clone = await request.json()
+
+            if not isinstance(token_to_clone, str):
+                return JSONResponse(
+                    {"error": "Token to clone must be a string."}, status_code=400
+                )
+
+            old_state = await app.state_manager.get_state(token_to_clone)
+
+            new_state = _deep_copy(old_state)
+
+            new_token = uuid.uuid4().hex
+
+            all_states = [new_state]
+
+            found_new = True
+
+            while found_new:
+                found_new = False
+
+                for state in all_states:
+                    for substate in state.substates.values():
+                        substate._was_touched = True
+
+                        if substate not in all_states:
+                            all_states.append(substate)
+
+                            found_new = True
+
+            await app.state_manager.set_state(
+                _substate_key(new_token, new_state), new_state
+            )
+
+            return JSONResponse(new_token)
+
+        app._api.add_route(
+            CLONE_STATE,
+            clone_state,
+            methods=["POST"],
+        )

reflex/plugins/base.py

@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, ParamSpec, Protocol, TypedDict
 from typing_extensions import Unpack
 
 if TYPE_CHECKING:
-    from reflex.app import UnevaluatedPage
+    from reflex.app import App, UnevaluatedPage
 
 
 class CommonContext(TypedDict):
@@ -44,6 +44,12 @@ class PreCompileContext(CommonContext):
     unevaluated_pages: Sequence["UnevaluatedPage"]
 
 
+class PostCompileContext(CommonContext):
+    """Context for post-compile hooks."""
+
+    app: "App"
+
+
 class Plugin:
     """Base class for all plugins."""
 
@@ -104,6 +110,13 @@ class Plugin:
             context: The context for the plugin.
         """
 
+    def post_compile(self, **context: Unpack[PostCompileContext]) -> None:
+        """Called after the compilation of the plugin.
+
+        Args:
+            context: The context for the plugin.
+        """
+
     def __repr__(self):
         """Return a string representation of the plugin.
 

reflex/state.py

@@ -509,7 +509,7 @@ class BaseState(EvenMoreBasicBaseState):
 
         new_backend_vars = {
             name: value
-            for name, value in cls.__dict__.items()
+            for name, value in list(cls.__dict__.items())
             if types.is_backend_base_variable(name, cls)
         }
         # Add annotated backend vars that may not have a default value.

{reflex-0.8.6a0.dist-info → reflex-0.8.6a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reflex
-Version: 0.8.6a0
+Version: 0.8.6a1
 Summary: Web apps in pure Python.
 Project-URL: homepage, https://reflex.dev
 Project-URL: repository, https://github.com/reflex-dev/reflex

{reflex-0.8.6a0.dist-info → reflex-0.8.6a1.dist-info}/RECORD

@@ -2,7 +2,7 @@ reflex/__init__.py,sha256=_1PVYjDeA6_JyfXvL6OuKjjO6AX2oMiNcAq8AEHf6xw,10161
 reflex/__init__.pyi,sha256=0D46kHVUJPE_kgYL-BjraERu-MXNCPsQTZQShrijmeQ,10148
 reflex/__main__.py,sha256=6cVrGEyT3j3tEvlEVUatpaYfbB5EF3UVY-6vc_Z7-hw,108
 reflex/admin.py,sha256=Nbc38y-M8iaRBvh1W6DQu_D3kEhO8JFvxrog4q2cB_E,434
-reflex/app.py,sha256=2Jay0aocK1_rzR9Gbao2dxugXGdEuo_IoCQgCKQfdgY,77374
+reflex/app.py,sha256=g5bpc29jekySAvA0a3advQ1dUM-y4cKAM8vMB7oNAsM,77731
 reflex/assets.py,sha256=l5O_mlrTprC0lF7Rc_McOe3a0OtSLnRdNl_PqCpDCBA,3431
 reflex/base.py,sha256=Oh664QL3fZEHErhUasFqP7fE4olYf1y-9Oj6uZI2FCU,1173
 reflex/config.py,sha256=iL5hbdel04GzTtNiPwoNQ6xACJO1SkuvwzAyodDv028,19875
@@ -13,7 +13,7 @@ reflex/page.py,sha256=ssCbMVFuIy60vH-YhJUzN0OxzUwXFCCD3ej56dVjp3g,3525
 reflex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reflex/reflex.py,sha256=BkHnxKp3uZEllLWBbH8Y_EfSu80L4_0FEbSbZkg7JyQ,22506
 reflex/route.py,sha256=ZRlXl1HeKWM3W-yFCnRFIA_6I3W9JBqe2CKzCpWX7Vg,7699
-reflex/state.py,sha256=BunDGopplAH9qUagCv8j80m-IHnVX2LfF-bYo2aDspg,93055
+reflex/state.py,sha256=V84fFvC8yzs0hoqgAgF1BMK81yWm6UZMGUxSJBQPf6o,93061
 reflex/style.py,sha256=JxbXXA4MTnXrk0XHEoMBoNC7J-M2oL5Hl3W_QmXvmBg,13222
 reflex/testing.py,sha256=MgYxW3MLjd8TX5Lv7EWW-go3Fu_gL07i9HpxPJpBUDw,39951
 reflex/.templates/apps/blank/assets/favicon.ico,sha256=baxxgDAQ2V4-G5Q4S2yK5uUJTUGkv-AOWBQ0xd6myUo,4286
@@ -82,7 +82,7 @@ reflex/components/base/body.py,sha256=KLPOhxVsKyjPwrY9AziCOOG_c9ckOqIhI4n2i3_Q3N
 reflex/components/base/body.pyi,sha256=VG8N-0ocwgFOJR9mXfu6a5BtIY_jR7mK9lX3Y22g9GQ,8655
 reflex/components/base/document.py,sha256=Fr7y22NbeKeiz8kWPH2q5BpFjKdq-AmY-sxZilee_H8,636
 reflex/components/base/document.pyi,sha256=PlvclB2vyGcQ8zLJpFac_37LHeXLwpe8Qh0UqzXAPfE,11788
-reflex/components/base/error_boundary.py,sha256=hGp0yjFjBLutbplZkI3fkHBjIWBlYOoZ_1ISth7jEV0,6351
+reflex/components/base/error_boundary.py,sha256=sp6W7G6uFWdls6nzl17XSizSSm6zS2OAu3ZqTE2W1CE,6438
 reflex/components/base/error_boundary.pyi,sha256=SvKnZ09jwIL_4-J5BKVZuATclrB4o3Eh457iTmmoNb4,2926
 reflex/components/base/fragment.py,sha256=ys7wkokq-N8WBxa9fqkEaNIrBlSximyD7vqlFVe02hQ,342
 reflex/components/base/fragment.pyi,sha256=1vl8p6SCWd0_QzYVAb-Em70IX6FwIWFvJLxMl5OJTwU,2401
@@ -352,15 +352,16 @@ reflex/experimental/hooks.py,sha256=CHYGrAE5t8riltrJmDFgJ4D2Vhmhw-y3B3MSGNlOQow,
 reflex/istate/__init__.py,sha256=afq_pCS5B_REC-Kl3Rbaa538uWi59xNz4INeuENcWnk,2039
 reflex/istate/data.py,sha256=KrANpAQFTZcfFWnxyykgK1FRNuXTwTin1WYpyqymV-U,7547
 reflex/istate/dynamic.py,sha256=xOQ9upZVPf6ngqcLQZ9HdAAYmoWwJ8kRFPH34Q5HTiM,91
-reflex/istate/manager.py,sha256=te5uQN6n-ctOGrf6NbZUVy6BKgWapLlMtCzDaO6RthU,30412
+reflex/istate/manager.py,sha256=8EezUlCGkkg_UIs1xa460kZFdJ9qHo6naazphT4jqns,30446
 reflex/istate/proxy.py,sha256=Q8JrV1m6drVcTNJL9JuN-nKUXclazs96OHl_fhR0UBk,25928
 reflex/istate/storage.py,sha256=gCPoiZxuG-Rw0y-Pz3OC7rv4o08dQ_jK1fE2u8Jhxqg,4339
 reflex/istate/wrappers.py,sha256=p8uuioXRbR5hperwbOJHUcWdu7hukLikQdoR7qrnKsI,909
 reflex/middleware/__init__.py,sha256=x7xTeDuc73Hjj43k1J63naC9x8vzFxl4sq7cCFBX7sk,111
 reflex/middleware/hydrate_middleware.py,sha256=1ch7bx2ZhojOR15b-LHD2JztrWCnpPJjTe8MWHJe-5Y,1510
 reflex/middleware/middleware.py,sha256=p5VVoIgQ_NwOg_GOY6g0S4fmrV76_VE1zt-HiwbMw-s,1158
-reflex/plugins/__init__.py,sha256=aARE63iAH2pNVdGmFU4qkXDiPoeBxINfkFzwk-NZe0w,351
-reflex/plugins/base.py,sha256=Jgj7Xrpk0ztM57kl0hN3PUHYrGSXu7oj7OPyIWQJUcE,3059
+reflex/plugins/__init__.py,sha256=jrMWQqMxCwDwgwQYTygeR_pIewMcvIFwAnngPbjSumQ,439
+reflex/plugins/_screenshot.py,sha256=CAOaRpbrpTTIswwCXqhv7WYShB86Ao9MVv6dcXJzRb4,3958
+reflex/plugins/base.py,sha256=5BgzCM7boj9kJ6FGzVzVlgQk-crJuVmOLCl1PXvv4-E,3372
 reflex/plugins/shared_tailwind.py,sha256=UXUndEEcYBZ02klymw-vSZv01IZVLJG3oSaBHpQ617U,6426
 reflex/plugins/sitemap.py,sha256=Jj47uSMnkxndl7awkl48EhlQylBfY00WuMBNyTBcZHA,6186
 reflex/plugins/tailwind_v3.py,sha256=7bXI-zsGoS1pW27-_gskxGaUOQ7NQMPcYkoI5lnmIMA,4819
@@ -399,8 +400,8 @@ reflex/vars/number.py,sha256=tO7pnvFaBsedq1HWT4skytnSqHWMluGEhUbjAUMx8XQ,28190
 reflex/vars/object.py,sha256=BDmeiwG8v97s_BnR1Egq3NxOKVjv9TfnREB3cz0zZtk,17322
 reflex/vars/sequence.py,sha256=1kBrqihspyjyQ1XDqFPC8OpVGtZs_EVkOdIKBro5ilA,55249
 scripts/hatch_build.py,sha256=-4pxcLSFmirmujGpQX9UUxjhIC03tQ_fIQwVbHu9kc0,1861
-reflex-0.8.6a0.dist-info/METADATA,sha256=2RaX5LhTj99Yn0i6IbUON1L_nKWz5EHINSBMcYHeGpY,12541
-reflex-0.8.6a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reflex-0.8.6a0.dist-info/entry_points.txt,sha256=Rxt4dXc7MLBNt5CSHTehVPuSe9Xqow4HLX55nD9tQQ0,45
-reflex-0.8.6a0.dist-info/licenses/LICENSE,sha256=dw3zLrp9f5ObD7kqS32vWfhcImfO52PMmRqvtxq_YEE,11358
-reflex-0.8.6a0.dist-info/RECORD,,
+reflex-0.8.6a1.dist-info/METADATA,sha256=teDs-RrGAMBksliapUSw7JC4U6yaHZoVfD25pKGzFzY,12541
+reflex-0.8.6a1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reflex-0.8.6a1.dist-info/entry_points.txt,sha256=Rxt4dXc7MLBNt5CSHTehVPuSe9Xqow4HLX55nD9tQQ0,45
+reflex-0.8.6a1.dist-info/licenses/LICENSE,sha256=dw3zLrp9f5ObD7kqS32vWfhcImfO52PMmRqvtxq_YEE,11358
+reflex-0.8.6a1.dist-info/RECORD,,