Add number entity for usage_quota option

ufozone · Dec 18, 2023 · fe734ba · fe734ba
1 parent 2adf086
commit fe734ba
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/custom_components/unifi_voucher/number.py b/custom_components/unifi_voucher/number.py
@@ -3,6 +3,7 @@
 
 from homeassistant.core import HomeAssistant
 from homeassistant.const import (
+    UnitOfInformation,
     UnitOfTime,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -16,6 +17,7 @@
     DOMAIN,
     CONF_VOUCHER_QUOTA,
     CONF_VOUCHER_DURATION,
+    CONF_VOUCHER_USAGE_QUOTA,
     DEFAULT_VOUCHER,
 )
 from .coordinator import UnifiVoucherCoordinator
@@ -41,6 +43,12 @@ async def async_setup_entry(
             native_unit_of_measurement=UnitOfTime.HOURS,
             translation_key=CONF_VOUCHER_DURATION,
         ),
+        NumberEntityDescription(
+            key=CONF_VOUCHER_USAGE_QUOTA,
+            icon="mdi:database-sync",
+            native_unit_of_measurement=UnitOfInformation.MEGABYTES,
+            translation_key=CONF_VOUCHER_USAGE_QUOTA,
+        ),
     ]
 
     async_add_entities(