reflex 0.7.4a2__py3-none-any.whl → 0.7.5a1__py3-none-any.whl

reflex/reflex.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import atexit
 from pathlib import Path
+from typing import TYPE_CHECKING
 
 import typer
 import typer.core
@@ -429,7 +430,7 @@ def logout(
 
     loglevel = loglevel or get_config().loglevel
 
-    logout(loglevel)  # pyright: ignore [reportArgumentType]
+    logout(_convert_reflex_loglevel_to_reflex_cli_loglevel(loglevel))
 
 
 db_cli = typer.Typer()
@@ -581,7 +582,6 @@ def deploy(
     ),
 ):
     """Deploy the app to the Reflex hosting service."""
-    from reflex_cli.constants.base import LogLevel as HostingLogLevel
     from reflex_cli.utils import dependency
     from reflex_cli.v2 import cli as hosting_cli
     from reflex_cli.v2.deployments import check_version
@@ -604,21 +604,6 @@ def deploy(
     # Set the log level.
     console.set_log_level(loglevel)
 
-    def convert_reflex_loglevel_to_reflex_cli_loglevel(
-        loglevel: constants.LogLevel,
-    ) -> HostingLogLevel:
-        if loglevel == constants.LogLevel.DEBUG:
-            return HostingLogLevel.DEBUG
-        if loglevel == constants.LogLevel.INFO:
-            return HostingLogLevel.INFO
-        if loglevel == constants.LogLevel.WARNING:
-            return HostingLogLevel.WARNING
-        if loglevel == constants.LogLevel.ERROR:
-            return HostingLogLevel.ERROR
-        if loglevel == constants.LogLevel.CRITICAL:
-            return HostingLogLevel.CRITICAL
-        return HostingLogLevel.INFO
-
     # Only check requirements if interactive.
     # There is user interaction for requirements update.
     if interactive:
@@ -652,7 +637,7 @@ def deploy(
         envfile=envfile,
         hostname=hostname,
         interactive=interactive,
-        loglevel=convert_reflex_loglevel_to_reflex_cli_loglevel(loglevel),
+        loglevel=_convert_reflex_loglevel_to_reflex_cli_loglevel(loglevel),
         token=token,
         project=project,
         project_name=project_name,
@@ -676,6 +661,36 @@ def rename(
     prerequisites.rename_app(new_name, loglevel)
 
 
+if TYPE_CHECKING:
+    from reflex_cli.constants.base import LogLevel as HostingLogLevel
+
+
+def _convert_reflex_loglevel_to_reflex_cli_loglevel(
+    loglevel: constants.LogLevel,
+) -> HostingLogLevel:
+    """Convert a Reflex log level to a Reflex CLI log level.
+
+    Args:
+        loglevel: The Reflex log level to convert.
+
+    Returns:
+        The converted Reflex CLI log level.
+    """
+    from reflex_cli.constants.base import LogLevel as HostingLogLevel
+
+    if loglevel == constants.LogLevel.DEBUG:
+        return HostingLogLevel.DEBUG
+    if loglevel == constants.LogLevel.INFO:
+        return HostingLogLevel.INFO
+    if loglevel == constants.LogLevel.WARNING:
+        return HostingLogLevel.WARNING
+    if loglevel == constants.LogLevel.ERROR:
+        return HostingLogLevel.ERROR
+    if loglevel == constants.LogLevel.CRITICAL:
+        return HostingLogLevel.CRITICAL
+    return HostingLogLevel.INFO
+
+
 cli.add_typer(db_cli, name="db", help="Subcommands for managing the database schema.")
 cli.add_typer(script_cli, name="script", help="Subcommands running helper scripts.")
 cli.add_typer(

reflex/state.py

@@ -593,8 +593,8 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
             if cls._item_is_event_handler(name, fn)
         }
 
-        for mixin in cls._mixins():  # pyright: ignore [reportAssignmentType]
-            for name, value in mixin.__dict__.items():
+        for mixin_cls in cls._mixins():
+            for name, value in mixin_cls.__dict__.items():
                 if name in cls.inherited_vars:
                     continue
                 if is_computed_var(value):
@@ -605,7 +605,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
                     cls.computed_vars[newcv._js_expr] = newcv
                     cls.vars[newcv._js_expr] = newcv
                     continue
-                if types.is_backend_base_variable(name, mixin):  # pyright: ignore [reportArgumentType]
+                if types.is_backend_base_variable(name, mixin_cls):
                     cls.backend_vars[name] = copy.deepcopy(value)
                     continue
                 if events.get(name) is not None:
@@ -907,7 +907,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
             raise ValueError(f"Only one parent state is allowed {parent_states}.")
         # The first non-mixin state in the mro is our parent.
         for base in cls.mro()[1:]:
-            if base._mixin or not issubclass(base, BaseState):
+            if not issubclass(base, BaseState) or base._mixin:
                 continue
             if base is BaseState:
                 break

reflex/style.py

@@ -237,10 +237,10 @@ def format_style_key(key: str) -> tuple[str, ...]:
 EMPTY_VAR_DATA = VarData()
 
 
-class Style(dict):
+class Style(dict[str, Any]):
     """A style dictionary."""
 
-    def __init__(self, style_dict: dict | None = None, **kwargs):
+    def __init__(self, style_dict: dict[str, Any] | None = None, **kwargs):
         """Initialize the style.
 
         Args:

reflex/testing.py

@@ -27,6 +27,7 @@ from typing import (
     Callable,
     Coroutine,
     Optional,
+    Sequence,
     Type,
     TypeVar,
 )
@@ -268,6 +269,10 @@ class AppHarness:
                     loglevel=reflex.constants.LogLevel.INFO,
                 )
                 self.app_module_path.write_text(source_code)
+        else:
+            # Just initialize the web folder.
+            with chdir(self.app_path):
+                reflex.utils.prerequisites.initialize_frontend_dependencies()
         with chdir(self.app_path):
             # ensure config and app are reloaded when testing different app
             reflex.config.get_config(reload=True)
@@ -290,8 +295,10 @@ class AppHarness:
         if self.app_instance and isinstance(
             self.app_instance._state_manager, StateManagerRedis
         ):
+            if self.app_instance._state is None:
+                raise RuntimeError("State is not set.")
             # Create our own redis connection for testing.
-            self.state_manager = StateManagerRedis.create(self.app_instance._state)  # pyright: ignore [reportArgumentType]
+            self.state_manager = StateManagerRedis.create(self.app_instance._state)
         else:
             self.state_manager = (
                 self.app_instance._state_manager if self.app_instance else None
@@ -455,6 +462,10 @@ class AppHarness:
 
     def stop(self) -> None:
         """Stop the frontend and backend servers."""
+        # Quit browsers first to avoid any lingering events being sent during shutdown.
+        for driver in self._frontends:
+            driver.quit()
+
         self._reload_state_module()
 
         if self.backend is not None:
@@ -485,8 +496,6 @@ class AppHarness:
             self.backend_thread.join()
         if self.frontend_output_thread is not None:
             self.frontend_output_thread.join()
-        for driver in self._frontends:
-            driver.quit()
 
         # Cleanup decorated pages added during testing
         for page in self._decorated_pages:
@@ -764,7 +773,7 @@ class AppHarness:
         self,
         element: "WebElement",
         timeout: TimeoutType = None,
-        exp_not_equal: str = "",
+        exp_not_equal: str | Sequence[str] = "",
     ) -> str | None:
         """Poll element.get_attribute("value") for change.
 
@@ -779,8 +788,11 @@ class AppHarness:
         Raises:
             TimeoutError: when the timeout expires before value changes
         """
+        exp_not_equal = (
+            (exp_not_equal,) if isinstance(exp_not_equal, str) else exp_not_equal
+        )
         if not self._poll_for(
-            target=lambda: element.get_attribute("value") != exp_not_equal,
+            target=lambda: element.get_attribute("value") not in exp_not_equal,
             timeout=timeout,
         ):
             raise TimeoutError(

reflex/utils/console.py

@@ -201,10 +201,9 @@ def _get_first_non_framework_frame() -> FrameType | None:
     # Exclude utility modules that should never be the source of deprecated reflex usage.
     exclude_modules = [click, rx, typer, typing_extensions]
     exclude_roots = [
-        p.parent.resolve()
-        if (p := Path(m.__file__)).name == "__init__.py"  # pyright: ignore [reportArgumentType]
-        else p.resolve()
+        p.parent.resolve() if (p := Path(file)).name == "__init__.py" else p.resolve()
         for m in exclude_modules
+        if (file := m.__file__)
     ]
     # Specifically exclude the reflex cli module.
     if reflex_bin := shutil.which(b"reflex"):

reflex/utils/exec.py

@@ -330,6 +330,7 @@ def run_uvicorn_backend(host: str, port: int, loglevel: LogLevel):
         log_level=loglevel.value,
         reload=True,
         reload_dirs=list(map(str, get_reload_paths())),
+        reload_delay=0.1,
     )
 
 
@@ -356,6 +357,9 @@ def run_granian_backend(host: str, port: int, loglevel: LogLevel):
         log_level=LogLevels(loglevel.value),
         reload=True,
         reload_paths=get_reload_paths(),
+        reload_ignore_worker_failure=True,
+        reload_tick=100,
+        workers_kill_timeout=2,
     ).serve()
 
 

reflex/utils/imports.py

@@ -4,11 +4,11 @@ from __future__ import annotations
 
 import dataclasses
 from collections import defaultdict
-from typing import DefaultDict, Union
+from typing import DefaultDict, Mapping, Sequence, Union
 
 
 def merge_imports(
-    *imports: ImportDict | ParsedImportDict | ImmutableParsedImportDict,
+    *imports: ImportDict | ParsedImportDict | ParsedImportTuple,
 ) -> ParsedImportDict:
     """Merge multiple import dicts together.
 
@@ -43,7 +43,9 @@ def merge_imports(
     return all_imports
 
 
-def parse_imports(imports: ImportDict | ParsedImportDict) -> ParsedImportDict:
+def parse_imports(
+    imports: ImmutableImportDict | ImmutableParsedImportDict,
+) -> ParsedImportDict:
     """Parse the import dict into a standard format.
 
     Args:
@@ -53,10 +55,12 @@ def parse_imports(imports: ImportDict | ParsedImportDict) -> ParsedImportDict:
         The parsed import dict.
     """
 
-    def _make_list(value: ImportTypes) -> list[str | ImportVar] | list[ImportVar]:
+    def _make_list(
+        value: ImmutableImportTypes,
+    ) -> list[str | ImportVar] | list[ImportVar]:
         if isinstance(value, (str, ImportVar)):
             return [value]
-        return value
+        return list(value)
 
     return {
         package: [
@@ -68,7 +72,7 @@ def parse_imports(imports: ImportDict | ParsedImportDict) -> ParsedImportDict:
 
 
 def collapse_imports(
-    imports: ParsedImportDict | ImmutableParsedImportDict,
+    imports: ParsedImportDict | ParsedImportTuple,
 ) -> ParsedImportDict:
     """Remove all duplicate ImportVar within an ImportDict.
 
@@ -132,6 +136,9 @@ class ImportVar:
 
 
 ImportTypes = Union[str, ImportVar, list[str | ImportVar], list[ImportVar]]
+ImmutableImportTypes = Union[str, ImportVar, Sequence[str | ImportVar]]
 ImportDict = dict[str, ImportTypes]
+ImmutableImportDict = Mapping[str, ImmutableImportTypes]
 ParsedImportDict = dict[str, list[ImportVar]]
-ImmutableParsedImportDict = tuple[tuple[str, tuple[ImportVar, ...]], ...]
+ImmutableParsedImportDict = Mapping[str, Sequence[ImportVar]]
+ParsedImportTuple = tuple[tuple[str, tuple[ImportVar, ...]], ...]

reflex/utils/net.py

@@ -1,8 +1,13 @@
 """Helpers for downloading files from the network."""
 
+import functools
+import time
+from typing import Callable, ParamSpec, TypeVar
+
 import httpx
 
-from ..config import environment
+from reflex.utils.decorator import once
+
 from . import console
 
 
@@ -12,30 +17,114 @@ def _httpx_verify_kwarg() -> bool:
     Returns:
         True if SSL verification is enabled, False otherwise
     """
+    from ..config import environment
+
     return not environment.SSL_NO_VERIFY.get()
 
 
-def get(url: str, **kwargs) -> httpx.Response:
-    """Make an HTTP GET request.
+_P = ParamSpec("_P")
+_T = TypeVar("_T")
+
+
+def _wrap_https_func(
+    func: Callable[_P, _T],
+) -> Callable[_P, _T]:
+    """Wrap an HTTPS function with logging.
 
     Args:
-        url: The URL to request.
-        **kwargs: Additional keyword arguments to pass to httpx.get.
+        func: The function to wrap.
 
     Returns:
-        The response object.
+        The wrapped function.
+    """
+
+    @functools.wraps(func)
+    def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> _T:
+        url = args[0]
+        console.debug(f"Sending HTTPS request to {args[0]}")
+        initial_time = time.time()
+        try:
+            response = func(*args, **kwargs)
+        except httpx.ConnectError as err:
+            if "CERTIFICATE_VERIFY_FAILED" in str(err):
+                # If the error is a certificate verification error, recommend mitigating steps.
+                console.error(
+                    f"Certificate verification failed for {url}. Set environment variable SSL_CERT_FILE to the "
+                    "path of the certificate file or SSL_NO_VERIFY=1 to disable verification."
+                )
+            raise
+        else:
+            console.debug(
+                f"Received response from {url} in {time.time() - initial_time:.3f} seconds"
+            )
+            return response
+
+    return wrapper
+
 
-    Raises:
-        httpx.ConnectError: If the connection cannot be established.
+def _is_ipv4_supported() -> bool:
+    """Determine if the system supports IPv4.
+
+    Returns:
+        True if the system supports IPv4, False otherwise.
     """
-    kwargs.setdefault("verify", _httpx_verify_kwarg())
     try:
-        return httpx.get(url, **kwargs)
-    except httpx.ConnectError as err:
-        if "CERTIFICATE_VERIFY_FAILED" in str(err):
-            # If the error is a certificate verification error, recommend mitigating steps.
-            console.error(
-                f"Certificate verification failed for {url}. Set environment variable SSL_CERT_FILE to the "
-                "path of the certificate file or SSL_NO_VERIFY=1 to disable verification."
-            )
-        raise
+        httpx.head("http://1.1.1.1", timeout=3)
+    except httpx.RequestError:
+        return False
+    else:
+        return True
+
+
+def _is_ipv6_supported() -> bool:
+    """Determine if the system supports IPv6.
+
+    Returns:
+        True if the system supports IPv6, False otherwise.
+    """
+    try:
+        httpx.head("http://[2606:4700:4700::1111]", timeout=3)
+    except httpx.RequestError:
+        return False
+    else:
+        return True
+
+
+def _should_use_ipv6() -> bool:
+    """Determine if the system supports IPv6.
+
+    Returns:
+        True if the system supports IPv6, False otherwise.
+    """
+    return not _is_ipv4_supported() and _is_ipv6_supported()
+
+
+def _httpx_local_address_kwarg() -> str:
+    """Get the value of the HTTPX local_address keyword argument.
+
+    Returns:
+        The local address to bind to
+    """
+    from ..config import environment
+
+    return environment.REFLEX_HTTP_CLIENT_BIND_ADDRESS.get() or (
+        "::" if _should_use_ipv6() else "0.0.0.0"
+    )
+
+
+@once
+def _httpx_client() -> httpx.Client:
+    """Get an HTTPX client.
+
+    Returns:
+        An HTTPX client.
+    """
+    return httpx.Client(
+        transport=httpx.HTTPTransport(
+            local_address=_httpx_local_address_kwarg(),
+            verify=_httpx_verify_kwarg(),
+        )
+    )
+
+
+get = _wrap_https_func(_httpx_client().get)