Set correct entity category for all entites #25

ufozone · Feb 24, 2024 · 5bc41b6 · 5bc41b6
1 parent 4e766a6
commit 5bc41b6
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/custom_components/unifi_voucher/button.py b/custom_components/unifi_voucher/button.py
@@ -11,7 +11,10 @@
  ButtonEntity,
  ButtonEntityDescription,
 )
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity import (
+ Entity,
+ EntityCategory,
+)
 
 from .const import (
  DOMAIN,
@@ -62,6 +65,7 @@ async def async_setup_entry(
  icon="mdi:update",
  translation_key="update",
  device_class=ButtonDeviceClass.UPDATE,
+ entity_category=EntityCategory.DIAGNOSTIC,
  press_action=lambda coordinator: coordinator.async_update_vouchers(),
  ),
  ]

diff --git a/custom_components/unifi_voucher/image.py b/custom_components/unifi_voucher/image.py
@@ -15,7 +15,10 @@
  ImageEntity,
  ImageEntityDescription,
 )
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity import (
+ Entity,
+ EntityCategory,
+)
 
 import homeassistant.util.dt as dt_util
 
@@ -42,6 +45,7 @@ async def async_setup_entry(
  translation_key=ATTR_QR_CODE,
  icon="mdi:qrcode",
  device_class=None,
+ entity_category=EntityCategory.DIAGNOSTIC,
  ),
  ]
 

diff --git a/custom_components/unifi_voucher/number.py b/custom_components/unifi_voucher/number.py
@@ -11,7 +11,10 @@
  NumberEntity,
  NumberEntityDescription,
 )
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity import (
+ Entity,
+ EntityCategory,
+)
 
 from .const import (
  DOMAIN,
@@ -36,18 +39,21 @@ async def async_setup_entry(
  key=CONF_VOUCHER_QUOTA,
  icon="mdi:numeric-9-plus",
  translation_key=CONF_VOUCHER_QUOTA,
+ entity_category=EntityCategory.CONFIG,
  ),
  NumberEntityDescription(
  key=CONF_VOUCHER_DURATION,
  icon="mdi:clock-outline",
  native_unit_of_measurement=UnitOfTime.HOURS,
  translation_key=CONF_VOUCHER_DURATION,
+ entity_category=EntityCategory.CONFIG,
  ),
  NumberEntityDescription(
  key=CONF_VOUCHER_USAGE_QUOTA,
  icon="mdi:database-sync",
  native_unit_of_measurement=UnitOfInformation.MEGABYTES,
  translation_key=CONF_VOUCHER_USAGE_QUOTA,
+ entity_category=EntityCategory.CONFIG,
  ),
  ]