Split build stuff to new repo

hassio/tools.py

@@ -0,0 +1,58 @@
+"""Tools file for HassIO."""
+import asyncio
+import logging
+import re
+import socket
+
+import aiohttp
+import async_timeout
+
+from .const import URL_HASSIO_VERSION, URL_HASSIO_VERSION_BETA
+
+_LOGGER = logging.getLogger(__name__)
+
+_RE_VERSION = re.compile(r"VERSION=(.*)")
+
+
+async def fetch_current_versions(websession, beta=False):
+    """Fetch current versions from github.
+
+    Is a coroutine.
+    """
+    url = URL_HASSIO_VERSION_BETA if beta else URL_HASSIO_VERSION
+    try:
+        with async_timeout.timeout(10, loop=websession.loop):
+            async with websession.get(url) as request:
+                return await request.json(content_type=None)
+
+    except (ValueError, aiohttp.ClientError, asyncio.TimeoutError) as err:
+        _LOGGER.warning("Can't fetch versions from %s! %s", url, err)
+
+
+def get_version_from_env(env_list):
+    """Extract Version from ENV list."""
+    for env in env_list:
+        found = _RE_VERSION.match(env)
+        if found:
+            return found.group(1)
+
+    _LOGGER.error("Can't find VERSION in env")
+    return None
+
+
+def get_local_ip(loop):
+    """Retrieve local IP address.
+
+    Need run inside executor.
+    """
+    try:
+        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+
+        # Use Google Public DNS server to determine own IP
+        sock.connect(('8.8.8.8', 80))
+
+        return sock.getsockname()[0]
+    except socket.error:
+        return socket.gethostbyname(socket.gethostname())
+    finally:
+        sock.close()