pymobiledevice3 4.27.4__py3-none-any.whl → 5.1.2__py3-none-any.whl

pymobiledevice3/tunneld/api.py

@@ -6,22 +6,25 @@ from pymobiledevice3.exceptions import TunneldConnectionError
 from pymobiledevice3.remote.remote_service_discovery import RemoteServiceDiscoveryService
 from pymobiledevice3.utils import get_asyncio_loop
 
-TUNNELD_DEFAULT_ADDRESS = ('127.0.0.1', 49151)
+TUNNELD_DEFAULT_ADDRESS = ("127.0.0.1", 49151)
 
 
-async def async_get_tunneld_devices(tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS) \
-        -> list[RemoteServiceDiscoveryService]:
+async def async_get_tunneld_devices(
+    tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS,
+) -> list[RemoteServiceDiscoveryService]:
     tunnels = _list_tunnels(tunneld_address)
     return await _create_rsds_from_tunnels(tunnels)
 
 
-def get_tunneld_devices(tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS) \
-        -> list[RemoteServiceDiscoveryService]:
+def get_tunneld_devices(
+    tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS,
+) -> list[RemoteServiceDiscoveryService]:
     return get_asyncio_loop().run_until_complete(async_get_tunneld_devices(tunneld_address))
 
 
-async def async_get_tunneld_device_by_udid(udid: str, tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS) \
-        -> Optional[RemoteServiceDiscoveryService]:
+async def async_get_tunneld_device_by_udid(
+    udid: str, tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS
+) -> Optional[RemoteServiceDiscoveryService]:
     tunnels = _list_tunnels(tunneld_address)
     if udid not in tunnels:
         return None
@@ -29,27 +32,29 @@ async def async_get_tunneld_device_by_udid(udid: str, tunneld_address: tuple[str
     return rsds[0]
 
 
-def get_tunneld_device_by_udid(udid: str, tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS) \
-        -> Optional[RemoteServiceDiscoveryService]:
+def get_tunneld_device_by_udid(
+    udid: str, tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS
+) -> Optional[RemoteServiceDiscoveryService]:
     return get_asyncio_loop().run_until_complete(async_get_tunneld_device_by_udid(udid, tunneld_address))
 
 
 def _list_tunnels(tunneld_address: tuple[str, int] = TUNNELD_DEFAULT_ADDRESS) -> dict[str, list[dict]]:
     try:
         # Get the list of tunnels from the specified address
-        resp = requests.get(f'http://{tunneld_address[0]}:{tunneld_address[1]}')
+        resp = requests.get(f"http://{tunneld_address[0]}:{tunneld_address[1]}")
         tunnels = resp.json()
-    except requests.exceptions.ConnectionError:
-        raise TunneldConnectionError()
+    except requests.exceptions.ConnectionError as e:
+        raise TunneldConnectionError() from e
     return tunnels
 
 
 async def _create_rsds_from_tunnels(tunnels: dict[str, list[dict]]) -> list[RemoteServiceDiscoveryService]:
     rsds = []
-    for udid, details in tunnels.items():
+    for _udid, details in tunnels.items():
         for tunnel_details in details:
-            rsd = RemoteServiceDiscoveryService((tunnel_details['tunnel-address'], tunnel_details['tunnel-port']),
-                                                name=tunnel_details['interface'])
+            rsd = RemoteServiceDiscoveryService(
+                (tunnel_details["tunnel-address"], tunnel_details["tunnel-port"]), name=tunnel_details["interface"]
+            )
             try:
                 await rsd.connect()
                 rsds.append(rsd)