remove psutil as a required dep (except in a windows) (#5498)

* remove psutil as a required dep (except in a windows)

* remove maximum for dev

* install psutil

Author: adhami3310

pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
   "jinja2 >=3.1.2,<4.0",
   "packaging >=24.2,<26",
   "platformdirs >=4.3.7,<5.0",
-  "psutil >=7.0.0,<8.0",
+  "psutil >=7.0.0,<8.0; sys_platform == 'win32'",
   "pydantic >=1.10.21,<3.0",
   "python-socketio >=5.12.0,<6.0",
   "python-multipart >=0.0.20,<1.0",
@@ -165,6 +165,7 @@ dev = [
   "playwright >=1.51",
   "plotly >=6.0",
   "pre-commit ==4.2.0",
+  "psutil >=7.0.0",
   "psycopg[binary] >=3.2",
   "pyright >=1.1.400",
   "pytest >=8.3",

reflex/istate/manager.py

@@ -179,6 +179,7 @@ def _default_token_expiration() -> int:
 
 def reset_disk_state_manager():
     """Reset the disk state manager."""
+    console.debug("Resetting disk state manager.")
     states_directory = prerequisites.get_states_dir()
     if states_directory.exists():
         for path in states_directory.iterdir():

reflex/testing.py

@@ -24,7 +24,6 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Literal, TypeVar
 
-import psutil
 import uvicorn
 
 import reflex
@@ -478,6 +477,8 @@ def __enter__(self) -> AppHarness:
 
     def stop(self) -> None:
         """Stop the frontend and backend servers."""
+        import psutil
+
         # Quit browsers first to avoid any lingering events being sent during shutdown.
         for driver in self._frontends:
             driver.quit()

reflex/utils/exec.py

@@ -15,8 +15,6 @@
 from typing import Any, NamedTuple, TypedDict
 from urllib.parse import urljoin
 
-import psutil
-
 from reflex import constants
 from reflex.config import get_config
 from reflex.constants.base import LogLevel
@@ -131,12 +129,16 @@ def get_changes(old: dict[str, str], new: dict[str, str]) -> Change:
 def kill(proc_pid: int):
     """Kills a process and all its child processes.
 
+    Requires the `psutil` library to be installed.
+
     Args:
-        proc_pid (int): The process ID of the process to be killed.
+        proc_pid: The process ID of the process to be killed.
 
     Example:
         >>> kill(1234)
     """
+    import psutil
+
     process = psutil.Process(proc_pid)
     for proc in process.children(recursive=True):
         proc.kill()

reflex/utils/prerequisites.py

@@ -116,7 +116,7 @@ def check_latest_package_version(package_name: str):
         # Get the latest version from PyPI
         current_version = importlib.metadata.version(package_name)
         url = f"https://pypi.org/pypi/{package_name}/json"
-        response = net.get(url)
+        response = net.get(url, timeout=2)
         latest_version = response.json()["info"]["version"]
         console.debug(f"Latest version of {package_name}: {latest_version}")
         if get_or_set_last_reflex_version_check_datetime():

reflex/utils/processes.py

@@ -4,17 +4,17 @@
 
 import collections
 import contextlib
-import importlib.metadata
 import os
 import signal
+import socket
 import subprocess
 from collections.abc import Callable, Generator, Sequence
 from concurrent import futures
+from contextlib import closing
 from pathlib import Path
 from typing import Any, Literal, overload
 
 import click
-import psutil
 from redis.exceptions import RedisError
 from rich.progress import Progress
 
@@ -52,29 +52,6 @@ def get_num_workers() -> int:
     return (os.cpu_count() or 1) * 2 + 1
 
 
-def get_process_on_port(port: int) -> psutil.Process | None:
-    """Get the process on the given port.
-
-    Args:
-        port: The port.
-
-    Returns:
-        The process on the given port.
-    """
-    for proc in psutil.process_iter(["pid", "name", "cmdline"]):
-        with contextlib.suppress(
-            psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess
-        ):
-            if importlib.metadata.version("psutil") >= "6.0.0":
-                conns = proc.net_connections(kind="inet")
-            else:
-                conns = proc.connections(kind="inet")
-            for conn in conns:
-                if conn.laddr.port == int(port):
-                    return proc
-    return None
-
-
 def is_process_on_port(port: int) -> bool:
     """Check if a process is running on the given port.
 
@@ -84,18 +61,8 @@ def is_process_on_port(port: int) -> bool:
     Returns:
         Whether a process is running on the given port.
     """
-    return get_process_on_port(port) is not None
-
-
-def kill_process_on_port(port: int):
-    """Kill the process on the given port.
-
-    Args:
-        port: The port.
-    """
-    if get_process_on_port(port) is not None:
-        with contextlib.suppress(psutil.AccessDenied):
-            get_process_on_port(port).kill()  # pyright: ignore [reportOptionalMemberAccess]
+    with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
+        return sock.connect_ex(("127.0.0.1", port)) == 0
 
 
 def change_port(port: int, _type: str) -> int:
@@ -133,13 +100,13 @@ def handle_port(service_name: str, port: int, auto_increment: bool) -> int:
     Raises:
         Exit:when the port is in use.
     """
-    if (process := get_process_on_port(port)) is None:
+    console.debug(f"Checking if {service_name.capitalize()} port: {port} is in use.")
+    if not is_process_on_port(port):
+        console.debug(f"{service_name.capitalize()} port: {port} is not in use.")
         return port
     if auto_increment:
         return change_port(port, service_name)
-    console.error(
-        f"{service_name.capitalize()} port: {port} is already in use by PID: {process.pid}."
-    )
+    console.error(f"{service_name.capitalize()} port: {port} is already in use.")
     raise click.exceptions.Exit
 
 

reflex/utils/telemetry.py

@@ -13,7 +13,6 @@
 from typing import TypedDict
 
 import httpx
-import psutil
 
 from reflex import constants
 from reflex.environment import environment
@@ -89,18 +88,6 @@ def get_reflex_enterprise_version() -> str | None:
         return None
 
 
-def get_memory() -> int:
-    """Get the total memory in MB.
-
-    Returns:
-        The total memory in MB.
-    """
-    try:
-        return psutil.virtual_memory().total >> 20
-    except ValueError:  # needed to pass ubuntu test
-        return 0
-
-
 def _raise_on_missing_project_hash() -> bool:
     """Check if an error should be raised when project hash is missing.
 
@@ -128,7 +115,6 @@ class _Properties(TypedDict):
     bun_version: str | None
     reflex_enterprise_version: str | None
     cpu_count: int
-    memory: int
     cpu_info: dict
 
 
@@ -182,7 +168,6 @@ def _get_event_defaults() -> _DefaultEvent | None:
             ),
             "reflex_enterprise_version": get_reflex_enterprise_version(),
             "cpu_count": get_cpu_count(),
-            "memory": get_memory(),
             "cpu_info": dataclasses.asdict(cpuinfo) if cpuinfo else {},
         },
     }

scripts/wait_for_listening_port.py

@@ -9,17 +9,12 @@
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
-# psutil is already a dependency of Reflex itself - so it's OK to use
-import psutil
-
 
 def _pid_exists(pid: int):
-    # os.kill(pid, 0) doesn't work on Windows (actually kills the PID)
-    # psutil.pid_exists() doesn't work on Windows (does os.kill underneath)
-    # psutil.pids() seems to return the right thing. Inefficient but doesn't matter - keeps things simple.
-    #
     # Note: For windows, the pid here is really the "winpid".
-    return pid in psutil.pids()
+    import psutil
+
+    return psutil.pid_exists(pid)
 
 
 def _wait_for_port(port: int, server_pid: int, timeout: float) -> tuple[bool, str]:

tests/integration/init-test/in_docker_test_script.sh

@@ -31,6 +31,7 @@ pip install -U pip
 echo "Installing reflex from local repo code"
 cp -r /reflex-repo ~/reflex-repo
 pip install ~/reflex-repo
+pip install psutil
 
 redis-server &
 

tests/units/test_telemetry.py

@@ -15,9 +15,6 @@ def test_telemetry():
     # Check that the CPU count and memory are greater than 0.
     assert telemetry.get_cpu_count() > 0
 
-    # Check that the available memory is greater than 0
-    assert telemetry.get_memory() > 0
-
     # Check that the Reflex version is not None.
     assert telemetry.get_reflex_version() is not None