mirror of
https://github.com/home-assistant/core
synced 2024-09-12 15:16:21 +02:00
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
"""
|
|
homeassistant.components.demo
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Sets up a demo environment that mimics interaction with devices.
|
|
"""
|
|
import time
|
|
|
|
import homeassistant.core as ha
|
|
import homeassistant.bootstrap as bootstrap
|
|
import homeassistant.loader as loader
|
|
from homeassistant.const import (
|
|
CONF_PLATFORM, ATTR_ENTITY_ID)
|
|
|
|
DOMAIN = "demo"
|
|
|
|
DEPENDENCIES = ['conversation', 'introduction', 'zone']
|
|
|
|
COMPONENTS_WITH_DEMO_PLATFORM = [
|
|
'device_tracker', 'light', 'media_player', 'notify', 'switch', 'sensor',
|
|
'thermostat', 'camera', 'binary_sensor', 'alarm_control_panel', 'lock']
|
|
|
|
|
|
def setup(hass, config):
|
|
""" Setup a demo environment. """
|
|
group = loader.get_component('group')
|
|
configurator = loader.get_component('configurator')
|
|
|
|
config.setdefault(ha.DOMAIN, {})
|
|
config.setdefault(DOMAIN, {})
|
|
|
|
if config[DOMAIN].get('hide_demo_state') != 1:
|
|
hass.states.set('a.Demo_Mode', 'Enabled')
|
|
|
|
# Setup sun
|
|
if not hass.config.latitude:
|
|
hass.config.latitude = 32.87336
|
|
|
|
if not hass.config.longitude:
|
|
hass.config.longitude = 117.22743
|
|
|
|
bootstrap.setup_component(hass, 'sun')
|
|
|
|
# Setup demo platforms
|
|
for component in COMPONENTS_WITH_DEMO_PLATFORM:
|
|
bootstrap.setup_component(
|
|
hass, component, {component: {CONF_PLATFORM: 'demo'}})
|
|
|
|
# Setup room groups
|
|
lights = sorted(hass.states.entity_ids('light'))
|
|
switches = sorted(hass.states.entity_ids('switch'))
|
|
media_players = sorted(hass.states.entity_ids('media_player'))
|
|
group.setup_group(hass, 'living room', [lights[2], lights[1], switches[0],
|
|
media_players[1]])
|
|
group.setup_group(hass, 'bedroom', [lights[0], switches[1],
|
|
media_players[0]])
|
|
|
|
# Setup scripts
|
|
bootstrap.setup_component(
|
|
hass, 'script',
|
|
{'script': {
|
|
'demo': {
|
|
'alias': 'Toggle {}'.format(lights[0].split('.')[1]),
|
|
'sequence': [{
|
|
'execute_service': 'light.turn_off',
|
|
'service_data': {ATTR_ENTITY_ID: lights[0]}
|
|
}, {
|
|
'delay': {'seconds': 5}
|
|
}, {
|
|
'execute_service': 'light.turn_on',
|
|
'service_data': {ATTR_ENTITY_ID: lights[0]}
|
|
}, {
|
|
'delay': {'seconds': 5}
|
|
}, {
|
|
'execute_service': 'light.turn_off',
|
|
'service_data': {ATTR_ENTITY_ID: lights[0]}
|
|
}]
|
|
}}})
|
|
|
|
# Setup scenes
|
|
bootstrap.setup_component(
|
|
hass, 'scene',
|
|
{'scene': [
|
|
{'name': 'Romantic lights',
|
|
'entities': {
|
|
lights[0]: True,
|
|
lights[1]: {'state': 'on', 'xy_color': [0.33, 0.66],
|
|
'brightness': 200},
|
|
}},
|
|
{'name': 'Switch on and off',
|
|
'entities': {
|
|
switches[0]: True,
|
|
switches[1]: False,
|
|
}},
|
|
]})
|
|
|
|
# Setup configurator
|
|
configurator_ids = []
|
|
|
|
def hue_configuration_callback(data):
|
|
""" Fake callback, mark config as done. """
|
|
time.sleep(2)
|
|
|
|
# First time it is called, pretend it failed.
|
|
if len(configurator_ids) == 1:
|
|
configurator.notify_errors(
|
|
configurator_ids[0],
|
|
"Failed to register, please try again.")
|
|
|
|
configurator_ids.append(0)
|
|
else:
|
|
configurator.request_done(configurator_ids[0])
|
|
|
|
request_id = configurator.request_config(
|
|
hass, "Philips Hue", hue_configuration_callback,
|
|
description=("Press the button on the bridge to register Philips Hue "
|
|
"with Home Assistant."),
|
|
description_image="/static/images/config_philips_hue.jpg",
|
|
submit_caption="I have pressed the button"
|
|
)
|
|
|
|
configurator_ids.append(request_id)
|
|
|
|
return True
|