maint: Remove duplicate code

RogerSelwyn · Apr 24, 2024 · fb4f67b · fb4f67b
1 parent bae957a
commit fb4f67b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/custom_components/mqtt_discoverystream_alt/classes/base_entity.py b/custom_components/mqtt_discoverystream_alt/classes/base_entity.py
@@ -21,6 +21,8 @@
 class DiscoveryEntity:
     """Base discovery entity class."""
 
+    PUBLISH_STATE = False
+
     def __init__(self, hass, publish_retain, platform, publish_state):
         """Initialise the climate class."""
         self._hass = hass

diff --git a/custom_components/mqtt_discoverystream_alt/discovery.py b/custom_components/mqtt_discoverystream_alt/discovery.py
@@ -238,10 +238,9 @@ def _build_discovery_class(self, entity_type):
         """Build the discovery class."""
         modulename = __name__.removesuffix(".discovery")
         module = importlib.import_module(f".classes.{entity_type}", package=modulename)
-        if hasattr(module.DiscoveryItem, "PUBLISH_STATE"):
-            publish_state = module.DiscoveryItem.PUBLISH_STATE
-        else:
-            publish_state = True
+
+        publish_state = module.DiscoveryItem.PUBLISH_STATE
+
         module_class = module.DiscoveryItem(
             self._hass,
             self._publish_retain,