From 3c0cb3fd6afb68d155057f193fb28522dc575592 Mon Sep 17 00:00:00 2001
From: benedikt-bartscher
 <31854409+benedikt-bartscher@users.noreply.github.com>
Date: Thu, 11 Jul 2024 20:21:34 +0200
Subject: [PATCH] migrate deprecated connections to net_connections (#3641)

* migrate deprecated connections to net_connections

* check for psutil version
---
 reflex/utils/processes.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/reflex/utils/processes.py b/reflex/utils/processes.py
index ea94cc2ff..c435af7d0 100644
--- a/reflex/utils/processes.py
+++ b/reflex/utils/processes.py
@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import collections
 import contextlib
+import importlib.metadata
 import os
 import signal
 import subprocess
@@ -58,8 +59,12 @@ def get_process_on_port(port) -> Optional[psutil.Process]:
     """
     for proc in psutil.process_iter(["pid", "name", "cmdline"]):
         try:
-            for conns in proc.connections(kind="inet"):
-                if conns.laddr.port == int(port):
+            if importlib.metadata.version("psutil") >= "6.0.0":
+                conns = proc.net_connections(kind="inet")  # type: ignore
+            else:
+                conns = proc.connections(kind="inet")
+            for conn in conns:
+                if conn.laddr.port == int(port):
                     return proc
         except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
             pass