reflex 0.6.8a2__py3-none-any.whl → 0.7.0__py3-none-any.whl

reflex/utils/processes.py

@@ -15,12 +15,14 @@ from typing import Callable, Generator, List, Optional, Tuple, Union
 import psutil
 import typer
 from redis.exceptions import RedisError
+from rich.progress import Progress
 
 from reflex import constants
+from reflex.config import environment
 from reflex.utils import console, path_ops, prerequisites
 
 
-def kill(pid):
+def kill(pid: int):
     """Kill a process.
 
     Args:
@@ -48,7 +50,7 @@ def get_num_workers() -> int:
     return (os.cpu_count() or 1) * 2 + 1
 
 
-def get_process_on_port(port) -> Optional[psutil.Process]:
+def get_process_on_port(port: int) -> Optional[psutil.Process]:
     """Get the process on the given port.
 
     Args:
@@ -62,7 +64,7 @@ def get_process_on_port(port) -> Optional[psutil.Process]:
             psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess
         ):
             if importlib.metadata.version("psutil") >= "6.0.0":
-                conns = proc.net_connections(kind="inet")  # type: ignore
+                conns = proc.net_connections(kind="inet")
             else:
                 conns = proc.connections(kind="inet")
             for conn in conns:
@@ -71,7 +73,7 @@ def get_process_on_port(port) -> Optional[psutil.Process]:
     return None
 
 
-def is_process_on_port(port) -> bool:
+def is_process_on_port(port: int) -> bool:
     """Check if a process is running on the given port.
 
     Args:
@@ -83,7 +85,7 @@ def is_process_on_port(port) -> bool:
     return get_process_on_port(port) is not None
 
 
-def kill_process_on_port(port):
+def kill_process_on_port(port: int):
     """Kill the process on the given port.
 
     Args:
@@ -91,10 +93,10 @@ def kill_process_on_port(port):
     """
     if get_process_on_port(port) is not None:
         with contextlib.suppress(psutil.AccessDenied):
-            get_process_on_port(port).kill()  # type: ignore
+            get_process_on_port(port).kill()  # pyright: ignore [reportOptionalMemberAccess]
 
 
-def change_port(port: str, _type: str) -> str:
+def change_port(port: int, _type: str) -> int:
     """Change the port.
 
     Args:
@@ -105,7 +107,7 @@ def change_port(port: str, _type: str) -> str:
         The new port.
 
     """
-    new_port = str(int(port) + 1)
+    new_port = port + 1
     if is_process_on_port(new_port):
         return change_port(new_port, _type)
     console.info(
@@ -114,7 +116,7 @@ def change_port(port: str, _type: str) -> str:
     return new_port
 
 
-def handle_port(service_name: str, port: str, default_port: str) -> str:
+def handle_port(service_name: str, port: int, default_port: int) -> int:
     """Change port if the specified port is in use and is not explicitly specified as a CLI arg or config arg.
     otherwise tell the user the port is in use and exit the app.
 
@@ -133,7 +135,7 @@ def handle_port(service_name: str, port: str, default_port: str) -> str:
         Exit:when the port is in use.
     """
     if is_process_on_port(port):
-        if int(port) == int(default_port):
+        if port == int(default_port):
             return change_port(port, service_name)
         else:
             console.error(f"{service_name.capitalize()} port: {port} is already in use")
@@ -141,7 +143,12 @@ def handle_port(service_name: str, port: str, default_port: str) -> str:
     return port
 
 
-def new_process(args, run: bool = False, show_logs: bool = False, **kwargs):
+def new_process(
+    args: str | list[str] | list[str | None] | list[str | Path | None],
+    run: bool = False,
+    show_logs: bool = False,
+    **kwargs,
+):
     """Wrapper over subprocess.Popen to unify the launch of child processes.
 
     Args:
@@ -156,24 +163,30 @@ def new_process(args, run: bool = False, show_logs: bool = False, **kwargs):
     Raises:
         Exit: When attempting to run a command with a None value.
     """
-    node_bin_path = str(path_ops.get_node_bin_path())
-    if not node_bin_path and not prerequisites.CURRENTLY_INSTALLING_NODE:
-        console.warn(
-            "The path to the Node binary could not be found. Please ensure that Node is properly "
-            "installed and added to your system's PATH environment variable or try running "
-            "`reflex init` again."
-        )
-    if None in args:
+    # Check for invalid command first.
+    if isinstance(args, list) and None in args:
         console.error(f"Invalid command: {args}")
         raise typer.Exit(1)
-    # Add the node bin path to the PATH environment variable.
+
+    path_env: str = os.environ.get("PATH", "")
+
+    # Add node_bin_path to the PATH environment variable.
+    if not environment.REFLEX_BACKEND_ONLY.get():
+        node_bin_path = str(path_ops.get_node_bin_path())
+        if not node_bin_path and not prerequisites.CURRENTLY_INSTALLING_NODE:
+            console.warn(
+                "The path to the Node binary could not be found. Please ensure that Node is properly "
+                "installed and added to your system's PATH environment variable or try running "
+                "`reflex init` again."
+            )
+        path_env = os.pathsep.join([node_bin_path, path_env])
+
     env: dict[str, str] = {
         **os.environ,
-        "PATH": os.pathsep.join(
-            [node_bin_path if node_bin_path else "", os.environ["PATH"]]
-        ),  # type: ignore
+        "PATH": path_env,
         **kwargs.pop("env", {}),
     }
+
     kwargs = {
         "env": env,
         "stderr": None if show_logs else subprocess.STDOUT,
@@ -185,7 +198,7 @@ def new_process(args, run: bool = False, show_logs: bool = False, **kwargs):
     }
     console.debug(f"Running command: {args}")
     fn = subprocess.run if run else subprocess.Popen
-    return fn(args, **kwargs)
+    return fn(args, **kwargs)  # pyright: ignore [reportCallIssue, reportArgumentType]
 
 
 @contextlib.contextmanager
@@ -206,14 +219,14 @@ def run_concurrently_context(
         return
 
     # Convert the functions to tuples.
-    fns = [fn if isinstance(fn, tuple) else (fn,) for fn in fns]  # type: ignore
+    fns = [fn if isinstance(fn, tuple) else (fn,) for fn in fns]  # pyright: ignore [reportAssignmentType]
 
     # Run the functions concurrently.
     executor = None
     try:
         executor = futures.ThreadPoolExecutor(max_workers=len(fns))
         # Submit the tasks.
-        tasks = [executor.submit(*fn) for fn in fns]  # type: ignore
+        tasks = [executor.submit(*fn) for fn in fns]  # pyright: ignore [reportArgumentType]
 
         # Yield control back to the main thread while tasks are running.
         yield tasks
@@ -241,7 +254,7 @@ def run_concurrently(*fns: Union[Callable, Tuple]) -> None:
 def stream_logs(
     message: str,
     process: subprocess.Popen,
-    progress=None,
+    progress: Progress | None = None,
     suppress_errors: bool = False,
     analytics_enabled: bool = False,
 ):
@@ -361,7 +374,7 @@ def get_command_with_loglevel(command: list[str]) -> list[str]:
         The updated command list
     """
     npm_path = path_ops.get_npm_path()
-    npm_path = str(Path(npm_path).resolve()) if npm_path else npm_path
+    npm_path = str(npm_path) if npm_path else None
 
     if command[0] == npm_path:
         return [*command, "--loglevel", "silly"]
@@ -369,10 +382,10 @@ def get_command_with_loglevel(command: list[str]) -> list[str]:
 
 
 def run_process_with_fallback(
-    args,
+    args: list[str],
     *,
-    show_status_message,
-    fallback=None,
+    show_status_message: str,
+    fallback: str | list | None = None,
     analytics_enabled: bool = False,
     **kwargs,
 ):
@@ -411,7 +424,7 @@ def run_process_with_fallback(
             )
 
 
-def execute_command_and_return_output(command) -> str | None:
+def execute_command_and_return_output(command: str) -> str | None:
     """Execute a command and return the output.
 
     Args:

reflex/utils/pyi_generator.py

@@ -75,7 +75,6 @@ DEFAULT_IMPORTS = {
         "EventHandler",
         "EventSpec",
         "EventType",
-        "BASE_STATE",
         "KeyInputInfo",
     ],
     "reflex.style": ["Style"],
@@ -83,7 +82,7 @@ DEFAULT_IMPORTS = {
 }
 
 
-def _walk_files(path):
+def _walk_files(path: str | Path):
     """Walk all files in a path.
     This can be replaced with Path.walk() in python3.12.
 
@@ -114,7 +113,9 @@ def _relative_to_pwd(path: Path) -> Path:
     return path
 
 
-def _get_type_hint(value, type_hint_globals, is_optional=True) -> str:
+def _get_type_hint(
+    value: Any, type_hint_globals: dict, is_optional: bool = True
+) -> str:
     """Resolve the type hint for value.
 
     Args:
@@ -229,7 +230,7 @@ def _generate_imports(
     """
     return [
         *[
-            ast.ImportFrom(module=name, names=[ast.alias(name=val) for val in values])
+            ast.ImportFrom(module=name, names=[ast.alias(name=val) for val in values])  # pyright: ignore [reportCallIssue]
             for name, values in DEFAULT_IMPORTS.items()
         ],
         ast.Import([ast.alias("reflex")]),
@@ -367,7 +368,7 @@ def _extract_class_props_as_ast_nodes(
                     # Try to get default from pydantic field definition.
                     default = target_class.__fields__[name].default
                     if isinstance(default, Var):
-                        default = default._decode()  # type: ignore
+                        default = default._decode()
 
             kwargs.append(
                 (
@@ -383,7 +384,7 @@ def _extract_class_props_as_ast_nodes(
     return kwargs
 
 
-def type_to_ast(typ, cls: type) -> ast.AST:
+def type_to_ast(typ: Any, cls: type) -> ast.AST:
     """Converts any type annotation into its AST representation.
     Handles nested generic types, unions, etc.
 
@@ -434,14 +435,16 @@ def type_to_ast(typ, cls: type) -> ast.AST:
     if len(arg_nodes) == 1:
         slice_value = arg_nodes[0]
     else:
-        slice_value = ast.Tuple(elts=arg_nodes, ctx=ast.Load())
+        slice_value = ast.Tuple(elts=arg_nodes, ctx=ast.Load())  # pyright: ignore [reportArgumentType]
 
     return ast.Subscript(
-        value=ast.Name(id=base_name), slice=ast.Index(value=slice_value), ctx=ast.Load()
+        value=ast.Name(id=base_name),
+        slice=ast.Index(value=slice_value),  # pyright: ignore [reportArgumentType]
+        ctx=ast.Load(),
     )
 
 
-def _get_parent_imports(func):
+def _get_parent_imports(func: Callable):
     _imports = {"reflex.vars": ["Var"]}
     for type_hint in inspect.get_annotations(func).values():
         try:
@@ -498,7 +501,7 @@ def _generate_component_create_functiondef(
 
     def figure_out_return_type(annotation: Any):
         if inspect.isclass(annotation) and issubclass(annotation, inspect._empty):
-            return ast.Name(id="EventType[..., BASE_STATE]")
+            return ast.Name(id="EventType[Any]")
 
         if not isinstance(annotation, str) and get_origin(annotation) is tuple:
             arguments = get_args(annotation)
@@ -514,8 +517,10 @@ def _generate_component_create_functiondef(
             # Get all prefixes of the type arguments
             all_count_args_type = [
                 ast.Name(
-                    f"EventType[[{', '.join([ast.unparse(arg) for arg in type_args[:i]])}], BASE_STATE]"
+                    f"EventType[{', '.join([ast.unparse(arg) for arg in type_args[:i]])}]"
                 )
+                if i > 0
+                else ast.Name("EventType[()]")
                 for i in range(len(type_args) + 1)
             ]
 
@@ -528,7 +533,7 @@ def _generate_component_create_functiondef(
             inside_of_tuple = annotation.removeprefix("Tuple[").removesuffix("]")
 
             if inside_of_tuple == "()":
-                return ast.Name(id="EventType[[], BASE_STATE]")
+                return ast.Name(id="EventType[()]")
 
             arguments = [""]
 
@@ -555,16 +560,16 @@ def _generate_component_create_functiondef(
             ]
 
             all_count_args_type = [
-                ast.Name(
-                    f"EventType[[{', '.join(arguments_without_var[:i])}], BASE_STATE]"
-                )
+                ast.Name(f"EventType[{', '.join(arguments_without_var[:i])}]")
+                if i > 0
+                else ast.Name("EventType[()]")
                 for i in range(len(arguments) + 1)
             ]
 
             return ast.Name(
                 id=f"Union[{', '.join(map(ast.unparse, all_count_args_type))}]"
             )
-        return ast.Name(id="EventType[..., BASE_STATE]")
+        return ast.Name(id="EventType[Any]")
 
     event_triggers = clz().get_event_triggers()
 
@@ -575,7 +580,7 @@ def _generate_component_create_functiondef(
                 arg=trigger,
                 annotation=ast.Subscript(
                     ast.Name("Optional"),
-                    ast.Index(  # type: ignore
+                    ast.Index(  # pyright: ignore [reportArgumentType]
                         value=ast.Name(
                             id=ast.unparse(
                                 figure_out_return_type(
@@ -618,10 +623,10 @@ def _generate_component_create_functiondef(
         defaults=[],
     )
 
-    definition = ast.FunctionDef(
+    definition = ast.FunctionDef(  # pyright: ignore [reportCallIssue]
         name="create",
         args=create_args,
-        body=[
+        body=[  # pyright: ignore [reportArgumentType]
             ast.Expr(
                 value=ast.Constant(
                     value=_generate_docstrings(
@@ -630,7 +635,7 @@ def _generate_component_create_functiondef(
                 ),
             ),
             ast.Expr(
-                value=ast.Ellipsis(),
+                value=ast.Constant(value=Ellipsis),
             ),
         ],
         decorator_list=[
@@ -641,7 +646,7 @@ def _generate_component_create_functiondef(
                 else [ast.Name(id="classmethod")]
             ),
         ],
-        lineno=node.lineno if node is not None else None,
+        lineno=node.lineno if node is not None else None,  # pyright: ignore [reportArgumentType]
         returns=ast.Constant(value=clz.__name__),
     )
     return definition
@@ -680,7 +685,7 @@ def _generate_staticmethod_call_functiondef(
             else []
         ),
     )
-    definition = ast.FunctionDef(
+    definition = ast.FunctionDef(  # pyright: ignore [reportCallIssue]
         name="__call__",
         args=call_args,
         body=[
@@ -690,11 +695,12 @@ def _generate_staticmethod_call_functiondef(
             ),
         ],
         decorator_list=[ast.Name(id="staticmethod")],
-        lineno=node.lineno if node is not None else None,
+        lineno=node.lineno if node is not None else None,  # pyright: ignore [reportArgumentType]
         returns=ast.Constant(
             value=_get_type_hint(
                 typing.get_type_hints(clz.__call__).get("return", None),
                 type_hint_globals,
+                is_optional=False,
             )
         ),
     )
@@ -726,17 +732,17 @@ def _generate_namespace_call_functiondef(
     clz = classes[clz_name]
 
     if not hasattr(clz.__call__, "__self__"):
-        return _generate_staticmethod_call_functiondef(node, clz, type_hint_globals)  # type: ignore
+        return _generate_staticmethod_call_functiondef(node, clz, type_hint_globals)  # pyright: ignore [reportArgumentType]
 
     # Determine which class is wrapped by the namespace __call__ method
     component_clz = clz.__call__.__self__
 
-    if clz.__call__.__func__.__name__ != "create":
+    if clz.__call__.__func__.__name__ != "create":  # pyright: ignore [reportFunctionMemberAccess]
         return None
 
     definition = _generate_component_create_functiondef(
         node=None,
-        clz=component_clz,  # type: ignore
+        clz=component_clz,  # pyright: ignore [reportArgumentType]
         type_hint_globals=type_hint_globals,
     )
     definition.name = "__call__"
@@ -816,7 +822,7 @@ class StubGenerator(ast.NodeTransformer):
             The modified Module node.
         """
         self.generic_visit(node)
-        return self._remove_docstring(node)  # type: ignore
+        return self._remove_docstring(node)  # pyright: ignore [reportReturnType]
 
     def visit_Import(
         self, node: ast.Import | ast.ImportFrom
@@ -914,7 +920,7 @@ class StubGenerator(ast.NodeTransformer):
             node.body.append(call_definition)
         if not node.body:
             # We should never return an empty body.
-            node.body.append(ast.Expr(value=ast.Ellipsis()))
+            node.body.append(ast.Expr(value=ast.Constant(value=Ellipsis)))
         self.current_class = None
         return node
 
@@ -941,9 +947,9 @@ class StubGenerator(ast.NodeTransformer):
             if node.name.startswith("_") and node.name != "__call__":
                 return None  # remove private methods
 
-            if node.body[-1] != ast.Expr(value=ast.Ellipsis()):
+            if node.body[-1] != ast.Expr(value=ast.Constant(value=Ellipsis)):
                 # Blank out the function body for public functions.
-                node.body = [ast.Expr(value=ast.Ellipsis())]
+                node.body = [ast.Expr(value=ast.Constant(value=Ellipsis))]
         return node
 
     def visit_Assign(self, node: ast.Assign) -> ast.Assign | None:
@@ -1050,7 +1056,7 @@ class PyiGenerator:
         pyi_path.write_text(pyi_content)
         logger.info(f"Wrote {relpath}")
 
-    def _get_init_lazy_imports(self, mod, new_tree):
+    def _get_init_lazy_imports(self, mod: tuple | ModuleType, new_tree: ast.AST):
         # retrieve the _SUBMODULES and _SUBMOD_ATTRS from an init file if present.
         sub_mods = getattr(mod, "_SUBMODULES", None)
         sub_mod_attrs = getattr(mod, "_SUBMOD_ATTRS", None)
@@ -1077,7 +1083,7 @@ class PyiGenerator:
                 + (
                     "  # type: ignore"
                     if mod in pyright_ignore_imports
-                    else "  # noqa"  # ignore ruff formatting here for cases like rx.list.
+                    else "  # noqa: F401"  # ignore ruff formatting here for cases like rx.list.
                     if isinstance(mod, tuple)
                     else ""
                 )
@@ -1136,7 +1142,7 @@ class PyiGenerator:
             if pyi_path:
                 self.written_files.append(pyi_path)
 
-    def scan_all(self, targets, changed_files: list[Path] | None = None):
+    def scan_all(self, targets: list, changed_files: list[Path] | None = None):
         """Scan all targets for class inheriting Component and generate the .pyi files.
 
         Args:

reflex/utils/registry.py

@@ -22,15 +22,15 @@ def latency(registry: str) -> int:
         return 10_000_000
 
 
-def average_latency(registry, attempts: int = 3) -> int:
+def average_latency(registry: str, attempts: int = 3) -> int:
     """Get the average latency of a registry.
 
     Args:
-        registry (str): The URL of the registry.
-        attempts (int): The number of attempts to make. Defaults to 10.
+        registry: The URL of the registry.
+        attempts: The number of attempts to make. Defaults to 10.
 
     Returns:
-        int: The average latency of the registry in microseconds.
+        The average latency of the registry in microseconds.
     """
     return sum(latency(registry) for _ in range(attempts)) // attempts
 

reflex/utils/serializers.py

@@ -476,7 +476,7 @@ try:
         base64_image = base64.b64encode(image_bytes).decode("utf-8")
         try:
             # Newer method to get the mime type, but does not always work.
-            mime_type = image.get_format_mimetype()  # type: ignore
+            mime_type = image.get_format_mimetype()  # pyright: ignore [reportAttributeAccessIssue]
         except AttributeError:
             try:
                 # Fallback method

reflex/utils/telemetry.py

@@ -122,7 +122,7 @@ def _prepare_event(event: str, **kwargs) -> dict:
         return {}
 
     if UTC is None:
-        # for python 3.9 & 3.10
+        # for python 3.10
         stamp = datetime.utcnow().isoformat()
     else:
         # for python 3.11 & 3.12
@@ -156,12 +156,13 @@ def _prepare_event(event: str, **kwargs) -> dict:
 def _send_event(event_data: dict) -> bool:
     try:
         httpx.post(POSTHOG_API_URL, json=event_data)
-        return True
     except Exception:
         return False
+    else:
+        return True
 
 
-def _send(event, telemetry_enabled, **kwargs):
+def _send(event: str, telemetry_enabled: bool | None, **kwargs):
     from reflex.config import get_config
 
     # Get the telemetry_enabled from the config if it is not specified.
@@ -188,7 +189,7 @@ def send(event: str, telemetry_enabled: bool | None = None, **kwargs):
         kwargs: Additional data to send with the event.
     """
 
-    async def async_send(event, telemetry_enabled, **kwargs):
+    async def async_send(event: str, telemetry_enabled: bool | None, **kwargs):
         return _send(event, telemetry_enabled, **kwargs)
 
     try: