dvershinin · May 18, 2024 11:47 · May 18, 2024
diff --git a/reset_network_adapter.py b/reset_network_adapter.py
@@ -0,0 +1,61 @@
+#!/usr/bin/python3
+
+import subprocess
+
+REALTEK_PRODUCT_ID = "0bda:8153"
+UHUBCTL_PATH = "/opt/homebrew/bin/uhubctl"
+
+
+def log_message(message):
+    """ Log messages to the system log using logger. """
+    subprocess.run(['/usr/bin/logger', '-p', 'user.info', '-t', 'networkAdapterReset',
+                    f"[networkAdapterReset]: {message}"])
+    # Print to stderr as well
+    print(message)
+
+
+def find_realtek_device():
+    """ Finds the hub and port for the Realtek device using uhubctl. """
+    try:
+        result = subprocess.run([UHUBCTL_PATH], stdout=subprocess.PIPE, text=True)
+        lines = result.stdout.splitlines()
+
+        current_hub_id = None
+        for i, line in enumerate(lines):
+            # e.g. first line in hub information: "Current status for hub 0-2 ..."
+            if "Current status for hub" in line:
+                current_hub_id = line.split()[4]
+            if REALTEK_PRODUCT_ID in line:  # This is the product ID for the Realtek device
+                # E.g. '  Port 4: 0203 power 5gbps U0 enable connect [0bda:8153 Realtek USB 10/100/1000 LAN 001000001]'
+                port_line = line.strip()
+                port_id = port_line.split()[1].split(':')[0]
+                return current_hub_id, port_id
+    except Exception as e:
+        log_message(f"Error finding Realtek device: {str(e)}")
+    return None, None
+
+
+def power_cycle_usb(hub_id, port_id):
+    """ Power cycles the USB port. """
+    try:
+        # Turning off the power to the port
+        run_args = [UHUBCTL_PATH, '--location', hub_id, '--ports', port_id, '--action',
+                    'cycle', '--delay', '1']
+        subprocess.run(run_args, check=True)
+        log_message(f"Successfully power cycled hub {hub_id}, port {port_id}.")
+    except subprocess.CalledProcessError as e:
+        log_message(f"Failed to toggle USB: {str(e)}")
+
+
+def main():
+    log_message("Starting network adapter reset script.")
+    hub_id, port_id = find_realtek_device()
+    if hub_id and port_id:
+        log_message(f"Detected hub: {hub_id}, port: {port_id}")
+        power_cycle_usb(hub_id, port_id)
+    else:
+        log_message("Realtek USB 10/100/1000 LAN device not found.")
+
+
+if __name__ == "__main__":
+    main()