""" homeassistant.components.demo ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Sets up a demo environment that mimics interaction with devices. """ import time import homeassistant as ha import homeassistant.bootstrap as bootstrap import homeassistant.loader as loader from homeassistant.const import ( CONF_PLATFORM, ATTR_ENTITY_PICTURE, ATTR_ENTITY_ID) DOMAIN = "demo" DEPENDENCIES = [] COMPONENTS_WITH_DEMO_PLATFORM = [ 'switch', 'light', 'thermostat', 'sensor', 'media_player', 'notify'] 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 = hass.states.entity_ids('light') switches = hass.states.entity_ids('switch') group.setup_group(hass, 'living room', [lights[0], lights[1], switches[0]]) group.setup_group(hass, 'bedroom', [lights[2], switches[1]]) # Setup IP Camera bootstrap.setup_component( hass, 'camera', {'camera': { 'platform': 'generic', 'name': 'IP Camera', 'still_image_url': 'http://194.218.96.92/jpg/image.jpg', }}) # Setup scripts bootstrap.setup_component( hass, 'script', {'script': { 'demo': { 'alias': 'Demo {}'.format(lights[0]), '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 fake device tracker hass.states.set("device_tracker.paulus", "home", {ATTR_ENTITY_PICTURE: "http://graph.facebook.com/297400035/picture"}) hass.states.set("device_tracker.anne_therese", "not_home", {ATTR_ENTITY_PICTURE: "http://graph.facebook.com/621994601/picture"}) hass.states.set("group.all_devices", "home", { "auto": True, ATTR_ENTITY_ID: [ "device_tracker.paulus", "device_tracker.anne_therese" ] }) # 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