Update component code imports

Limych · Jun 13, 2022 · 26facb3 · 26facb3
1 parent 738c98f
commit 26facb3
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 20 deletions.
diff --git a/custom_components/integration_blueprint/__init__.py b/custom_components/integration_blueprint/__init__.py
@@ -9,14 +9,15 @@
 from datetime import timedelta
 
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from homeassistant.helpers.typing import ConfigType
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .api import IntegrationBlueprintApiClient
-from .const import CONF_PASSWORD, CONF_USERNAME, DOMAIN, PLATFORMS, STARTUP_MESSAGE
+from .const import DOMAIN, PLATFORMS, STARTUP_MESSAGE
 
 SCAN_INTERVAL = timedelta(seconds=30)
 

diff --git a/custom_components/integration_blueprint/binary_sensor.py b/custom_components/integration_blueprint/binary_sensor.py
@@ -1,8 +1,9 @@
 """Binary sensor platform for integration_blueprint."""
 from homeassistant.components.binary_sensor import BinarySensorEntity
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 
-from .const import BINARY_SENSOR, BINARY_SENSOR_DEVICE_CLASS, DEFAULT_NAME, DOMAIN
+from .const import BINARY_SENSOR_DEVICE_CLASS, DEFAULT_NAME, DOMAIN
 from .entity import IntegrationBlueprintEntity
 
 
@@ -18,7 +19,7 @@ class IntegrationBlueprintBinarySensor(IntegrationBlueprintEntity, BinarySensorE
     @property
     def name(self):
         """Return the name of the binary_sensor."""
-        return f"{DEFAULT_NAME}_{BINARY_SENSOR}"
+        return f"{DEFAULT_NAME}_{Platform.BINARY_SENSOR}"
 
     @property
     def device_class(self):

diff --git a/custom_components/integration_blueprint/config_flow.py b/custom_components/integration_blueprint/config_flow.py
@@ -4,17 +4,13 @@
 import voluptuous as vol
 from homeassistant import config_entries
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import callback
 from homeassistant.helpers.aiohttp_client import async_create_clientsession
 from homeassistant.helpers.typing import ConfigType
 
 from .api import IntegrationBlueprintApiClient
-from .const import (  # pylint: disable=unused-import
-    CONF_PASSWORD,
-    CONF_USERNAME,
-    DOMAIN,
-    PLATFORMS,
-)
+from .const import DOMAIN, PLATFORMS  # pylint: disable=unused-import
 
 
 class BlueprintFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):

diff --git a/custom_components/integration_blueprint/const.py b/custom_components/integration_blueprint/const.py
@@ -1,8 +1,10 @@
 """Constants for integration_blueprint."""
 
-# Base component constants
 from typing import Final
 
+from homeassistant.const import Platform
+
+# Base component constants
 NAME: Final = "Integration blueprint"
 DOMAIN: Final = "integration_blueprint"
 VERSION: Final = "0.1.0"
@@ -26,15 +28,14 @@
 BINARY_SENSOR_DEVICE_CLASS: Final = "connectivity"
 
 # Platforms
-BINARY_SENSOR: Final = "binary_sensor"
-SENSOR: Final = "sensor"
-SWITCH: Final = "switch"
-PLATFORMS: Final = [BINARY_SENSOR, SENSOR, SWITCH]
+PLATFORMS: Final = [
+    Platform.BINARY_SENSOR,
+    Platform.SENSOR,
+    Platform.SWITCH,
+]
 
 # Configuration and options
 CONF_ENABLED: Final = "enabled"
-CONF_USERNAME: Final = "username"
-CONF_PASSWORD: Final = "password"
 
 # Defaults
 DEFAULT_NAME: Final = DOMAIN

diff --git a/custom_components/integration_blueprint/sensor.py b/custom_components/integration_blueprint/sensor.py
@@ -1,8 +1,9 @@
 """Sensor platform for integration_blueprint."""
 from homeassistant.components.sensor import SensorEntity
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 
-from .const import DEFAULT_NAME, DOMAIN, ICON, SENSOR
+from .const import DEFAULT_NAME, DOMAIN, ICON
 from .entity import IntegrationBlueprintEntity
 
 
@@ -18,7 +19,7 @@ class IntegrationBlueprintSensor(IntegrationBlueprintEntity, SensorEntity):
     @property
     def name(self):
         """Return the name of the sensor."""
-        return f"{DEFAULT_NAME}_{SENSOR}"
+        return f"{DEFAULT_NAME}_{Platform.SENSOR}"
 
     @property
     def native_value(self):

diff --git a/custom_components/integration_blueprint/switch.py b/custom_components/integration_blueprint/switch.py
@@ -1,8 +1,9 @@
 """Switch platform for integration_blueprint."""
 from homeassistant.components.switch import SwitchEntity
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 
-from .const import DEFAULT_NAME, DOMAIN, ICON, SWITCH
+from .const import DEFAULT_NAME, DOMAIN, ICON
 from .entity import IntegrationBlueprintEntity
 
 
@@ -28,7 +29,7 @@ async def async_turn_off(self, **kwargs):  # pylint: disable=unused-argument
     @property
     def name(self):
         """Return the name of the switch."""
-        return f"{DEFAULT_NAME}_{SWITCH}"
+        return f"{DEFAULT_NAME}_{Platform.SWITCH}"
 
     @property
     def icon(self):