evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Convert rest sensors to async using httpx (#41973)

Browse source
This commit is contained in:
J. Nick Koston 2020-10-16 19:21:13 -05:00 • committed by GitHub
parent 39adf14079
commit ad6ce5fa83
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
12 changed files with 1094 additions and 970 deletions
Download patch file Download diff file Expand all files Collapse all files

14
homeassistant/components/pvoutput/sensor.py
View file

@ -43,7 +43,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
)
def setup_platform(hass, config, add_entities, discovery_info=None):
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
"""Set up the PVOutput sensor."""
name = config.get(CONF_NAME)
api_key = config.get(CONF_API_KEY)
@ -54,13 +54,13 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
headers = {"X-Pvoutput-Apikey": api_key, "X-Pvoutput-SystemId": system_id}
rest = RestData(method, _ENDPOINT, auth, headers, payload, verify_ssl)
rest.update()
await rest.async_update()
if rest.data is None:
_LOGGER.error("Unable to fetch data from PVOutput")
return False
add_entities([PvoutputSensor(rest, name)], True)
async_add_entities([PvoutputSensor(rest, name)], True)
class PvoutputSensor(Entity):
@ -112,11 +112,15 @@ class PvoutputSensor(Entity):
ATTR_VOLTAGE: self.pvcoutput.voltage,
}
def update(self):
async def async_update(self):
"""Get the latest data from the PVOutput API and updates the state."""
try:
self.rest.update()
await self.rest.async_update()
self.pvcoutput = self.status._make(self.rest.data.split(","))
except TypeError:
self.pvcoutput = None
_LOGGER.error("Unable to fetch data from PVOutput. %s", self.rest.data)
async def async_will_remove_from_hass(self):
"""Shutdown the session."""
await self.rest.async_remove()