jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 20:27:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle unhandled IQVIA data update exception (#32144)

Browse Source 
* Handle unhandled IQVIA data update exception

* Cleanup

* Ask for forgiveness, not permission

* Use warning-level logs

* Fix log messages
This commit is contained in:
Aaron Bach 2020-02-24 13:02:33 -07:00 committed by GitHub
parent 9801810552
commit db40b2fc32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
homeassistant/components/iqvia/sensor.py
View File

@ -134,21 +134,34 @@ class IndexSensor(IQVIAEntity):
async def async_update(self):
"""Update the sensor."""
if not self._iqvia.data:
_LOGGER.warning(
"IQVIA didn't return data for %s; trying again later", self.name
)
return
data = {}
if self._type in (TYPE_ALLERGY_TODAY, TYPE_ALLERGY_TOMORROW):
data = self._iqvia.data[TYPE_ALLERGY_INDEX].get("Location")
elif self._type in (TYPE_ASTHMA_TODAY, TYPE_ASTHMA_TOMORROW):
data = self._iqvia.data[TYPE_ASTHMA_INDEX].get("Location")
elif self._type == TYPE_DISEASE_TODAY:
data = self._iqvia.data[TYPE_DISEASE_INDEX].get("Location")
if not data:
try:
if self._type in (TYPE_ALLERGY_TODAY, TYPE_ALLERGY_TOMORROW):
data = self._iqvia.data[TYPE_ALLERGY_INDEX].get("Location")
elif self._type in (TYPE_ASTHMA_TODAY, TYPE_ASTHMA_TOMORROW):
data = self._iqvia.data[TYPE_ASTHMA_INDEX].get("Location")
elif self._type == TYPE_DISEASE_TODAY:
data = self._iqvia.data[TYPE_DISEASE_INDEX].get("Location")
except KeyError:
_LOGGER.warning(
"IQVIA didn't return data for %s; trying again later", self.name
)
return
key = self._type.split("_")[-1].title()
[period] = [p for p in data["periods"] if p["Type"] == key]
try:
[period] = [p for p in data["periods"] if p["Type"] == key]
except ValueError:
_LOGGER.warning(
"IQVIA didn't return data for %s; trying again later", self.name
)
return
[rating] = [
i["label"]
for i in RATING_MAPPING