162 lines
5.0 KiB
Python
162 lines
5.0 KiB
Python
"""
|
|
Support to check for available updates.
|
|
|
|
For more details about this component, please refer to the documentation at
|
|
https://home-assistant.io/components/updater/
|
|
"""
|
|
import json
|
|
import logging
|
|
import os
|
|
import platform
|
|
import uuid
|
|
from datetime import datetime, timedelta
|
|
# pylint: disable=no-name-in-module, import-error
|
|
from distutils.version import StrictVersion
|
|
|
|
import requests
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
import homeassistant.util.dt as dt_util
|
|
from homeassistant.const import __version__ as CURRENT_VERSION
|
|
from homeassistant.const import ATTR_FRIENDLY_NAME
|
|
from homeassistant.helpers import event
|
|
|
|
REQUIREMENTS = ['distro==1.0.4']
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ATTR_RELEASE_NOTES = 'release_notes'
|
|
|
|
CONF_REPORTING = 'reporting'
|
|
|
|
DOMAIN = 'updater'
|
|
|
|
ENTITY_ID = 'updater.updater'
|
|
|
|
UPDATER_URL = 'https://updater.home-assistant.io/'
|
|
UPDATER_UUID_FILE = '.uuid'
|
|
|
|
CONFIG_SCHEMA = vol.Schema({DOMAIN: {
|
|
vol.Optional(CONF_REPORTING, default=True): cv.boolean
|
|
}}, extra=vol.ALLOW_EXTRA)
|
|
|
|
RESPONSE_SCHEMA = vol.Schema({
|
|
vol.Required('version'): str,
|
|
vol.Required('release-notes'): cv.url,
|
|
})
|
|
|
|
|
|
def _create_uuid(hass, filename=UPDATER_UUID_FILE):
|
|
"""Create UUID and save it in a file."""
|
|
with open(hass.config.path(filename), 'w') as fptr:
|
|
_uuid = uuid.uuid4().hex
|
|
fptr.write(json.dumps({'uuid': _uuid}))
|
|
return _uuid
|
|
|
|
|
|
def _load_uuid(hass, filename=UPDATER_UUID_FILE):
|
|
"""Load UUID from a file or return None."""
|
|
try:
|
|
with open(hass.config.path(filename)) as fptr:
|
|
jsonf = json.loads(fptr.read())
|
|
return uuid.UUID(jsonf['uuid'], version=4).hex
|
|
except (ValueError, AttributeError):
|
|
return None
|
|
except FileNotFoundError:
|
|
return _create_uuid(hass, filename)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Set up the updater component."""
|
|
if 'dev' in CURRENT_VERSION:
|
|
# This component only makes sense in release versions
|
|
_LOGGER.warning("Running on 'dev', only analytics will be submitted")
|
|
|
|
config = config.get(DOMAIN, {})
|
|
huuid = _load_uuid(hass) if config.get(CONF_REPORTING) else None
|
|
|
|
# Update daily, start 1 hour after startup
|
|
_dt = datetime.now() + timedelta(hours=1)
|
|
event.track_time_change(
|
|
hass, lambda _: check_newest_version(hass, huuid),
|
|
hour=_dt.hour, minute=_dt.minute, second=_dt.second)
|
|
|
|
return True
|
|
|
|
|
|
def check_newest_version(hass, huuid):
|
|
"""Check if a new version is available and report if one is."""
|
|
result = get_newest_version(huuid)
|
|
|
|
if result is None:
|
|
return
|
|
|
|
newest, releasenotes = result
|
|
|
|
if newest is None or 'dev' in CURRENT_VERSION:
|
|
return
|
|
|
|
if StrictVersion(newest) > StrictVersion(CURRENT_VERSION):
|
|
_LOGGER.info("The latest available version is %s", newest)
|
|
hass.states.set(
|
|
ENTITY_ID, newest, {ATTR_FRIENDLY_NAME: 'Update Available',
|
|
ATTR_RELEASE_NOTES: releasenotes}
|
|
)
|
|
elif StrictVersion(newest) == StrictVersion(CURRENT_VERSION):
|
|
_LOGGER.info("You are on the latest version (%s) of Home Assistant",
|
|
newest)
|
|
|
|
|
|
def get_newest_version(huuid):
|
|
"""Get the newest Home Assistant version."""
|
|
info_object = {
|
|
'arch': platform.machine(),
|
|
'dev': ('dev' in CURRENT_VERSION),
|
|
'docker': False,
|
|
'os_name': platform.system(),
|
|
'python_version': platform.python_version(),
|
|
'timezone': dt_util.DEFAULT_TIME_ZONE.zone,
|
|
'uuid': huuid,
|
|
'version': CURRENT_VERSION,
|
|
'virtualenv': (os.environ.get('VIRTUAL_ENV') is not None),
|
|
}
|
|
|
|
if platform.system() == 'Windows':
|
|
info_object['os_version'] = platform.win32_ver()[0]
|
|
elif platform.system() == 'Darwin':
|
|
info_object['os_version'] = platform.mac_ver()[0]
|
|
elif platform.system() == 'FreeBSD':
|
|
info_object['os_version'] = platform.release()
|
|
elif platform.system() == 'Linux':
|
|
import distro
|
|
linux_dist = distro.linux_distribution(full_distribution_name=False)
|
|
info_object['distribution'] = linux_dist[0]
|
|
info_object['os_version'] = linux_dist[1]
|
|
info_object['docker'] = os.path.isfile('/.dockerenv')
|
|
|
|
if not huuid:
|
|
info_object = {}
|
|
|
|
res = None
|
|
try:
|
|
req = requests.post(UPDATER_URL, json=info_object, timeout=5)
|
|
res = req.json()
|
|
res = RESPONSE_SCHEMA(res)
|
|
|
|
_LOGGER.info(("Submitted analytics to Home Assistant servers. "
|
|
"Information submitted includes %s"), info_object)
|
|
return (res['version'], res['release-notes'])
|
|
except requests.RequestException:
|
|
_LOGGER.error("Could not contact Home Assistant Update to check "
|
|
"for updates")
|
|
return None
|
|
|
|
except ValueError:
|
|
_LOGGER.error("Received invalid response from Home Assistant Update")
|
|
return None
|
|
|
|
except vol.Invalid:
|
|
_LOGGER.error('Got unexpected response: %s', res)
|
|
return None
|