Add NPSClient.get()

src/lvmnps/nps/core.py

@@ -105,6 +105,16 @@ async def refresh(self):
 
         pass
 
+    def get(self, outlet: int | str):
+        """Retrieves an outlet by ID or name."""
+
+        if isinstance(outlet, int):
+            return get_outlet_by_id(self.outlets, outlet)
+        elif isinstance(outlet, str):
+            return get_outlet_by_name(self.outlets, outlet)
+        else:
+            raise TypeError("Invalid outlet type. Only int and str are allowed.")
+
     async def set_state(
         self,
         outlets: OutletArgType,

tests/test_nps.py

@@ -10,7 +10,7 @@
 
 import pytest
 
-from lvmnps.nps.core import NPSClient
+from lvmnps.nps.core import NPSClient, OutletModel
 
 
 async def test_npsclient(nps_test_client: NPSClient):
@@ -69,3 +69,13 @@ async def test_outlet_cycle(nps_test_client: NPSClient):
     await nps_test_client.cycle(outlet, delay=0.1)
 
     assert outlet.state is True
+
+
+@pytest.mark.parametrize("outlet", [1, "test_1"])
+async def test_client_get(nps_test_client: NPSClient, outlet: str | int):
+    assert isinstance(nps_test_client.get(outlet), OutletModel)
+
+
+async def test_client_get_invalid_type(nps_test_client: NPSClient):
+    with pytest.raises(TypeError):
+        nps_test_client.get(None)  # type: ignore