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

Optimize foreacast.io API calls

Browse Source
This commit is contained in:
Paulus Schoutsen 2016-06-01 09:17:15 -07:00
parent 1f602be80a
commit f8bdc835f8
1 changed files with 57 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
homeassistant/components/sensor/forecast.py
View File

@ -131,26 +131,38 @@ class ForeCastSensor(Entity):
"""Return the unit system of this entity.""" """Return the unit system of this entity."""
return self._unit_system return self._unit_system
# pylint: disable=too-many-branches # pylint: disable=too-many-branches,too-many-statements
def update(self): def update(self):
"""Get the latest data from Forecast.io and updates the states.""" """Get the latest data from Forecast.io and updates the states."""
import forecastio import forecastio
self.forecast_client.update() self.forecast_client.update()
data = self.forecast_client.data.currently()
data_minutely = self.forecast_client.data.minutely() try:
data_hourly = self.forecast_client.data.hourly() if self.type == 'minutely_summary':
data_daily = self.forecast_client.data.daily() self.forecast_client.update_minutely()
self._state = self.forecast_client.data_minutely.summary
return
elif self.type == 'hourly_summary':
self.forecast_client.update_hourly()
self._state = self.forecast_client.data_hourly.summary
return
elif self.type == 'daily_summary':
self.forecast_client.update_daily()
self._state = self.forecast_client.data_daily.summary
return
except forecastio.utils.PropertyUnavailable:
return
self.forecast_client.update_currently()
data = self.forecast_client.data_currently
try: try:
if self.type == 'summary': if self.type == 'summary':
self._state = data.summary self._state = data.summary
elif self.type == 'minutely_summary':
self._state = data_minutely.summary
elif self.type == 'hourly_summary':
self._state = data_hourly.summary
elif self.type == 'daily_summary':
self._state = data_daily.summary
elif self.type == 'icon': elif self.type == 'icon':
self._state = data.icon self._state = data.icon
elif self.type == 'nearest_storm_distance': elif self.type == 'nearest_storm_distance':
@ -191,14 +203,22 @@ class ForeCastSensor(Entity):
class ForeCastData(object): class ForeCastData(object):
"""Gets the latest data from Forecast.io.""" """Gets the latest data from Forecast.io."""
# pylint: disable=too-many-instance-attributes
def __init__(self, api_key, latitude, longitude, units): def __init__(self, api_key, latitude, longitude, units):
"""Initialize the data object.""" """Initialize the data object."""
self._api_key = api_key self._api_key = api_key
self.latitude = latitude self.latitude = latitude
self.longitude = longitude self.longitude = longitude
self.units = units
self.data = None self.data = None
self.unit_system = None self.unit_system = None
self.units = units self.data_currently = None
self.data_minutely = None
self.data_hourly = None
self.data_daily = None
self.update() self.update()
@Throttle(MIN_TIME_BETWEEN_UPDATES) @Throttle(MIN_TIME_BETWEEN_UPDATES)
@ -206,9 +226,28 @@ class ForeCastData(object):
"""Get the latest data from Forecast.io.""" """Get the latest data from Forecast.io."""
import forecastio import forecastio
forecast = forecastio.load_forecast(self._api_key, self.data = forecastio.load_forecast(self._api_key,
self.latitude, self.latitude,
self.longitude, self.longitude,
units=self.units) units=self.units)
self.data = forecast self.unit_system = self.data.json['flags']['units']
self.unit_system = forecast.json['flags']['units']
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update_currently(self):
"""Update currently data."""
self.data_currently = self.data.currently()
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update_minutely(self):
"""Update minutely data."""
self.data_minutely = self.data.minutely()
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update_hourly(self):
"""Update hourly data."""
self.data_hourly = self.data.hourly()
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update_daily(self):
"""Update daily data."""
self.data_daily = self.data.daily()