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

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

164
homeassistant/components/yr/sensor.py
View file

@ -12,55 +12,63 @@ import voluptuous as vol
import homeassistant.helpers.config_validation as cv
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import (
CONF_LATITUDE, CONF_LONGITUDE, CONF_ELEVATION, CONF_MONITORED_CONDITIONS,
ATTR_ATTRIBUTION, CONF_NAME)
CONF_LATITUDE,
CONF_LONGITUDE,
CONF_ELEVATION,
CONF_MONITORED_CONDITIONS,
ATTR_ATTRIBUTION,
CONF_NAME,
)
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import (async_track_utc_time_change,
async_call_later)
from homeassistant.helpers.event import async_track_utc_time_change, async_call_later
from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__)
ATTRIBUTION = "Weather forecast from met.no, delivered by the Norwegian " \
"Meteorological Institute."
ATTRIBUTION = (
"Weather forecast from met.no, delivered by the Norwegian "
"Meteorological Institute."
)
# https://api.met.no/license_data.html
SENSOR_TYPES = {
'symbol': ['Symbol', None],
'precipitation': ['Precipitation', 'mm'],
'temperature': ['Temperature', '°C'],
'windSpeed': ['Wind speed', 'm/s'],
'windGust': ['Wind gust', 'm/s'],
'pressure': ['Pressure', 'hPa'],
'windDirection': ['Wind direction', '°'],
'humidity': ['Humidity', '%'],
'fog': ['Fog', '%'],
'cloudiness': ['Cloudiness', '%'],
'lowClouds': ['Low clouds', '%'],
'mediumClouds': ['Medium clouds', '%'],
'highClouds': ['High clouds', '%'],
'dewpointTemperature': ['Dewpoint temperature', '°C'],
"symbol": ["Symbol", None],
"precipitation": ["Precipitation", "mm"],
"temperature": ["Temperature", "°C"],
"windSpeed": ["Wind speed", "m/s"],
"windGust": ["Wind gust", "m/s"],
"pressure": ["Pressure", "hPa"],
"windDirection": ["Wind direction", "°"],
"humidity": ["Humidity", "%"],
"fog": ["Fog", "%"],
"cloudiness": ["Cloudiness", "%"],
"lowClouds": ["Low clouds", "%"],
"mediumClouds": ["Medium clouds", "%"],
"highClouds": ["High clouds", "%"],
"dewpointTemperature": ["Dewpoint temperature", "°C"],
}
CONF_FORECAST = 'forecast'
CONF_FORECAST = "forecast"
DEFAULT_FORECAST = 0
DEFAULT_NAME = 'yr'
DEFAULT_NAME = "yr"
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_ELEVATION): vol.Coerce(int),
vol.Optional(CONF_FORECAST, default=DEFAULT_FORECAST): vol.Coerce(int),
vol.Optional(CONF_LATITUDE): cv.latitude,
vol.Optional(CONF_LONGITUDE): cv.longitude,
vol.Optional(CONF_MONITORED_CONDITIONS, default=['symbol']):
vol.All(cv.ensure_list, vol.Length(min=1), [vol.In(SENSOR_TYPES)]),
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
})
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{
vol.Optional(CONF_ELEVATION): vol.Coerce(int),
vol.Optional(CONF_FORECAST, default=DEFAULT_FORECAST): vol.Coerce(int),
vol.Optional(CONF_LATITUDE): cv.latitude,
vol.Optional(CONF_LONGITUDE): cv.longitude,
vol.Optional(CONF_MONITORED_CONDITIONS, default=["symbol"]): vol.All(
cv.ensure_list, vol.Length(min=1), [vol.In(SENSOR_TYPES)]
),
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
}
)
async def async_setup_platform(hass, config, async_add_entities,
discovery_info=None):
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
"""Set up the Yr.no sensor."""
elevation = config.get(CONF_ELEVATION, hass.config.elevation or 0)
forecast = config.get(CONF_FORECAST)
@ -72,11 +80,7 @@ async def async_setup_platform(hass, config, async_add_entities,
_LOGGER.error("Latitude or longitude not set in Home Assistant config")
return False
coordinates = {
'lat': str(latitude),
'lon': str(longitude),
'msl': str(elevation),
}
coordinates = {"lat": str(latitude), "lon": str(longitude), "msl": str(elevation)}
dev = []
for sensor_type in config[CONF_MONITORED_CONDITIONS]:
@ -84,8 +88,7 @@ async def async_setup_platform(hass, config, async_add_entities,
async_add_entities(dev)
weather = YrData(hass, coordinates, forecast, dev)
async_track_utc_time_change(hass, weather.updating_devices,
minute=31, second=0)
async_track_utc_time_change(hass, weather.updating_devices, minute=31, second=0)
await weather.fetching_data()
@ -103,7 +106,7 @@ class YrSensor(Entity):
@property
def name(self):
"""Return the name of the sensor."""
return '{} {}'.format(self.client_name, self._name)
return "{} {}".format(self.client_name, self._name)
@property
def state(self):
@ -118,17 +121,17 @@ class YrSensor(Entity):
@property
def entity_picture(self):
"""Weather symbol if type is symbol."""
if self.type != 'symbol':
if self.type != "symbol":
return None
return "https://api.met.no/weatherapi/weathericon/1.1/" \
"?symbol={0};content_type=image/png".format(self._state)
return (
"https://api.met.no/weatherapi/weathericon/1.1/"
"?symbol={0};content_type=image/png".format(self._state)
)
@property
def device_state_attributes(self):
"""Return the state attributes."""
return {
ATTR_ATTRIBUTION: ATTRIBUTION,
}
return {ATTR_ATTRIBUTION: ATTRIBUTION}
@property
def unit_of_measurement(self):
@ -141,8 +144,9 @@ class YrData:
def __init__(self, hass, coordinates, forecast, devices):
"""Initialize the data object."""
self._url = 'https://aa015h6buqvih86i1.api.met.no/'\
'weatherapi/locationforecast/1.9/'
self._url = (
"https://aa015h6buqvih86i1.api.met.no/" "weatherapi/locationforecast/1.9/"
)
self._urlparams = coordinates
self._forecast = forecast
self.devices = devices
@ -157,14 +161,14 @@ class YrData:
"""Retry in 15 to 20 minutes."""
minutes = 15 + randrange(6)
_LOGGER.error("Retrying in %i minutes: %s", minutes, err)
async_call_later(self.hass, minutes*60, self.fetching_data)
async_call_later(self.hass, minutes * 60, self.fetching_data)
try:
websession = async_get_clientsession(self.hass)
with async_timeout.timeout(10):
resp = await websession.get(
self._url, params=self._urlparams)
resp = await websession.get(self._url, params=self._urlparams)
if resp.status != 200:
try_again('{} returned {}'.format(resp.url, resp.status))
try_again("{} returned {}".format(resp.url, resp.status))
return
text = await resp.text()
@ -173,13 +177,13 @@ class YrData:
return
try:
self.data = xmltodict.parse(text)['weatherdata']
self.data = xmltodict.parse(text)["weatherdata"]
except (ExpatError, IndexError) as err:
try_again(err)
return
await self.updating_devices()
async_call_later(self.hass, 60*60, self.fetching_data)
async_call_later(self.hass, 60 * 60, self.fetching_data)
async def updating_devices(self, *_):
"""Find the current data from self.data."""
@ -197,16 +201,17 @@ class YrData:
ordered_entries = []
for time_entry in self.data['product']['time']:
valid_from = dt_util.parse_datetime(time_entry['@from'])
valid_to = dt_util.parse_datetime(time_entry['@to'])
for time_entry in self.data["product"]["time"]:
valid_from = dt_util.parse_datetime(time_entry["@from"])
valid_to = dt_util.parse_datetime(time_entry["@to"])
if now >= valid_to:
# Has already passed. Never select this.
continue
average_dist = (abs((valid_to - forecast_time).total_seconds()) +
abs((valid_from - forecast_time).total_seconds()))
average_dist = abs((valid_to - forecast_time).total_seconds()) + abs(
(valid_from - forecast_time).total_seconds()
)
ordered_entries.append((average_dist, time_entry))
@ -219,25 +224,34 @@ class YrData:
new_state = None
for (_, selected_time_entry) in ordered_entries:
loc_data = selected_time_entry['location']
loc_data = selected_time_entry["location"]
if dev.type not in loc_data:
continue
if dev.type == 'precipitation':
new_state = loc_data[dev.type]['@value']
elif dev.type == 'symbol':
new_state = loc_data[dev.type]['@number']
elif dev.type in ('temperature', 'pressure', 'humidity',
'dewpointTemperature'):
new_state = loc_data[dev.type]['@value']
elif dev.type in ('windSpeed', 'windGust'):
new_state = loc_data[dev.type]['@mps']
elif dev.type == 'windDirection':
new_state = float(loc_data[dev.type]['@deg'])
elif dev.type in ('fog', 'cloudiness', 'lowClouds',
'mediumClouds', 'highClouds'):
new_state = loc_data[dev.type]['@percent']
if dev.type == "precipitation":
new_state = loc_data[dev.type]["@value"]
elif dev.type == "symbol":
new_state = loc_data[dev.type]["@number"]
elif dev.type in (
"temperature",
"pressure",
"humidity",
"dewpointTemperature",
):
new_state = loc_data[dev.type]["@value"]
elif dev.type in ("windSpeed", "windGust"):
new_state = loc_data[dev.type]["@mps"]
elif dev.type == "windDirection":
new_state = float(loc_data[dev.type]["@deg"])
elif dev.type in (
"fog",
"cloudiness",
"lowClouds",
"mediumClouds",
"highClouds",
):
new_state = loc_data[dev.type]["@percent"]
break