core/homeassistant/components/litejet.py

"""Allows the LiteJet lighting system to be controlled by Home Assistant.

For more details about this component, please refer to the documentation at
https://home-assistant.io/components/litejet/
"""
import logging

import voluptuous as vol

from homeassistant.helpers import discovery
from homeassistant.const import CONF_PORT
import homeassistant.helpers.config_validation as cv

REQUIREMENTS = ['pylitejet==0.1']

_LOGGER = logging.getLogger(__name__)

CONF_EXCLUDE_NAMES = 'exclude_names'
CONF_INCLUDE_SWITCHES = 'include_switches'

DOMAIN = 'litejet'

CONFIG_SCHEMA = vol.Schema({
    DOMAIN: vol.Schema({
        vol.Required(CONF_PORT): cv.string,
        vol.Optional(CONF_EXCLUDE_NAMES): vol.All(cv.ensure_list, [cv.string]),
        vol.Optional(CONF_INCLUDE_SWITCHES, default=False): cv.boolean
    })
}, extra=vol.ALLOW_EXTRA)


def setup(hass, config):
    """Initialize the LiteJet component."""
    from pylitejet import LiteJet

    url = config[DOMAIN].get(CONF_PORT)

    hass.data['litejet_system'] = LiteJet(url)
    hass.data['litejet_config'] = config[DOMAIN]

    discovery.load_platform(hass, 'light', DOMAIN, {}, config)
    if config[DOMAIN].get(CONF_INCLUDE_SWITCHES):
        discovery.load_platform(hass, 'switch', DOMAIN, {}, config)
    discovery.load_platform(hass, 'scene', DOMAIN, {}, config)

    return True


def is_ignored(hass, name):
    """Determine if a load, switch, or scene should be ignored."""
    for prefix in hass.data['litejet_config'].get(CONF_EXCLUDE_NAMES, []):
        if name.startswith(prefix):
            return True
    return False