Update azure-pipelines-release.yml for Azure Pipelines

0.106.3

CODEOWNERS

@@ -68,6 +68,7 @@ homeassistant/components/config/* @home-assistant/core
 homeassistant/components/configurator/* @home-assistant/core
 homeassistant/components/conversation/* @home-assistant/core
 homeassistant/components/coolmaster/* @OnFreund
+homeassistant/components/coronavirus/* @home_assistant/core
 homeassistant/components/counter/* @fabaff
 homeassistant/components/cover/* @home-assistant/core
 homeassistant/components/cpuspeed/* @fabaff

azure-pipelines-release.yml

@@ -41,7 +41,7 @@ stages:
           jq curl
 
         release="$(Build.SourceBranchName)"
-        created_by="$(curl -s https://api.github.com/repos/home-assistant/home-assistant/releases/tags/${release} | jq --raw-output '.author.login')"
+        created_by="$(curl -s https://api.github.com/repos/home-assistant/core/releases/tags/${release} | jq --raw-output '.author.login')"
 
         if [[ "${created_by}" =~ ^(balloob|pvizeli|fabaff|robbiet480|bramkragten|frenck)$ ]]; then
           exit 0

homeassistant/components/coronavirus/.translations/en.json

@@ -0,0 +1,13 @@
+{
+    "config": {
+        "step": {
+            "user": {
+                "data": {
+                    "country": "Country"
+                },
+                "title": "Pick a country to monitor"
+            }
+        },
+        "title": "Coronavirus"
+    }
+}

homeassistant/components/coronavirus/__init__.py

@@ -0,0 +1,75 @@
+"""The Coronavirus integration."""
+import asyncio
+from datetime import timedelta
+import logging
+
+import aiohttp
+import async_timeout
+import coronavirus
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import aiohttp_client, update_coordinator
+
+from .const import DOMAIN
+
+PLATFORMS = ["sensor"]
+
+
+async def async_setup(hass: HomeAssistant, config: dict):
+    """Set up the Coronavirus component."""
+    # Make sure coordinator is initialized.
+    await get_coordinator(hass)
+    return True
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
+    """Set up Coronavirus from a config entry."""
+    for component in PLATFORMS:
+        hass.async_create_task(
+            hass.config_entries.async_forward_entry_setup(entry, component)
+        )
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
+    """Unload a config entry."""
+    unload_ok = all(
+        await asyncio.gather(
+            *[
+                hass.config_entries.async_forward_entry_unload(entry, component)
+                for component in PLATFORMS
+            ]
+        )
+    )
+
+    return unload_ok
+
+
+async def get_coordinator(hass):
+    """Get the data update coordinator."""
+    if DOMAIN in hass.data:
+        return hass.data[DOMAIN]
+
+    async def async_get_cases():
+        try:
+            with async_timeout.timeout(10):
+                return {
+                    case.id: case
+                    for case in await coronavirus.get_cases(
+                        aiohttp_client.async_get_clientsession(hass)
+                    )
+                }
+        except (asyncio.TimeoutError, aiohttp.ClientError):
+            raise update_coordinator.UpdateFailed
+
+    hass.data[DOMAIN] = update_coordinator.DataUpdateCoordinator(
+        hass,
+        logging.getLogger(__name__),
+        name=DOMAIN,
+        update_method=async_get_cases,
+        update_interval=timedelta(hours=1),
+    )
+    await hass.data[DOMAIN].async_refresh()
+    return hass.data[DOMAIN]

homeassistant/components/coronavirus/config_flow.py

@@ -0,0 +1,41 @@
+"""Config flow for Coronavirus integration."""
+import logging
+
+import voluptuous as vol
+
+from homeassistant import config_entries
+
+from . import get_coordinator
+from .const import DOMAIN, OPTION_WORLDWIDE  # pylint:disable=unused-import
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+    """Handle a config flow for Coronavirus."""
+
+    VERSION = 1
+    CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL
+
+    _options = None
+
+    async def async_step_user(self, user_input=None):
+        """Handle the initial step."""
+        errors = {}
+
+        if self._options is None:
+            self._options = {OPTION_WORLDWIDE: "Worldwide"}
+            coordinator = await get_coordinator(self.hass)
+            for case_id in sorted(coordinator.data):
+                self._options[case_id] = coordinator.data[case_id].country
+
+        if user_input is not None:
+            return self.async_create_entry(
+                title=self._options[user_input["country"]], data=user_input
+            )
+
+        return self.async_show_form(
+            step_id="user",
+            data_schema=vol.Schema({vol.Required("country"): vol.In(self._options)}),
+            errors=errors,
+        )

homeassistant/components/coronavirus/const.py

@@ -0,0 +1,6 @@
+"""Constants for the Coronavirus integration."""
+from coronavirus import DEFAULT_SOURCE
+
+DOMAIN = "coronavirus"
+OPTION_WORLDWIDE = "__worldwide"
+ATTRIBUTION = f"Data provided by {DEFAULT_SOURCE.NAME}"

homeassistant/components/coronavirus/manifest.json

@@ -0,0 +1,12 @@
+{
+  "domain": "coronavirus",
+  "name": "Coronavirus (COVID-19)",
+  "config_flow": true,
+  "documentation": "https://www.home-assistant.io/integrations/coronavirus",
+  "requirements": ["coronavirus==1.0.1"],
+  "ssdp": [],
+  "zeroconf": [],
+  "homekit": {},
+  "dependencies": [],
+  "codeowners": ["@home_assistant/core"]
+}

homeassistant/components/coronavirus/sensor.py

@@ -0,0 +1,69 @@
+"""Sensor platform for the Corona virus."""
+from homeassistant.const import ATTR_ATTRIBUTION
+from homeassistant.helpers.entity import Entity
+
+from . import get_coordinator
+from .const import ATTRIBUTION, OPTION_WORLDWIDE
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    """Defer sensor setup to the shared sensor module."""
+    coordinator = await get_coordinator(hass)
+
+    async_add_entities(
+        CoronavirusSensor(coordinator, config_entry.data["country"], info_type)
+        for info_type in ("confirmed", "recovered", "deaths", "current")
+    )
+
+
+class CoronavirusSensor(Entity):
+    """Sensor representing corona virus data."""
+
+    name = None
+    unique_id = None
+
+    def __init__(self, coordinator, country, info_type):
+        """Initialize coronavirus sensor."""
+        if country == OPTION_WORLDWIDE:
+            self.name = f"Worldwide {info_type}"
+        else:
+            self.name = f"{coordinator.data[country].country} {info_type}"
+        self.unique_id = f"{country}-{info_type}"
+        self.coordinator = coordinator
+        self.country = country
+        self.info_type = info_type
+
+    @property
+    def available(self):
+        """Return if sensor is available."""
+        return self.coordinator.last_update_success and (
+            self.country in self.coordinator.data or self.country == OPTION_WORLDWIDE
+        )
+
+    @property
+    def state(self):
+        """State of the sensor."""
+        if self.country == OPTION_WORLDWIDE:
+            return sum(
+                getattr(case, self.info_type) for case in self.coordinator.data.values()
+            )
+
+        return getattr(self.coordinator.data[self.country], self.info_type)
+
+    @property
+    def unit_of_measurement(self):
+        """Return unit of measurement."""
+        return "people"
+
+    @property
+    def device_state_attributes(self):
+        """Return device attributes."""
+        return {ATTR_ATTRIBUTION: ATTRIBUTION}
+
+    async def async_added_to_hass(self):
+        """When entity is added to hass."""
+        self.coordinator.async_add_listener(self.async_write_ha_state)
+
+    async def async_will_remove_from_hass(self):
+        """When entity will be removed from hass."""
+        self.coordinator.async_remove_listener(self.async_write_ha_state)

homeassistant/components/coronavirus/strings.json

@@ -0,0 +1,13 @@
+{
+  "config": {
+    "title": "Coronavirus",
+    "step": {
+      "user": {
+        "title": "Pick a country to monitor",
+        "data": {
+          "country": "Country"
+        }
+      }
+    }
+  }
+}

homeassistant/components/frontend/manifest.json

@@ -3,7 +3,7 @@
   "name": "Home Assistant Frontend",
   "documentation": "https://www.home-assistant.io/integrations/frontend",
   "requirements": [
-    "home-assistant-frontend==20200220.4"
+    "home-assistant-frontend==20200220.5"
   ],
   "dependencies": [
     "api",

homeassistant/components/lovelace/__init__.py

@@ -104,6 +104,9 @@ class LovelaceStorage:
 
     async def async_save(self, config):
         """Save config."""
+        if self._hass.config.safe_mode:
+            raise HomeAssistantError("Deleting not supported in safe mode")
+
         if self._data is None:
             await self._load()
         self._data["config"] = config
@@ -112,6 +115,9 @@ class LovelaceStorage:
 
     async def async_delete(self):
         """Delete config."""
+        if self._hass.config.safe_mode:
+            raise HomeAssistantError("Deleting not supported in safe mode")
+
         await self.async_save(None)
 
     async def _load(self):

homeassistant/components/rest/sensor.py

@@ -202,17 +202,19 @@ class RestSensor(Entity):
         self.rest.update()
         value = self.rest.data
         _LOGGER.debug("Data fetched from resource: %s", value)
-        content_type = self.rest.headers.get("content-type")
+        if self.rest.headers is not None:
+            # If the http request failed, headers will be None
+            content_type = self.rest.headers.get("content-type")
 
-        if content_type and content_type.startswith("text/xml"):
-            try:
-                value = json.dumps(xmltodict.parse(value))
-                _LOGGER.debug("JSON converted from XML: %s", value)
-            except ExpatError:
-                _LOGGER.warning(
-                    "REST xml result could not be parsed and converted to JSON."
-                )
-                _LOGGER.debug("Erroneous XML: %s", value)
+            if content_type and content_type.startswith("text/xml"):
+                try:
+                    value = json.dumps(xmltodict.parse(value))
+                    _LOGGER.debug("JSON converted from XML: %s", value)
+                except ExpatError:
+                    _LOGGER.warning(
+                        "REST xml result could not be parsed and converted to JSON."
+                    )
+                    _LOGGER.debug("Erroneous XML: %s", value)
 
         if self._json_attrs:
             self._attributes = {}

homeassistant/components/unifi/device_tracker.py

@@ -338,4 +338,4 @@ class UniFiDeviceTracker(ScannerEntity):
     @property
     def should_poll(self):
         """No polling needed."""
-        return False
+        return True

homeassistant/components/unifi/unifi_client.py

@@ -62,4 +62,4 @@ class UniFiClient(Entity):
     @property
     def should_poll(self) -> bool:
         """No polling needed."""
-        return False
+        return True

homeassistant/const.py

@@ -1,7 +1,7 @@
 """Constants used by Home Assistant components."""
 MAJOR_VERSION = 0
 MINOR_VERSION = 106
-PATCH_VERSION = "1"
+PATCH_VERSION = "3"
 __short_version__ = f"{MAJOR_VERSION}.{MINOR_VERSION}"
 __version__ = f"{__short_version__}.{PATCH_VERSION}"
 REQUIRED_PYTHON_VER = (3, 7, 0)

homeassistant/generated/config_flows.py

@@ -17,6 +17,7 @@ FLOWS = [
     "cast",
     "cert_expiry",
     "coolmaster",
+    "coronavirus",
     "daikin",
     "deconz",
     "dialogflow",

homeassistant/package_constraints.txt

@@ -11,7 +11,7 @@ cryptography==2.8
 defusedxml==0.6.0
 distro==1.4.0
 hass-nabucasa==0.31
-home-assistant-frontend==20200220.4
+home-assistant-frontend==20200220.5
 importlib-metadata==1.5.0
 jinja2>=2.10.3
 netdisco==2.6.0

requirements_all.txt

@@ -398,6 +398,9 @@ connect-box==0.2.5
 # homeassistant.components.xiaomi_miio
 construct==2.9.45
 
+# homeassistant.components.coronavirus
+coronavirus==1.0.1
+
 # homeassistant.scripts.credstash
 # credstash==1.15.0
 
@@ -683,7 +686,7 @@ hole==0.5.0
 holidays==0.10.1
 
 # homeassistant.components.frontend
-home-assistant-frontend==20200220.4
+home-assistant-frontend==20200220.5
 
 # homeassistant.components.zwave
 homeassistant-pyozw==0.1.8

requirements_test_all.txt

@@ -143,6 +143,9 @@ colorlog==4.1.0
 # homeassistant.components.xiaomi_miio
 construct==2.9.45
 
+# homeassistant.components.coronavirus
+coronavirus==1.0.1
+
 # homeassistant.scripts.credstash
 # credstash==1.15.0
 
@@ -254,7 +257,7 @@ hole==0.5.0
 holidays==0.10.1
 
 # homeassistant.components.frontend
-home-assistant-frontend==20200220.4
+home-assistant-frontend==20200220.5
 
 # homeassistant.components.zwave
 homeassistant-pyozw==0.1.8

tests/components/coronavirus/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Coronavirus integration."""

tests/components/coronavirus/test_config_flow.py

@@ -0,0 +1,33 @@
+"""Test the Coronavirus config flow."""
+from asynctest import patch
+
+from homeassistant import config_entries, setup
+from homeassistant.components.coronavirus.const import DOMAIN, OPTION_WORLDWIDE
+
+
+async def test_form(hass):
+    """Test we get the form."""
+    await setup.async_setup_component(hass, "persistent_notification", {})
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+    assert result["type"] == "form"
+    assert result["errors"] == {}
+
+    with patch("coronavirus.get_cases", return_value=[],), patch(
+        "homeassistant.components.coronavirus.async_setup", return_value=True
+    ) as mock_setup, patch(
+        "homeassistant.components.coronavirus.async_setup_entry", return_value=True,
+    ) as mock_setup_entry:
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], {"country": OPTION_WORLDWIDE},
+        )
+
+    assert result2["type"] == "create_entry"
+    assert result2["title"] == "Worldwide"
+    assert result2["data"] == {
+        "country": OPTION_WORLDWIDE,
+    }
+    await hass.async_block_till_done()
+    assert len(mock_setup.mock_calls) == 1
+    assert len(mock_setup_entry.mock_calls) == 1

tests/components/lovelace/test_init.py

@@ -45,6 +45,16 @@ async def test_lovelace_from_storage(hass, hass_ws_client, hass_storage):
     assert not response["success"]
     assert response["error"]["code"] == "config_not_found"
 
+    await client.send_json(
+        {"id": 9, "type": "lovelace/config/save", "config": {"yo": "hello"}}
+    )
+    response = await client.receive_json()
+    assert not response["success"]
+
+    await client.send_json({"id": 10, "type": "lovelace/config/delete"})
+    response = await client.receive_json()
+    assert not response["success"]
+
 
 async def test_lovelace_from_storage_save_before_load(
     hass, hass_ws_client, hass_storage

tests/components/rest/test_sensor.py

@@ -589,6 +589,35 @@ class TestRestSensor(unittest.TestCase):
         assert mock_logger.warning.called
         assert mock_logger.debug.called
 
+    @patch("homeassistant.components.rest.sensor._LOGGER")
+    def test_update_with_failed_get(self, mock_logger):
+        """Test attributes get extracted from a XML result with bad xml."""
+        value_template = template("{{ value_json.toplevel.master_value }}")
+        value_template.hass = self.hass
+
+        self.rest.update = Mock(
+            "rest.RestData.update", side_effect=self.update_side_effect(None, None),
+        )
+        self.sensor = rest.RestSensor(
+            self.hass,
+            self.rest,
+            self.name,
+            self.unit_of_measurement,
+            self.device_class,
+            value_template,
+            ["key"],
+            self.force_update,
+            self.resource_template,
+            self.json_attrs_path,
+        )
+
+        self.sensor.update()
+        assert {} == self.sensor.device_state_attributes
+        assert mock_logger.warning.called
+        assert mock_logger.debug.called
+        assert self.sensor.state is None
+        assert self.sensor.available is False
+
 
 class TestRestData(unittest.TestCase):
     """Tests for RestData."""