refactor: Use HA "temperature" const

jcgoette · Oct 28, 2022 · 4f3be82 · 4f3be82
1 parent dbdb8b6
commit 4f3be82
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/custom_components/babybuddy/config_flow.py b/custom_components/babybuddy/config_flow.py
@@ -12,6 +12,7 @@
     CONF_PATH,
     CONF_PORT,
     CONF_SCAN_INTERVAL,
+    TEMPERATURE,
     UnitOfMass,
     UnitOfTemperature,
     UnitOfVolume,
@@ -22,7 +23,6 @@
 from .client import BabyBuddyClient
 from .const import (
     CONF_FEEDING_UNIT,
-    CONF_TEMPERATURE_UNIT,
     CONF_WEIGHT_UNIT,
     CONFIG_FLOW_VERSION,
     DEFAULT_NAME,
@@ -159,8 +159,8 @@ async def async_step_init(
 
         options = {
             vol.Optional(
-                CONF_TEMPERATURE_UNIT,
-                default=self.config_entry.options.get(CONF_TEMPERATURE_UNIT, None),
+                TEMPERATURE,
+                default=self.config_entry.options.get(TEMPERATURE, None),
             ): vol.In([UnitOfTemperature.CELSIUS, UnitOfTemperature.FAHRENHEIT]),
             vol.Optional(
                 CONF_WEIGHT_UNIT,

diff --git a/custom_components/babybuddy/const.py b/custom_components/babybuddy/const.py
@@ -18,7 +18,6 @@
 DOMAIN: Final = "babybuddy"
 
 CONF_FEEDING_UNIT: Final = "feedings"
-CONF_TEMPERATURE_UNIT: Final = "temperature"
 CONF_WEIGHT_UNIT: Final = "weight"
 
 DEFAULT_NAME: Final = "Baby Buddy"