Merge pull request #1189 from doronz88/feature/diagnostics-wifi

diagnostics: add `get_wifi()`
doronz88 · Sep 9, 2024 · 2f3b257 · 2f3b257
2 parents 21bbd04 + f3285f2
commit 2f3b257
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pymobiledevice3/cli/diagnostics.py b/pymobiledevice3/cli/diagnostics.py
@@ -5,6 +5,7 @@
 
 from pymobiledevice3.cli.cli_common import Command, print_json
 from pymobiledevice3.lockdown import LockdownClient
+from pymobiledevice3.lockdown_service_provider import LockdownServiceProvider
 from pymobiledevice3.services.diagnostics import DiagnosticsService
 
 logger = logging.getLogger(__name__)
@@ -89,3 +90,10 @@ def diagnostics_battery_monitor(service_provider: LockdownClient):
         }
         logger.info(info)
         time.sleep(1)
+
+
+@diagnostics.command('wifi', cls=Command)
+def diagnostics_wifi(service_provider: LockdownServiceProvider) -> None:
+    """ Query WiFi info from IORegistry """
+    raw_info = DiagnosticsService(lockdown=service_provider).get_wifi()
+    print_json(raw_info)
diff --git a/pymobiledevice3/services/diagnostics.py b/pymobiledevice3/services/diagnostics.py
@@ -1027,3 +1027,6 @@ def ioregistry(self, plane: str = None, name: str = None, ioclass: str = None):
 
     def get_battery(self) -> Mapping:
         return self.ioregistry(ioclass='IOPMPowerSource')
+
+    def get_wifi(self) -> Mapping:
+        return self.ioregistry(name='AppleBCMWLANSkywalkInterface')