def setup(hass, config):
"""Set up the streamlabs water integration."""
conf = config[DOMAIN]
api_key = conf.get(CONF_API_KEY)
location_id = conf.get(CONF_LOCATION_ID)
client = streamlabswater.StreamlabsClient(api_key)
locations = client.get_locations().get("locations")
if locations is None:
_LOGGER.error("Unable to retrieve locations. Verify API key")
return False
if location_id is None:
location = locations[0]
location_id = location["locationId"]
_LOGGER.info("Streamlabs Water Monitor auto-detected location_id=%s",
location_id)
else:
location = next(
(loc for loc in locations if location_id == loc["locationId"]),
None)
if location is None:
_LOGGER.error("Supplied location_id is invalid")
return False
location_name = location["name"]
hass.data[DOMAIN] = {
"client": client,
"location_id": location_id,
"location_name": location_name,
}
for platform in PLATFORMS:
discovery.load_platform(hass, platform, DOMAIN, {}, config)
def set_away_mode(service):
"""Set the StreamLabsWater Away Mode."""
away_mode = service.data.get(ATTR_AWAY_MODE)
client.update_location(location_id, away_mode)
hass.services.register(DOMAIN,
SERVICE_SET_AWAY_MODE,
set_away_mode,
schema=SET_AWAY_MODE_SCHEMA)
return True
def setup(hass, config):
"""Set up the streamlabs water component."""
from streamlabswater import streamlabswater
conf = config[DOMAIN]
api_key = conf.get(CONF_API_KEY)
location_id = conf.get(CONF_LOCATION_ID)
client = streamlabswater.StreamlabsClient(api_key)
locations = client.get_locations().get('locations')
if locations is None:
_LOGGER.error("Unable to retrieve locations. Verify API key")
return False
if location_id is None:
location = locations[0]
location_id = location['locationId']
_LOGGER.info("Streamlabs Water Monitor auto-detected location_id=%s",
location_id)
else:
location = next(
(l for l in locations if location_id == l['locationId']), None)
if location is None:
_LOGGER.error("Supplied location_id is invalid")
return False
location_name = location['name']
hass.data[DOMAIN] = {
'client': client,
'location_id': location_id,
'location_name': location_name
}
for component in STREAMLABSWATER_COMPONENTS:
discovery.load_platform(hass, component, DOMAIN, {}, config)
def set_away_mode(service):
"""Set the StreamLabsWater Away Mode."""
away_mode = service.data.get(ATTR_AWAY_MODE)
client.update_location(location_id, away_mode)
hass.services.register(DOMAIN,
SERVICE_SET_AWAY_MODE,
set_away_mode,
schema=SET_AWAY_MODE_SCHEMA)
return True