158 lines
5.3 KiB
Python
158 lines
5.3 KiB
Python
"""Module to help with parsing and generating configuration files."""
|
|
import logging
|
|
import os
|
|
from types import MappingProxyType
|
|
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.util.location as loc_util
|
|
from homeassistant.const import (
|
|
CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME, CONF_TEMPERATURE_UNIT,
|
|
CONF_TIME_ZONE, CONF_CUSTOMIZE)
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.util.yaml import load_yaml
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity import valid_entity_id
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
YAML_CONFIG_FILE = 'configuration.yaml'
|
|
CONFIG_DIR_NAME = '.homeassistant'
|
|
|
|
DEFAULT_CONFIG = (
|
|
# Tuples (attribute, default, auto detect property, description)
|
|
(CONF_NAME, 'Home', None, 'Name of the location where Home Assistant is '
|
|
'running'),
|
|
(CONF_LATITUDE, None, 'latitude', 'Location required to calculate the time'
|
|
' the sun rises and sets'),
|
|
(CONF_LONGITUDE, None, 'longitude', None),
|
|
(CONF_TEMPERATURE_UNIT, 'C', None, 'C for Celsius, F for Fahrenheit'),
|
|
(CONF_TIME_ZONE, 'UTC', 'time_zone', 'Pick yours from here: http://en.wiki'
|
|
'pedia.org/wiki/List_of_tz_database_time_zones'),
|
|
)
|
|
DEFAULT_COMPONENTS = {
|
|
'introduction:': 'Show links to resources in log and frontend',
|
|
'frontend:': 'Enables the frontend',
|
|
'updater:': 'Checks for available updates',
|
|
'discovery:': 'Discover some devices automatically',
|
|
'conversation:': 'Allows you to issue voice commands from the frontend',
|
|
'history:': 'Enables support for tracking state changes over time.',
|
|
'logbook:': 'View all events in a logbook',
|
|
'sun:': 'Track the sun',
|
|
'sensor:\n platform: yr': 'Prediction of weather',
|
|
}
|
|
|
|
|
|
def _valid_customize(value):
|
|
"""Config validator for customize."""
|
|
if not isinstance(value, dict):
|
|
raise vol.Invalid('Expected dictionary')
|
|
|
|
for key, val in value.items():
|
|
if not valid_entity_id(key):
|
|
raise vol.Invalid('Invalid entity ID: {}'.format(key))
|
|
|
|
if not isinstance(val, dict):
|
|
raise vol.Invalid('Value of {} is not a dictionary'.format(key))
|
|
|
|
return value
|
|
|
|
CORE_CONFIG_SCHEMA = vol.Schema({
|
|
CONF_NAME: vol.Coerce(str),
|
|
CONF_LATITUDE: cv.latitude,
|
|
CONF_LONGITUDE: cv.longitude,
|
|
CONF_TEMPERATURE_UNIT: cv.temperature_unit,
|
|
CONF_TIME_ZONE: cv.time_zone,
|
|
vol.Required(CONF_CUSTOMIZE,
|
|
default=MappingProxyType({})): _valid_customize,
|
|
})
|
|
|
|
|
|
def get_default_config_dir():
|
|
"""Put together the default configuration directory based on OS."""
|
|
data_dir = os.getenv('APPDATA') if os.name == "nt" \
|
|
else os.path.expanduser('~')
|
|
return os.path.join(data_dir, CONFIG_DIR_NAME)
|
|
|
|
|
|
def ensure_config_exists(config_dir, detect_location=True):
|
|
"""Ensure a config file exists in given configuration directory.
|
|
|
|
Creating a default one if needed.
|
|
Return path to the config file.
|
|
"""
|
|
config_path = find_config_file(config_dir)
|
|
|
|
if config_path is None:
|
|
print("Unable to find configuration. Creating default one in",
|
|
config_dir)
|
|
config_path = create_default_config(config_dir, detect_location)
|
|
|
|
return config_path
|
|
|
|
|
|
def create_default_config(config_dir, detect_location=True):
|
|
"""Create a default configuration file in given configuration directory.
|
|
|
|
Return path to new config file if success, None if failed.
|
|
"""
|
|
config_path = os.path.join(config_dir, YAML_CONFIG_FILE)
|
|
|
|
info = {attr: default for attr, default, _, _ in DEFAULT_CONFIG}
|
|
|
|
location_info = detect_location and loc_util.detect_location_info()
|
|
|
|
if location_info:
|
|
if location_info.use_fahrenheit:
|
|
info[CONF_TEMPERATURE_UNIT] = 'F'
|
|
|
|
for attr, default, prop, _ in DEFAULT_CONFIG:
|
|
if prop is None:
|
|
continue
|
|
info[attr] = getattr(location_info, prop) or default
|
|
|
|
# Writing files with YAML does not create the most human readable results
|
|
# So we're hard coding a YAML template.
|
|
try:
|
|
with open(config_path, 'w') as config_file:
|
|
config_file.write("homeassistant:\n")
|
|
|
|
for attr, _, _, description in DEFAULT_CONFIG:
|
|
if info[attr] is None:
|
|
continue
|
|
elif description:
|
|
config_file.write(" # {}\n".format(description))
|
|
config_file.write(" {}: {}\n".format(attr, info[attr]))
|
|
|
|
config_file.write("\n")
|
|
|
|
for component, description in DEFAULT_COMPONENTS.items():
|
|
config_file.write("# {}\n".format(description))
|
|
config_file.write("{}\n\n".format(component))
|
|
|
|
return config_path
|
|
|
|
except IOError:
|
|
print('Unable to create default configuration file', config_path)
|
|
return None
|
|
|
|
|
|
def find_config_file(config_dir):
|
|
"""Look in given directory for supported configuration files."""
|
|
config_path = os.path.join(config_dir, YAML_CONFIG_FILE)
|
|
|
|
return config_path if os.path.isfile(config_path) else None
|
|
|
|
|
|
def load_yaml_config_file(config_path):
|
|
"""Parse a YAML configuration file."""
|
|
conf_dict = load_yaml(config_path)
|
|
|
|
if not isinstance(conf_dict, dict):
|
|
_LOGGER.error(
|
|
'The configuration file %s does not contain a dictionary',
|
|
os.path.basename(config_path))
|
|
raise HomeAssistantError()
|
|
|
|
return conf_dict
|