From 245450a402979f29bb2d14034b8645db9eff807b Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Tue, 3 Sep 2019 10:17:03 +0200
Subject: [PATCH] Upgrade pyhaversion to 3.1.0 (#26232)

---
 homeassistant/components/version/manifest.json |  2 +-
 homeassistant/components/version/sensor.py     | 12 ++++++++++--
 requirements_all.txt                           |  2 +-
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/version/manifest.json b/homeassistant/components/version/manifest.json
index 2a48f91a6f8..815e7ff9a25 100644
--- a/homeassistant/components/version/manifest.json
+++ b/homeassistant/components/version/manifest.json
@@ -3,7 +3,7 @@
   "name": "Version",
   "documentation": "https://www.home-assistant.io/components/version",
   "requirements": [
-    "pyhaversion==3.0.2"
+    "pyhaversion==3.1.0"
   ],
   "dependencies": [],
   "codeowners": [
diff --git a/homeassistant/components/version/sensor.py b/homeassistant/components/version/sensor.py
index 438ea8f690c..3e00b87e984 100644
--- a/homeassistant/components/version/sensor.py
+++ b/homeassistant/components/version/sensor.py
@@ -28,7 +28,7 @@ ALL_IMAGES = [
     "odroid-c2",
     "odroid-xu",
 ]
-ALL_SOURCES = ["local", "pypi", "hassio", "docker"]
+ALL_SOURCES = ["local", "pypi", "hassio", "docker", "haio"]
 
 CONF_BETA = "beta"
 CONF_IMAGE = "image"
@@ -54,7 +54,13 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
 
 async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
     """Set up the Version sensor platform."""
-    from pyhaversion import LocalVersion, DockerVersion, HassioVersion, PyPiVersion
+    from pyhaversion import (
+        LocalVersion,
+        DockerVersion,
+        HassioVersion,
+        PyPiVersion,
+        HaIoVersion,
+    )
 
     beta = config.get(CONF_BETA)
     image = config.get(CONF_IMAGE)
@@ -74,6 +80,8 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
         haversion = VersionData(HassioVersion(hass.loop, session, branch, image))
     elif source == "docker":
         haversion = VersionData(DockerVersion(hass.loop, session, branch, image))
+    elif source == "haio":
+        haversion = VersionData(HaIoVersion(hass.loop, session))
     else:
         haversion = VersionData(LocalVersion(hass.loop, session))
 
diff --git a/requirements_all.txt b/requirements_all.txt
index 7fdf063ea54..60e580296d0 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -1193,7 +1193,7 @@ pygtfs==0.1.5
 pygtt==1.1.2
 
 # homeassistant.components.version
-pyhaversion==3.0.2
+pyhaversion==3.1.0
 
 # homeassistant.components.heos
 pyheos==0.6.0