add stream and dead sensors

Author: InTheDaylight14

custom_components/npm_switches/sensor.py

@@ -18,12 +18,12 @@ async def async_setup_entry(hass, entry, async_add_entities):
         if entry.data["include_redirection_hosts"]:
             entities.append(NpmSwitchesRedirSensor(coordinator, entry, "enabled"))
             entities.append(NpmSwitchesRedirSensor(coordinator, entry, "disabled"))
-        # if entry.data["include_stream_hosts"]:
-        #     entities.append(NpmSwitchesStreamSensor(coordinator, entry, "enabled"))
-        #     entities.append(NpmSwitchesStreamSensor(coordinator, entry, "disabled"))
-        # if entry.data["include_dead_hosts"]:
-        #     entities.append(NpmSwitchesDeadSensor(coordinator, entry, "enabled"))
-        #     entities.append(NpmSwitchesDeadSensor(coordinator, entry, "disabled"))
+        if entry.data["include_stream_hosts"]:
+            entities.append(NpmSwitchesStreamSensor(coordinator, entry, "enabled"))
+            entities.append(NpmSwitchesStreamSensor(coordinator, entry, "disabled"))
+        if entry.data["include_dead_hosts"]:
+            entities.append(NpmSwitchesDeadSensor(coordinator, entry, "enabled"))
+            entities.append(NpmSwitchesDeadSensor(coordinator, entry, "disabled"))
 
     async_add_entities(entities, True)
 
@@ -71,8 +71,8 @@ def __init__(
     ) -> None:
         """Initialize proxy switch entity."""
         super().__init__(coordinator, entry)
-        self.redir_id = name  # Unique ID relies on self.redir_id
-        self.sensor_name = self.redir_id
+        self.host_id = name  # Unique ID relies on self.host_id
+        self.sensor_name = self.host_id
         self.friendly_name = "NPM Redirection Hosts " + self.sensor_name.capitalize()
 
     # @property
@@ -87,6 +87,70 @@ def native_value(self):
             return self.coordinator.api.num_redir_enabled
         return self.coordinator.api.num_redir_disabled
 
+    @property
+    def icon(self):
+        """Return the icon of the sensor."""
+        return "mdi:counter"
+
+class NpmSwitchesStreamSensor(NpmSwitchesEntity, SensorEntity):
+    """NPM Switches Stream Sensor class."""
+
+    def __init__(
+        self,
+        coordinator: NpmSwitchesUpdateCoordinator,
+        entry: ConfigEntry,
+        name: str,
+    ) -> None:
+        """Initialize proxy switch entity."""
+        super().__init__(coordinator, entry)
+        self.host_id = name  # Unique ID relies on self.host_id
+        self.sensor_name = self.host_id
+        self.friendly_name = "NPM Steam Hosts " + self.sensor_name.capitalize()
+
+    # @property
+    # def name(self):
+    #     """Return the name of the sensor."""
+    #     return "npm_" + self.sensor_name + "_proxy_hosts"
+
+    @property
+    def native_value(self):
+        """Return the native value of the sensor."""
+        if self.sensor_name == "enabled":
+            return self.coordinator.api.num_stream_enabled
+        return self.coordinator.api.num_stream_disabled
+
+    @property
+    def icon(self):
+        """Return the icon of the sensor."""
+        return "mdi:counter"
+
+class NpmSwitchesDeadSensor(NpmSwitchesEntity, SensorEntity):
+    """NPM Switches Deam Sensor class."""
+
+    def __init__(
+        self,
+        coordinator: NpmSwitchesUpdateCoordinator,
+        entry: ConfigEntry,
+        name: str,
+    ) -> None:
+        """Initialize proxy switch entity."""
+        super().__init__(coordinator, entry)
+        self.host_id = name  # Unique ID relies on self.host_id
+        self.sensor_name = self.host_id
+        self.friendly_name = "NPM Dead Hosts " + self.sensor_name.capitalize()
+
+    # @property
+    # def name(self):
+    #     """Return the name of the sensor."""
+    #     return "npm_" + self.sensor_name + "_proxy_hosts"
+
+    @property
+    def native_value(self):
+        """Return the native value of the sensor."""
+        if self.sensor_name == "enabled":
+            return self.coordinator.api.num_dead_enabled
+        return self.coordinator.api.num_dead_disabled
+
     @property
     def icon(self):
         """Return the icon of the sensor."""