From e2c09c9d7bd13b5a21f75217f8542d6880236ec4 Mon Sep 17 00:00:00 2001 From: doronz88 Date: Tue, 3 Sep 2024 10:28:14 +0300 Subject: [PATCH] restore: fix restoring ios18 on python versions<3.11 --- pymobiledevice3/restore/restore.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pymobiledevice3/restore/restore.py b/pymobiledevice3/restore/restore.py index 4a2c5dbdc..8f8aeee3a 100644 --- a/pymobiledevice3/restore/restore.py +++ b/pymobiledevice3/restore/restore.py @@ -255,10 +255,9 @@ async def send_source_boot_object_v4(self, message: Mapping) -> None: if i == 0 and chunk.startswith(b'AEA1'): self.logger.debug('First chunk in a AEA') try: - async with asyncio.timeout(3): - message = await service.aio_recv_plist() - await self.send_url_asset(message) - except asyncio.TimeoutError: + message = await asyncio.wait_for(service.aio_recv_plist(), timeout=3) + await self.send_url_asset(message) + except asyncio.exceptions.TimeoutError: self.logger.debug('No URLAsset was requested. Assuming it is not necessary') # Send FileDataDone