Move Lock action to config item

cyr-ius · Dec 12, 2021 · f0ebb43 · f0ebb43
1 parent a9177f9
commit f0ebb43
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/custom_components/heatzy/manifest.json b/custom_components/heatzy/manifest.json
@@ -7,5 +7,5 @@
     "requirements": ["heatzypy==0.1.4.2"],
     "dependencies": [],
     "codeowners": ["@cyr-ius"],
-    "version": "4.8"
+    "version": "4.9"
 }
diff --git a/custom_components/heatzy/switch.py b/custom_components/heatzy/switch.py
@@ -3,7 +3,7 @@
 
 from heatzypy.exception import HeatzyException
 from homeassistant.components.switch import SwitchEntity
-from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.entity import DeviceInfo, EntityCategory
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
 from .const import DOMAIN
@@ -23,6 +23,8 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
 class LockSwitchEntity(CoordinatorEntity, SwitchEntity):
     """Lock Switch."""
 
+    entity_category = EntityCategory.CONFIG
+
     def __init__(self, coordinator, _unique_id):
         """Initialize switch."""
         super().__init__(coordinator)