80136f3591
* Change datetime.now() to dt_util.now() in cases where the functionality should stay the same These changes should not affect the functionality, rather cleanup our codebase. In general we would like integrations to not to use datetime.now() unless there's a very good reason for it, rather use our own dt_util.now() which makes the code aware of our current time zone. * Use datetime.utcnow() for season sensor to get offset-naive utc time * Revert "Use datetime.utcnow() for season sensor to get offset-naive utc time" This reverts commit 5f36463d9c7d52f8e11ffcec7e57dfbc7b21bdd1. * BOM sensor last_updated should be UTC as well * Run black * Remove unused last_partition_update variable
165 lines
4.3 KiB
Python
165 lines
4.3 KiB
Python
"""Demo platform that offers fake meteorological data."""
|
|
from datetime import timedelta
|
|
|
|
from homeassistant.components.weather import (
|
|
ATTR_FORECAST_CONDITION,
|
|
ATTR_FORECAST_PRECIPITATION,
|
|
ATTR_FORECAST_TEMP,
|
|
ATTR_FORECAST_TEMP_LOW,
|
|
ATTR_FORECAST_TIME,
|
|
WeatherEntity,
|
|
)
|
|
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
CONDITION_CLASSES = {
|
|
"cloudy": [],
|
|
"fog": [],
|
|
"hail": [],
|
|
"lightning": [],
|
|
"lightning-rainy": [],
|
|
"partlycloudy": [],
|
|
"pouring": [],
|
|
"rainy": ["shower rain"],
|
|
"snowy": [],
|
|
"snowy-rainy": [],
|
|
"sunny": ["sunshine"],
|
|
"windy": [],
|
|
"windy-variant": [],
|
|
"exceptional": [],
|
|
}
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Demo weather."""
|
|
add_entities(
|
|
[
|
|
DemoWeather(
|
|
"South",
|
|
"Sunshine",
|
|
21.6414,
|
|
92,
|
|
1099,
|
|
0.5,
|
|
TEMP_CELSIUS,
|
|
[
|
|
["rainy", 1, 22, 15],
|
|
["rainy", 5, 19, 8],
|
|
["cloudy", 0, 15, 9],
|
|
["sunny", 0, 12, 6],
|
|
["partlycloudy", 2, 14, 7],
|
|
["rainy", 15, 18, 7],
|
|
["fog", 0.2, 21, 12],
|
|
],
|
|
),
|
|
DemoWeather(
|
|
"North",
|
|
"Shower rain",
|
|
-12,
|
|
54,
|
|
987,
|
|
4.8,
|
|
TEMP_FAHRENHEIT,
|
|
[
|
|
["snowy", 2, -10, -15],
|
|
["partlycloudy", 1, -13, -14],
|
|
["sunny", 0, -18, -22],
|
|
["sunny", 0.1, -23, -23],
|
|
["snowy", 4, -19, -20],
|
|
["sunny", 0.3, -14, -19],
|
|
["sunny", 0, -9, -12],
|
|
],
|
|
),
|
|
]
|
|
)
|
|
|
|
|
|
class DemoWeather(WeatherEntity):
|
|
"""Representation of a weather condition."""
|
|
|
|
def __init__(
|
|
self,
|
|
name,
|
|
condition,
|
|
temperature,
|
|
humidity,
|
|
pressure,
|
|
wind_speed,
|
|
temperature_unit,
|
|
forecast,
|
|
):
|
|
"""Initialize the Demo weather."""
|
|
self._name = name
|
|
self._condition = condition
|
|
self._temperature = temperature
|
|
self._temperature_unit = temperature_unit
|
|
self._humidity = humidity
|
|
self._pressure = pressure
|
|
self._wind_speed = wind_speed
|
|
self._forecast = forecast
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return "{} {}".format("Demo Weather", self._name)
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""No polling needed for a demo weather condition."""
|
|
return False
|
|
|
|
@property
|
|
def temperature(self):
|
|
"""Return the temperature."""
|
|
return self._temperature
|
|
|
|
@property
|
|
def temperature_unit(self):
|
|
"""Return the unit of measurement."""
|
|
return self._temperature_unit
|
|
|
|
@property
|
|
def humidity(self):
|
|
"""Return the humidity."""
|
|
return self._humidity
|
|
|
|
@property
|
|
def wind_speed(self):
|
|
"""Return the wind speed."""
|
|
return self._wind_speed
|
|
|
|
@property
|
|
def pressure(self):
|
|
"""Return the pressure."""
|
|
return self._pressure
|
|
|
|
@property
|
|
def condition(self):
|
|
"""Return the weather condition."""
|
|
return [
|
|
k for k, v in CONDITION_CLASSES.items() if self._condition.lower() in v
|
|
][0]
|
|
|
|
@property
|
|
def attribution(self):
|
|
"""Return the attribution."""
|
|
return "Powered by Home Assistant"
|
|
|
|
@property
|
|
def forecast(self):
|
|
"""Return the forecast."""
|
|
reftime = dt_util.now().replace(hour=16, minute=00)
|
|
|
|
forecast_data = []
|
|
for entry in self._forecast:
|
|
data_dict = {
|
|
ATTR_FORECAST_TIME: reftime.isoformat(),
|
|
ATTR_FORECAST_CONDITION: entry[0],
|
|
ATTR_FORECAST_PRECIPITATION: entry[1],
|
|
ATTR_FORECAST_TEMP: entry[2],
|
|
ATTR_FORECAST_TEMP_LOW: entry[3],
|
|
}
|
|
reftime = reftime + timedelta(hours=4)
|
|
forecast_data.append(data_dict)
|
|
|
|
return forecast_data
|