From 7e756798a7a577a11dc5a4fbb0a08a7c64c65869 Mon Sep 17 00:00:00 2001 From: epenet Date: Thu, 6 Jan 2022 20:56:46 +0000 Subject: [PATCH] Adjust amcrest --- homeassistant/components/amcrest/__init__.py | 23 +++++++++++++------- 1 file changed, 15 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/amcrest/__init__.py b/homeassistant/components/amcrest/__init__.py index 3ee6e685eb5..78463e4a110 100644 --- a/homeassistant/components/amcrest/__init__.py +++ b/homeassistant/components/amcrest/__init__.py @@ -15,10 +15,6 @@ import voluptuous as vol from homeassistant.auth.models import User from homeassistant.auth.permissions.const import POLICY_CONTROL -from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR -from homeassistant.components.camera import DOMAIN as CAMERA -from homeassistant.components.sensor import DOMAIN as SENSOR -from homeassistant.components.switch import DOMAIN as SWITCH from homeassistant.const import ( ATTR_ENTITY_ID, CONF_AUTHENTICATION, @@ -34,6 +30,7 @@ from homeassistant.const import ( ENTITY_MATCH_ALL, ENTITY_MATCH_NONE, HTTP_BASIC_AUTHENTICATION, + Platform, ) from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.exceptions import Unauthorized, UnknownUser @@ -301,13 +298,15 @@ def setup(hass: HomeAssistant, config: ConfigType) -> bool: control_light, ) - discovery.load_platform(hass, CAMERA, DOMAIN, {CONF_NAME: name}, config) + discovery.load_platform( + hass, Platform.CAMERA, DOMAIN, {CONF_NAME: name}, config + ) event_codes = set() if binary_sensors: discovery.load_platform( hass, - BINARY_SENSOR, + Platform.BINARY_SENSOR, DOMAIN, {CONF_NAME: name, CONF_BINARY_SENSORS: binary_sensors}, config, @@ -324,12 +323,20 @@ def setup(hass: HomeAssistant, config: ConfigType) -> bool: if sensors: discovery.load_platform( - hass, SENSOR, DOMAIN, {CONF_NAME: name, CONF_SENSORS: sensors}, config + hass, + Platform.SENSOR, + DOMAIN, + {CONF_NAME: name, CONF_SENSORS: sensors}, + config, ) if switches: discovery.load_platform( - hass, SWITCH, DOMAIN, {CONF_NAME: name, CONF_SWITCHES: switches}, config + hass, + Platform.SWITCH, + DOMAIN, + {CONF_NAME: name, CONF_SWITCHES: switches}, + config, ) if not hass.data[DATA_AMCREST][DEVICES]: