Fix SABnzbd config check (#71549)

This commit is contained in:
Shai Ungar 2022-05-09 13:34:16 +03:00 committed by Paulus Schoutsen
parent 6c70a518eb
commit 18d440cc6f
2 changed files with 15 additions and 5 deletions

View File

@ -16,7 +16,6 @@ from homeassistant.const import (
CONF_PORT,
CONF_SENSORS,
CONF_SSL,
CONF_URL,
)
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
@ -46,7 +45,7 @@ from .const import (
UPDATE_INTERVAL,
)
from .sab import get_client
from .sensor import SENSOR_KEYS
from .sensor import OLD_SENSOR_KEYS
PLATFORMS = ["sensor"]
_LOGGER = logging.getLogger(__name__)
@ -80,12 +79,11 @@ CONFIG_SCHEMA = vol.Schema(
{
vol.Required(CONF_API_KEY): str,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): str,
vol.Required(CONF_URL): str,
vol.Optional(CONF_PATH): str,
vol.Optional(CONF_HOST, default=DEFAULT_HOST): cv.string,
vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
vol.Optional(CONF_SENSORS): vol.All(
cv.ensure_list, [vol.In(SENSOR_KEYS)]
cv.ensure_list, [vol.In(OLD_SENSOR_KEYS)]
),
vol.Optional(CONF_SSL, default=DEFAULT_SSL): cv.boolean,
},

View File

@ -103,7 +103,19 @@ SENSOR_TYPES: tuple[SabnzbdSensorEntityDescription, ...] = (
),
)
SENSOR_KEYS: list[str] = [desc.key for desc in SENSOR_TYPES]
OLD_SENSOR_KEYS = [
"current_status",
"speed",
"queue_size",
"queue_remaining",
"disk_size",
"disk_free",
"queue_count",
"day_size",
"week_size",
"month_size",
"total_size",
]
async def async_setup_entry(