change host type specific data when enabling or disabling

Author: InTheDaylight14

custom_components/npm_switches/api.py

@@ -177,7 +177,14 @@ async def enable_host(self, host_id: str, host_type: str) -> None:
         response = await self.api_wrapper("post", url, headers=self._headers)
 
         if response is True:
-            self.proxy_hosts_data[host_id]["enabled"] = 1
+            if host_type is "proxy-hosts":
+                self.proxy_hosts_data[host_id]["enabled"] = 1
+            if host_type is "redirection-hosts":
+                self.redir_hosts_data[host_id]["enabled"] = 1
+            if host_type is "streams":
+                self.stream_hosts_data[host_id]["enabled"] = 1
+            if host_type is "dead-hosts":
+                self.dead_hosts_data[host_id]["enabled"] = 1
         elif "error" in response.keys():
             _LOGGER.error(
                 "Error enabling host type %s host id %s. Error message: '%s'",
@@ -193,7 +200,14 @@ async def disable_host(self, host_id: str, host_type: str) -> None:
 
         response = await self.api_wrapper("post", url, headers=self._headers)
         if response is True:
-            self.proxy_hosts_data[host_id]["enabled"] = 0
+            if host_type is "proxy-hosts":
+                self.proxy_hosts_data[host_id]["enabled"] = 0
+            if host_type is "redirection-hosts":
+                self.redir_hosts_data[host_id]["enabled"] = 0
+            if host_type is "streams":
+                self.stream_hosts_data[host_id]["enabled"] = 0
+            if host_type is "dead-hosts":
+                self.dead_hosts_data[host_id]["enabled"] = 0
         elif "error" in response.keys():
             _LOGGER.error(
                 "Error enabling host type %s host id %s. Error message: '%s'",
@@ -215,16 +229,16 @@ def is_host_enabled(self, host_id: str, host_type: str) -> bool:
                 return True
             else:
                 return False
-        # elif host_type == "streams":
-            # if self.stream_hosts_data[host_id]["enabled"] == 1:
-            #     return True
-            # else:
-            #     return False
-        # elif host_type == "dead-hosts":
-            # if self.dead_hosts_data[host_id]["enabled"] == 1:
-            #     return True
-            # else:
-            #     return False
+        elif host_type == "streams":
+            if self.stream_hosts_data[host_id]["enabled"] == 1:
+                return True
+            else:
+                return False
+        elif host_type == "dead-hosts":
+            if self.dead_hosts_data[host_id]["enabled"] == 1:
+                return True
+            else:
+                return False
         else:
             return None
     @property