Fix LINT warnings

ufozone · Nov 11, 2023 · 5031f48 · 5031f48
1 parent 3136c51
commit 5031f48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/custom_components/unifi_voucher/coordinator.py b/custom_components/unifi_voucher/coordinator.py
@@ -142,6 +142,7 @@ def get_entry_option(
         self,
         conf_key: str,
     ) -> any:
+        """Get config entry option with default value as fallback."""
         _default_value = DEFAULT_VOUCHER.get(conf_key, {}).get("default")
         return self.config_entry.options.get(conf_key, _default_value)
 
@@ -244,13 +245,13 @@ async def async_create_voucher(
         """Create new voucher."""
         try:
             if number is None:
-                number = self.get_entry_option(CONF_VOUCHER_NUMBER)
+                number = int(self.get_entry_option(CONF_VOUCHER_NUMBER))
 
             if quota is None:
-                quota = self.get_entry_option(CONF_VOUCHER_QUOTA)
+                quota = int(self.get_entry_option(CONF_VOUCHER_QUOTA))
 
             if expire is None:
-                expire = self.get_entry_option(CONF_VOUCHER_EXPIRE)
+                expire = int(self.get_entry_option(CONF_VOUCHER_EXPIRE))
 
             await self.client.controller.request(
                 UnifiVoucherCreateRequest.create(

diff --git a/custom_components/unifi_voucher/number.py b/custom_components/unifi_voucher/number.py
@@ -1,9 +1,6 @@
 """UniFi Hotspot Manager number platform."""
 from __future__ import annotations
 
-from collections.abc import Callable, Coroutine
-from dataclasses import dataclass
-
 from homeassistant.core import HomeAssistant
 from homeassistant.const import (
     UnitOfTime,
@@ -16,7 +13,6 @@
 from homeassistant.helpers.entity import Entity
 
 from .const import (
-    LOGGER,
     DOMAIN,
     CONF_VOUCHER_QUOTA,
     CONF_VOUCHER_EXPIRE,