pymobiledevice3 4.14.6__py3-none-any.whl → 7.0.6__py3-none-any.whl

pymobiledevice3/services/power_assertion.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import contextlib
+from typing import Optional
 
 from pymobiledevice3.lockdown import LockdownClient
 from pymobiledevice3.lockdown_service_provider import LockdownServiceProvider
@@ -7,8 +8,8 @@ from pymobiledevice3.services.lockdown_service import LockdownService
 
 
 class PowerAssertionService(LockdownService):
-    RSD_SERVICE_NAME = 'com.apple.mobile.assertion_agent.shim.remote'
-    SERVICE_NAME = 'com.apple.mobile.assertion_agent'
+    RSD_SERVICE_NAME = "com.apple.mobile.assertion_agent.shim.remote"
+    SERVICE_NAME = "com.apple.mobile.assertion_agent"
 
     def __init__(self, lockdown: LockdownServiceProvider):
         if isinstance(lockdown, LockdownClient):
@@ -17,17 +18,17 @@ class PowerAssertionService(LockdownService):
             super().__init__(lockdown, self.RSD_SERVICE_NAME)
 
     @contextlib.contextmanager
-    def create_power_assertion(self, type_: str, name: str, timeout: float, details: str = None):
-        """ Trigger IOPMAssertionCreateWithName """
+    def create_power_assertion(self, type_: str, name: str, timeout: float, details: Optional[str] = None):
+        """Trigger IOPMAssertionCreateWithName"""
         msg = {
-            'CommandKey': 'CommandCreateAssertion',
-            'AssertionTypeKey': type_,
-            'AssertionNameKey': name,
-            'AssertionTimeoutKey': timeout,
+            "CommandKey": "CommandCreateAssertion",
+            "AssertionTypeKey": type_,
+            "AssertionNameKey": name,
+            "AssertionTimeoutKey": timeout,
         }
 
         if details is not None:
-            msg['AssertionDetailKey'] = details
+            msg["AssertionDetailKey"] = details
 
         self.service.send_recv_plist(msg)
         yield

pymobiledevice3/services/preboard.py

@@ -6,8 +6,8 @@ from pymobiledevice3.services.lockdown_service import LockdownService
 
 
 class PreboardService(LockdownService):
-    RSD_SERVICE_NAME = 'com.apple.preboardservice_v2.shim.remote'
-    SERVICE_NAME = 'com.apple.preboardservice_v2'
+    RSD_SERVICE_NAME = "com.apple.preboardservice_v2.shim.remote"
+    SERVICE_NAME = "com.apple.preboardservice_v2"
 
     def __init__(self, lockdown: LockdownServiceProvider):
         if isinstance(lockdown, LockdownClient):
@@ -16,7 +16,7 @@ class PreboardService(LockdownService):
             super().__init__(lockdown, self.RSD_SERVICE_NAME)
 
     def create_stashbag(self, manifest):
-        return self.service.send_recv_plist({'Command': 'CreateStashbag', 'Manifest': manifest})
+        return self.service.send_recv_plist({"Command": "CreateStashbag", "Manifest": manifest})
 
     def commit(self, manifest):
-        return self.service.send_recv_plist({'Command': 'CommitStashbag', 'Manifest': manifest})
+        return self.service.send_recv_plist({"Command": "CommitStashbag", "Manifest": manifest})

pymobiledevice3/services/remote_fetch_symbols.py

@@ -15,32 +15,39 @@ class DSCFile:
 
 
 class RemoteFetchSymbolsService(RemoteService):
-    SERVICE_NAME = 'com.apple.dt.remoteFetchSymbols'
+    SERVICE_NAME = "com.apple.dt.remoteFetchSymbols"
 
     def __init__(self, rsd: RemoteServiceDiscoveryService):
         super().__init__(rsd, self.SERVICE_NAME)
 
     async def get_dsc_file_list(self) -> list[DSCFile]:
         files: list[DSCFile] = []
-        response = await self.service.send_receive_request({'XPCDictionary_sideChannel': uuid.uuid4(), 'DSCFilePaths': []})
-        file_count = response['DSCFilePaths']
-        for i in range(file_count):
+        response = await self.service.send_receive_request({
+            "XPCDictionary_sideChannel": uuid.uuid4(),
+            "DSCFilePaths": [],
+        })
+        file_count = response["DSCFilePaths"]
+        for _i in range(file_count):
             response = await self.service.receive_response()
-            response = response['DSCFilePaths']
-            file_transfer = response['fileTransfer']
-            expected_length = file_transfer['expectedLength']
-            file_path = response['filePath']
+            response = response["DSCFilePaths"]
+            file_transfer = response["fileTransfer"]
+            expected_length = file_transfer["expectedLength"]
+            file_path = response["filePath"]
             files.append(DSCFile(file_path=file_path, file_size=expected_length))
         return files
 
     async def download(self, out: Path) -> None:
         files = await self.get_dsc_file_list()
-        for i, file in enumerate(files):
-            self.logger.info(f'Downloading {file}')
-            out_file = out / file.file_path[1:]  # trim the "/" prefix
-            out_file.parent.mkdir(parents=True, exist_ok=True)
-            with open(out_file, 'wb') as f:
-                with tqdm(total=files[i].file_size, dynamic_ncols=True) as pb:
+        with tqdm(total=len(files), dynamic_ncols=True, desc="Downloading DSC") as total_pb:
+            for i, file in enumerate(files):
+                total_pb.set_description("Downloading DSC")
+                out_file = out / file.file_path[1:]  # trim the "/" prefix
+                out_file.parent.mkdir(parents=True, exist_ok=True)
+                with (
+                    open(out_file, "wb") as f,
+                    tqdm(total=files[i].file_size, dynamic_ncols=True, desc=file.file_path.rsplit("/", 1)[-1]) as pb,
+                ):
                     async for chunk in self.service.iter_file_chunks(files[i].file_size, file_idx=i):
                         f.write(chunk)
                         pb.update(len(chunk))
+                total_pb.update(1)