feat: add devices property to Info

blacktop · Apr 10, 2023 · 512fd3a · 512fd3a
1 parent be25b71
commit 512fd3a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -41,9 +41,12 @@ import ipsw
 client = ipsw.IpswClient(base_url='tcp://127.0.0.1:8080')
 info = client.info.get("iPhone15,2_16.5_20F5028e_Restore.ipsw")
 print(f'{info.version} ({info.build})')
+for device in info.devices:
+    print(f'- {device}')
 ```
 ```bash
 16.5 (20F5028e)
+- iPhone 14 Pro
 ```
 
 ## Community

diff --git a/ipsw/models/info.py b/ipsw/models/info.py
@@ -29,6 +29,20 @@ def build(self):
         The iOS version.
         """
         return self.attrs["info"]["Plists"]["restore"].get("ProductBuildVersion", None)
+
+    @property
+    def devices(self):
+        """
+        The iOS devices.
+        """
+        devices = set()
+        for dt in self.attrs['info']['DeviceTrees'].values():
+            for child in dt['device-tree']['children']:
+                if 'product' in child:
+                    devices.add(child['product']['product-name'])
+        devlist = list(devices)
+        devlist.sort()
+        return devlist
 
 
 class InfoCollection(Collection):