Add find_device_by_identifier (#94)

CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+## 0.2.0 - 2021-10-07
+
+### Added
+
+- Add `find_device_by_identifier` to discovered
+
 ## 0.1.1 - 2021-10-07
 
 ### Changed

iolite_client/client.py

@@ -83,6 +83,22 @@ def find_room_by_name(self, name: str) -> Optional[Room]:
         """
         return self._find_room_by_attribute_value("name", name)
 
+    def find_device_by_identifier(self, identifier: str) -> Optional[Device]:
+        """Find a device by identifier.
+        :param identifier: The identifier of the device
+        :return: The matched device or None
+        """
+        for room in self.discovered_rooms.values():
+            if room.devices.get(identifier):
+                return room.devices.get(identifier)
+
+        for devices in self.unmapped_entities.values():
+            for device in devices:
+                if device.identifier == identifier:
+                    return device
+
+        return None
+
     def _find_room_by_attribute_value(
         self, attribute: str, value: str
     ) -> Optional[Room]:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "iolite-client"
-version = "0.1.1"
+version = "0.2.0"
 description = "API client for interacting with IOLite's remote API"
 authors = ["Malachi Soord <[email protected]>"]
 license = "MIT"

test/test_client.py

@@ -73,6 +73,27 @@ def test_add_heating_unmapped(self):
         self.assertTrue(len(self.discovered.unmapped_entities) == 0)
         self.assertEqual(self.bedroom_heating, self.discovered.get_rooms()[0].heating)
 
+    def test_find_device_by_identifier(self):
+        self.discovered.add_room(self.bedroom)
+        self.discovered.add_device(self.bedroom_switch)
+        self.assertEqual(
+            self.bedroom_switch,
+            self.discovered.find_device_by_identifier(self.bedroom_switch.identifier),
+        )
+
+    def test_find_device_by_identifier_unmapped(self):
+        self.discovered.add_device(self.bedroom_switch)
+        self.assertEqual(
+            self.bedroom_switch,
+            self.discovered.find_device_by_identifier(self.bedroom_switch.identifier),
+        )
+
+    def test_find_device_by_identifier_not_found(self):
+        self.assertEqual(
+            None,
+            self.discovered.find_device_by_identifier(self.bedroom_switch.identifier),
+        )
+
 
 if __name__ == "__main__":
     unittest.main()