Fixes issue for switch entity index (#18)

augustas2 · Jan 13, 2023 · 0d5339c · 0d5339c
1 parent 0e49511
commit 0d5339c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/eldes_alarm/switch.py b/custom_components/eldes_alarm/switch.py
@@ -75,19 +75,23 @@ def icon(self):
 
     async def async_turn_on(self):
         """Turn the entity on."""
+        output = self.data["outputs"][self.entity_index]
+
         await self.client.turn_on_output(
             self.imei,
-            self.entity_index
+            output['id']
         )
 
         self.data["outputs"][self.entity_index]["outputState"] = True
         self.async_write_ha_state()
 
     async def async_turn_off(self):
         """Turn the entity off."""
+        output = self.data["outputs"][self.entity_index]
+
         await self.client.turn_off_output(
             self.imei,
-            self.entity_index
+            output['id']
         )
 
         self.data["outputs"][self.entity_index]["outputState"] = False