Add alias to DOMAIN import (part 2) (#125561)

pull/125650/head
epenet 2024-09-10 08:40:02 +02:00 committed by GitHub
parent 130e7317bc
commit 675c467e12
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 24 additions and 16 deletions

View File

@ -9,7 +9,7 @@ from clearpasspy import ClearPass
import voluptuous as vol
from homeassistant.components.device_tracker import (
DOMAIN,
DOMAIN as DEVICE_TRACKER_DOMAIN,
PLATFORM_SCHEMA as DEVICE_TRACKER_PLATFORM_SCHEMA,
DeviceScanner,
)
@ -36,11 +36,13 @@ _LOGGER = logging.getLogger(__name__)
def get_scanner(hass: HomeAssistant, config: ConfigType) -> CPPMDeviceScanner | None:
"""Initialize Scanner."""
config = config[DEVICE_TRACKER_DOMAIN]
data = {
"server": config[DOMAIN][CONF_HOST],
"server": config[CONF_HOST],
"grant_type": GRANT_TYPE,
"secret": config[DOMAIN][CONF_API_KEY],
"client": config[DOMAIN][CONF_CLIENT_ID],
"secret": config[CONF_API_KEY],
"client": config[CONF_CLIENT_ID],
}
cppm = ClearPass(data)
if cppm.access_token is None:

View File

@ -13,7 +13,7 @@ from fortiosapi import FortiOSAPI
import voluptuous as vol
from homeassistant.components.device_tracker import (
DOMAIN,
DOMAIN as DEVICE_TRACKER_DOMAIN,
PLATFORM_SCHEMA as DEVICE_TRACKER_PLATFORM_SCHEMA,
DeviceScanner,
)
@ -37,9 +37,11 @@ PLATFORM_SCHEMA = DEVICE_TRACKER_PLATFORM_SCHEMA.extend(
def get_scanner(hass: HomeAssistant, config: ConfigType) -> FortiOSDeviceScanner | None:
"""Validate the configuration and return a FortiOSDeviceScanner."""
host = config[DOMAIN][CONF_HOST]
verify_ssl = config[DOMAIN][CONF_VERIFY_SSL]
token = config[DOMAIN][CONF_TOKEN]
config = config[DEVICE_TRACKER_DOMAIN]
host = config[CONF_HOST]
verify_ssl = config[CONF_VERIFY_SSL]
token = config[CONF_TOKEN]
fgt = FortiOSAPI()

View File

@ -9,7 +9,7 @@ from openwrt.ubus import Ubus
import voluptuous as vol
from homeassistant.components.device_tracker import (
DOMAIN,
DOMAIN as DEVICE_TRACKER_DOMAIN,
PLATFORM_SCHEMA as DEVICE_TRACKER_PLATFORM_SCHEMA,
DeviceScanner,
)
@ -38,14 +38,16 @@ PLATFORM_SCHEMA = DEVICE_TRACKER_PLATFORM_SCHEMA.extend(
def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None:
"""Validate the configuration and return an ubus scanner."""
dhcp_sw = config[DOMAIN][CONF_DHCP_SOFTWARE]
config = config[DEVICE_TRACKER_DOMAIN]
dhcp_sw = config[CONF_DHCP_SOFTWARE]
scanner: DeviceScanner
if dhcp_sw == "dnsmasq":
scanner = DnsmasqUbusDeviceScanner(config[DOMAIN])
scanner = DnsmasqUbusDeviceScanner(config)
elif dhcp_sw == "odhcpd":
scanner = OdhcpdUbusDeviceScanner(config[DOMAIN])
scanner = OdhcpdUbusDeviceScanner(config)
else:
scanner = UbusDeviceScanner(config[DOMAIN])
scanner = UbusDeviceScanner(config)
return scanner if scanner.success_init else None

View File

@ -8,7 +8,7 @@ from miio import DeviceException, WifiRepeater
import voluptuous as vol
from homeassistant.components.device_tracker import (
DOMAIN,
DOMAIN as DEVICE_TRACKER_DOMAIN,
PLATFORM_SCHEMA as DEVICE_TRACKER_PLATFORM_SCHEMA,
DeviceScanner,
)
@ -32,8 +32,10 @@ def get_scanner(
) -> XiaomiMiioDeviceScanner | None:
"""Return a Xiaomi MiIO device scanner."""
scanner = None
host = config[DOMAIN][CONF_HOST]
token = config[DOMAIN][CONF_TOKEN]
config = config[DEVICE_TRACKER_DOMAIN]
host = config[CONF_HOST]
token = config[CONF_TOKEN]
_LOGGER.info("Initializing with host %s (token %s...)", host, token[:5])