Merge branch 'exp-memo' of https://github.com/FarhanAliRaza/reflex into exp-memo

Author: FarhanAliRaza

pyi_hashes.json

@@ -23,7 +23,7 @@
   "reflex/components/core/window_events.pyi": "af33ccec866b9540ee7fbec6dbfbd151",
   "reflex/components/datadisplay/__init__.pyi": "52755871369acbfd3a96b46b9a11d32e",
   "reflex/components/datadisplay/code.pyi": "b86769987ef4d1cbdddb461be88539fd",
-  "reflex/components/datadisplay/dataeditor.pyi": "f8c1e816c9f22f4a7429f812214407f2",
+  "reflex/components/datadisplay/dataeditor.pyi": "9b85f3cf6156293cd9961eb17a0ea684",
   "reflex/components/datadisplay/shiki_code_block.pyi": "1d53e75b6be0d3385a342e7b3011babd",
   "reflex/components/el/__init__.pyi": "0adfd001a926a2a40aee94f6fa725ecc",
   "reflex/components/el/element.pyi": "c5974a92fbc310e42d0f6cfdd13472f4",
@@ -39,8 +39,8 @@
   "reflex/components/el/elements/tables.pyi": "686eb70ea7d8c4dafb0cc5c284e76184",
   "reflex/components/el/elements/typography.pyi": "684e83dde887dba12badd0fb75c87c04",
   "reflex/components/gridjs/datatable.pyi": "98a7e1b3f3b60cafcdfcd8879750ee42",
-  "reflex/components/lucide/icon.pyi": "dcb8773ef162f3ec5759efe11374cf5e",
-  "reflex/components/markdown/markdown.pyi": "dd74e8e9665b2a813ff799a7aa190b44",
+  "reflex/components/lucide/icon.pyi": "1db10f2b544908dd20c56ca4bc33d5e0",
+  "reflex/components/markdown/markdown.pyi": "4c4bca6fb0643abb90aca9c0bb87f722",
   "reflex/components/moment/moment.pyi": "e1952f1c2c82cef85d91e970d1be64ab",
   "reflex/components/plotly/plotly.pyi": "4311a0aae2abcc9226abb6a273f96372",
   "reflex/components/radix/__init__.pyi": "5d8e3579912473e563676bfc71f29191",
@@ -49,7 +49,7 @@
   "reflex/components/radix/primitives/base.pyi": "9ef34884fb6028dc017df5e2db639c81",
   "reflex/components/radix/primitives/dialog.pyi": "9ee73362bb59619c482b6b0d07033f37",
   "reflex/components/radix/primitives/drawer.pyi": "921e45dfaf5b9131ef27c561c3acca2e",
-  "reflex/components/radix/primitives/form.pyi": "78055e820703c98c3b838aa889566365",
+  "reflex/components/radix/primitives/form.pyi": "17002a3e9d7f52b3207614f6c1c9a24a",
   "reflex/components/radix/primitives/progress.pyi": "c917952d57ddb3e138a40c4005120d5e",
   "reflex/components/radix/primitives/slider.pyi": "4ff06f0025d47f166132909b09ab96f8",
   "reflex/components/radix/themes/__init__.pyi": "582b4a7ead62b2ae8605e17fa084c063",
@@ -71,7 +71,7 @@
   "reflex/components/radix/themes/components/dialog.pyi": "d2615f1a68c80ff930444d054b598c13",
   "reflex/components/radix/themes/components/dropdown_menu.pyi": "43f8770c9adf93c73398d68f79048424",
   "reflex/components/radix/themes/components/hover_card.pyi": "a96f4433237f9994decf935deff9f269",
-  "reflex/components/radix/themes/components/icon_button.pyi": "e930911d8ecbe61e5447e61c76a28ab6",
+  "reflex/components/radix/themes/components/icon_button.pyi": "f12a874bad243a81e5c8740a1d86c6bc",
   "reflex/components/radix/themes/components/inset.pyi": "bd7a2186b553bd4c86d83ff50c784066",
   "reflex/components/radix/themes/components/popover.pyi": "91f8edefeb232cc6d48690b1838144c2",
   "reflex/components/radix/themes/components/progress.pyi": "0e59587d5b3c8fe0d0067587f144e5b0",
@@ -106,7 +106,7 @@
   "reflex/components/radix/themes/typography/blockquote.pyi": "080c71899532f5dbf4cf143e7a5ad3bf",
   "reflex/components/radix/themes/typography/code.pyi": "7ffe785d55979cf8ff97ea040f3e2b64",
   "reflex/components/radix/themes/typography/heading.pyi": "0ebb38915cd0521fd59c569e04d288bb",
-  "reflex/components/radix/themes/typography/link.pyi": "e88c5d880a54548b6808c097ac62505b",
+  "reflex/components/radix/themes/typography/link.pyi": "64878125a37d47d676c9adf8156d8c41",
   "reflex/components/radix/themes/typography/text.pyi": "50f9ca15a941e4b77ddd12e77aa3c03e",
   "reflex/components/react_player/audio.pyi": "0e1690ff1f1f39bc748278d292238350",
   "reflex/components/react_player/react_player.pyi": "5ccd373b94ed1d3934ae6afc46bd6fe4",

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "reflex"
-version = "0.8.28dev1"
+version = "0.9.0dev1"
 description = "Web apps in pure Python."
 license.text = "Apache-2.0"
 authors = [
@@ -244,7 +244,7 @@ fail_fast = true
 
 [[tool.pre-commit.repos]]
 repo = "https://github.com/astral-sh/ruff-pre-commit"
-rev = "v0.15.1"
+rev = "v0.15.6"
 hooks = [
   { id = "ruff-format", args = [
     "reflex",

reflex/app.py

@@ -42,6 +42,7 @@
 from starlette.requests import ClientDisconnect, Request
 from starlette.responses import JSONResponse, Response, StreamingResponse
 from starlette.staticfiles import StaticFiles
+from typing_extensions import Unpack
 
 from reflex import constants
 from reflex.admin import AdminDash
@@ -84,6 +85,7 @@
     noop,
 )
 from reflex.experimental.memo import EXPERIMENTAL_MEMOS
+from reflex.istate.manager import StateModificationContext
 from reflex.istate.proxy import StateProxy
 from reflex.page import DECORATED_PAGES
 from reflex.route import (
@@ -610,11 +612,11 @@ def __repr__(self) -> str:
     def __call__(self) -> ASGIApp:
         """Run the backend api instance.
 
-        Raises:
-            ValueError: If the app has not been initialized.
-
         Returns:
             The backend api.
+
+        Raises:
+            ValueError: If the app has not been initialized.
         """
         from reflex.assets import remove_stale_external_asset_symlinks
         from reflex.vars.base import GLOBAL_CACHE
@@ -929,11 +931,11 @@ def _check_routes_conflict(self, new_route: str):
 
         Based on conflicts that React Router would throw if not intercepted.
 
-        Raises:
-            RouteValueError: exception showing which conflict exist with the route to be added
-
         Args:
             new_route: the route being newly added.
+
+        Raises:
+            RouteValueError: exception showing which conflict exist with the route to be added
         """
         from reflex.utils.exceptions import RouteValueError
 
@@ -1575,6 +1577,7 @@ async def modify_state(
         token: str,
         background: bool = False,
         previous_dirty_vars: dict[str, set[str]] | None = None,
+        **context: Unpack[StateModificationContext],
     ) -> AsyncIterator[BaseState]:
         """Modify the state out of band.
 
@@ -1595,7 +1598,7 @@ async def modify_state(
 
         # Get exclusive access to the state.
         async with self.state_manager.modify_state_with_links(
-            token, previous_dirty_vars=previous_dirty_vars
+            token, previous_dirty_vars=previous_dirty_vars, **context
         ) as state:
             # No other event handler can modify the state while in this context.
             yield state
@@ -1628,7 +1631,7 @@ def _process_background(
         if not handler.is_background:
             return None
 
-        substate = StateProxy(substate)
+        substate = StateProxy(substate, event)
 
         async def _coro():
             """Coroutine to process the event and emit updates inside an asyncio.Task.
@@ -1767,11 +1770,11 @@ async def process(
         headers: The client headers.
         client_ip: The client_ip.
 
-    Raises:
-        Exception: If a reflex specific error occurs during processing the event.
-
     Yields:
         The state updates after processing the event.
+
+    Raises:
+        Exception: If a reflex specific error occurs during processing the event.
     """
     from reflex.utils import telemetry
 
@@ -2046,7 +2049,7 @@ async def _ndjson_updates():
             """
             # Process the event.
             async with app.state_manager.modify_state_with_links(
-                event.substate_token
+                event.substate_token, event=event
             ) as state:
                 async for update in state._process(event):
                     # Postprocess the event.
@@ -2191,14 +2194,12 @@ async def emit_update(self, update: StateUpdate, token: str) -> None:
     async def on_event(self, sid: str, data: Any):
         """Event for receiving front-end websocket events.
 
-        Raises:
-            RuntimeError: If the Socket.IO is badly initialized.
-
         Args:
             sid: The Socket.IO session id.
             data: The event data.
 
         Raises:
+            RuntimeError: If the Socket.IO is badly initialized.
             EventDeserializationError: If the event data is not a dictionary.
         """
         fields = data

reflex/assets.py

@@ -72,12 +72,12 @@ def asset(
             the immediate caller 1. When using rx.asset via a helper function,
             increase this number for each helper function in the stack.
 
+    Returns:
+        The relative URL to the asset.
+
     Raises:
         FileNotFoundError: If the file does not exist.
         ValueError: If subfolder is provided for local assets.
-
-    Returns:
-        The relative URL to the asset.
     """
     assets = constants.Dirs.APP_ASSETS
     backend_only = EnvironmentVariables.REFLEX_BACKEND_ONLY.get()

reflex/compiler/compiler.py

@@ -33,7 +33,7 @@
 from reflex.utils.exceptions import ReflexError
 from reflex.utils.exec import is_prod_mode
 from reflex.utils.format import to_title_case
-from reflex.utils.imports import ImportVar
+from reflex.utils.imports import ImportVar, ParsedImportDict
 from reflex.utils.prerequisites import get_web_dir
 from reflex.vars.base import LiteralVar, Var
 
@@ -411,6 +411,47 @@ def _compile_memo_components(
     )
 
 
+def _get_shared_components_recursive(
+    component: BaseComponent,
+    rendered_components: dict[str, None],
+    all_import_dicts: list[ParsedImportDict],
+):
+    """Get the shared components for a component and its children.
+
+    A shared component is a StatefulComponent that appears in 2 or more
+    pages and is a candidate for writing to a common file and importing
+    into each page where it is used.
+
+    Args:
+        component: The component to collect shared StatefulComponents for.
+        rendered_components: A dict to store the rendered shared components in.
+        all_import_dicts: A list to store the imports of all shared components in.
+    """
+    for child in component.children:
+        # Depth-first traversal.
+        _get_shared_components_recursive(child, rendered_components, all_import_dicts)
+
+    # When the component is referenced by more than one page, render it
+    # to be included in the STATEFUL_COMPONENTS module.
+    # Skip this step in dev mode, thereby avoiding potential hot reload errors for larger apps
+    if isinstance(component, StatefulComponent) and component.references > 1:
+        # Reset this flag to render the actual component.
+        component.rendered_as_shared = False
+
+        # Include dynamic imports in the shared component.
+        if dynamic_imports := component._get_all_dynamic_imports():
+            rendered_components.update(dict.fromkeys(dynamic_imports))
+
+        # Include custom code in the shared component.
+        rendered_components.update(component._get_all_custom_code(export=True))
+
+        # Include all imports in the shared component.
+        all_import_dicts.append(component._get_all_imports())
+
+        # Indicate that this component now imports from the shared file.
+        component.rendered_as_shared = True
+
+
 def _compile_stateful_components(
     page_components: list[BaseComponent],
 ) -> str:
@@ -430,46 +471,10 @@ def _compile_stateful_components(
     all_import_dicts = []
     rendered_components = {}
 
-    def get_shared_components_recursive(component: BaseComponent):
-        """Get the shared components for a component and its children.
-
-        A shared component is a StatefulComponent that appears in 2 or more
-        pages and is a candidate for writing to a common file and importing
-        into each page where it is used.
-
-        Args:
-            component: The component to collect shared StatefulComponents for.
-        """
-        for child in component.children:
-            # Depth-first traversal.
-            get_shared_components_recursive(child)
-
-        # When the component is referenced by more than one page, render it
-        # to be included in the STATEFUL_COMPONENTS module.
-        # Skip this step in dev mode, thereby avoiding potential hot reload errors for larger apps
-        if (
-            isinstance(component, StatefulComponent)
-            and component.references > 1
-            and is_prod_mode()
-        ):
-            # Reset this flag to render the actual component.
-            component.rendered_as_shared = False
-
-            # Include dynamic imports in the shared component.
-            if dynamic_imports := component._get_all_dynamic_imports():
-                rendered_components.update(dict.fromkeys(dynamic_imports))
-
-            # Include custom code in the shared component.
-            rendered_components.update(component._get_all_custom_code(export=True))
-
-            # Include all imports in the shared component.
-            all_import_dicts.append(component._get_all_imports())
-
-            # Indicate that this component now imports from the shared file.
-            component.rendered_as_shared = True
-
     for page_component in page_components:
-        get_shared_components_recursive(page_component)
+        _get_shared_components_recursive(
+            page_component, rendered_components, all_import_dicts
+        )
 
     # Don't import from the file that we're about to create.
     all_imports = utils.merge_imports(*all_import_dicts)
@@ -637,7 +642,7 @@ def compile_stateful_components(
         progress_function()
         page_components.append(page_component)
 
-    code = _compile_stateful_components(page_components)
+    code = _compile_stateful_components(page_components) if is_prod_mode() else ""
     return output_path, code, page_components
 
 

reflex/compiler/utils.py

@@ -45,13 +45,13 @@ def compile_import_statement(fields: list[ImportVar]) -> tuple[str, list[str]]:
     Args:
         fields: The set of fields to import from the library.
 
-    Raises:
-        ValueError: If there is more than one default import.
-
     Returns:
         The libraries for default and rest.
         default: default library. When install "import def from library".
         rest: rest of libraries. When install "import {rest1, rest2} from library"
+
+    Raises:
+        ValueError: If there is more than one default import.
     """
     # ignore the ImportVar fields with render=False during compilation
     fields_set = {field for field in fields if field.render}
@@ -111,11 +111,11 @@ def compile_imports(import_dict: ParsedImportDict) -> list[_ImportDict]:
     Args:
         import_dict: The import dict to compile.
 
-    Raises:
-        ValueError: If an import in the dict is invalid.
-
     Returns:
         The list of import dict.
+
+    Raises:
+        ValueError: If an import in the dict is invalid.
     """
     collapsed_import_dict: ParsedImportDict = imports.collapse_imports(import_dict)
     validate_imports(collapsed_import_dict)

reflex/components/base/meta.py

@@ -14,11 +14,11 @@ class Title(elements.Title):
     def render(self) -> dict:
         """Render the title component.
 
-        Raises:
-            ValueError: If the title is not a single string.
-
         Returns:
             The rendered title component.
+
+        Raises:
+            ValueError: If the title is not a single string.
         """
         # Make sure the title is a single string.
         if len(self.children) != 1 or not isinstance(self.children[0], Bare):

reflex/components/component.py

@@ -1166,11 +1166,11 @@ def _add_style_recursive(
             style: A dict from component to styling.
             theme: The theme to apply. (for retro-compatibility with deprecated _apply_theme API)
 
-        Raises:
-            UserWarning: If `_add_style` has been overridden.
-
         Returns:
             The component with the additional style.
+
+        Raises:
+            UserWarning: If `_add_style` has been overridden.
         """
         # 1. Default style from `_add_style`/`add_style`.
         if type(self)._add_style != Component._add_style:

reflex/components/core/breakpoints.py

@@ -63,11 +63,11 @@ def create(
             lg: Styling when in the large width
             xl: Styling when in the extra-large width
 
-        Raises:
-            ValueError: If both custom and any other named parameters are provided.
-
         Returns:
             The responsive mapping.
+
+        Raises:
+            ValueError: If both custom and any other named parameters are provided.
         """
         thresholds = [initial, xs, sm, md, lg, xl]
 

reflex/components/datadisplay/code.py

@@ -382,7 +382,7 @@ class Theme:
 class CodeBlock(Component, MarkdownComponentMap):
     """A code block."""
 
-    library = "react-syntax-highlighter@16.1.0"
+    library = "react-syntax-highlighter@16.1.1"
 
     tag = "PrismAsyncLight"