diff --git a/homeassistant/components/waze_travel_time/const.py b/homeassistant/components/waze_travel_time/const.py index 37278543dfb..50384e79d92 100644 --- a/homeassistant/components/waze_travel_time/const.py +++ b/homeassistant/components/waze_travel_time/const.py @@ -1,6 +1,4 @@ """Constants for waze_travel_time.""" -from homeassistant.const import CONF_UNIT_SYSTEM_IMPERIAL, CONF_UNIT_SYSTEM_METRIC - DOMAIN = "waze_travel_time" CONF_DESTINATION = "destination" @@ -21,7 +19,9 @@ DEFAULT_AVOID_TOLL_ROADS = False DEFAULT_AVOID_SUBSCRIPTION_ROADS = False DEFAULT_AVOID_FERRIES = False -UNITS = [CONF_UNIT_SYSTEM_METRIC, CONF_UNIT_SYSTEM_IMPERIAL] +IMPERIAL_UNITS = "imperial" +METRIC_UNITS = "metric" +UNITS = [METRIC_UNITS, IMPERIAL_UNITS] REGIONS = ["US", "NA", "EU", "IL", "AU"] VEHICLE_TYPES = ["car", "taxi", "motorcycle"] diff --git a/homeassistant/components/waze_travel_time/sensor.py b/homeassistant/components/waze_travel_time/sensor.py index 66ee5376266..c4ff489cadf 100644 --- a/homeassistant/components/waze_travel_time/sensor.py +++ b/homeassistant/components/waze_travel_time/sensor.py @@ -16,8 +16,6 @@ from homeassistant.const import ( ATTR_ATTRIBUTION, CONF_NAME, CONF_REGION, - CONF_UNIT_SYSTEM_IMPERIAL, - CONF_UNIT_SYSTEM_METRIC, EVENT_HOMEASSISTANT_STARTED, LENGTH_KILOMETERS, LENGTH_MILES, @@ -49,6 +47,8 @@ from .const import ( DEFAULT_REALTIME, DEFAULT_VEHICLE_TYPE, DOMAIN, + IMPERIAL_UNITS, + METRIC_UNITS, ) _LOGGER = logging.getLogger(__name__) @@ -65,13 +65,13 @@ async def async_setup_entry( defaults = { CONF_REALTIME: DEFAULT_REALTIME, CONF_VEHICLE_TYPE: DEFAULT_VEHICLE_TYPE, - CONF_UNITS: CONF_UNIT_SYSTEM_METRIC, + CONF_UNITS: METRIC_UNITS, CONF_AVOID_FERRIES: DEFAULT_AVOID_FERRIES, CONF_AVOID_SUBSCRIPTION_ROADS: DEFAULT_AVOID_SUBSCRIPTION_ROADS, CONF_AVOID_TOLL_ROADS: DEFAULT_AVOID_TOLL_ROADS, } if hass.config.units is IMPERIAL_SYSTEM: - defaults[CONF_UNITS] = CONF_UNIT_SYSTEM_IMPERIAL + defaults[CONF_UNITS] = IMPERIAL_UNITS if not config_entry.options: new_data = config_entry.data.copy() @@ -248,7 +248,7 @@ class WazeTravelTimeData: self.duration, distance = routes[route] - if units == CONF_UNIT_SYSTEM_IMPERIAL: + if units == IMPERIAL_UNITS: # Convert to miles. self.distance = DistanceConverter.convert( distance, LENGTH_KILOMETERS, LENGTH_MILES