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

Fix Flu Near You exception re: stale coroutines (#34880)

Browse Source
This commit is contained in:
Aaron Bach 2020-04-29 13:29:44 -06:00 committed by GitHub
parent 481b2035db
commit b8a5597d3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/components/flunearyou/__init__.py
View File

@ -131,15 +131,6 @@ class FluNearYouData:
self.latitude = latitude self.latitude = latitude
self.longitude = longitude self.longitude = longitude
self._api_coros = {
CATEGORY_CDC_REPORT: self._client.cdc_reports.status_by_coordinates(
latitude, longitude
),
CATEGORY_USER_REPORT: self._client.user_reports.status_by_coordinates(
latitude, longitude
),
}
self._api_category_count = { self._api_category_count = {
CATEGORY_CDC_REPORT: 0, CATEGORY_CDC_REPORT: 0,
CATEGORY_USER_REPORT: 0, CATEGORY_USER_REPORT: 0,
@ -155,8 +146,17 @@ class FluNearYouData:
if self._api_category_count[api_category] == 0: if self._api_category_count[api_category] == 0:
return return
if api_category == CATEGORY_CDC_REPORT:
api_coro = self._client.cdc_reports.status_by_coordinates(
self.latitude, self.longitude
)
else:
api_coro = self._client.user_reports.status_by_coordinates(
self.latitude, self.longitude
)
try: try:
self.data[api_category] = await self._api_coros[api_category] self.data[api_category] = await api_coro
except FluNearYouError as err: except FluNearYouError as err:
LOGGER.error("Unable to get %s data: %s", api_category, err) LOGGER.error("Unable to get %s data: %s", api_category, err)
self.data[api_category] = None self.data[api_category] = None
@ -200,7 +200,7 @@ class FluNearYouData:
"""Update Flu Near You data.""" """Update Flu Near You data."""
tasks = [ tasks = [
self._async_get_data_from_api(api_category) self._async_get_data_from_api(api_category)
for api_category in self._api_coros for api_category in self._api_category_count
] ]
await asyncio.gather(*tasks) await asyncio.gather(*tasks)