112 lines
3.2 KiB
Python
112 lines
3.2 KiB
Python
"""Support for the CO2signal platform."""
|
|
import logging
|
|
|
|
import CO2Signal
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
|
from homeassistant.const import (
|
|
ATTR_ATTRIBUTION,
|
|
CONF_LATITUDE,
|
|
CONF_LONGITUDE,
|
|
CONF_TOKEN,
|
|
)
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
CONF_COUNTRY_CODE = "country_code"
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ATTRIBUTION = "Data provided by CO2signal"
|
|
|
|
MSG_LOCATION = (
|
|
"Please use either coordinates or the country code. "
|
|
"For the coordinates, "
|
|
"you need to use both latitude and longitude."
|
|
)
|
|
CO2_INTENSITY_UNIT = "CO2eq/kWh"
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_TOKEN): cv.string,
|
|
vol.Inclusive(CONF_LATITUDE, "coords", msg=MSG_LOCATION): cv.latitude,
|
|
vol.Inclusive(CONF_LONGITUDE, "coords", msg=MSG_LOCATION): cv.longitude,
|
|
vol.Optional(CONF_COUNTRY_CODE): cv.string,
|
|
}
|
|
)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the CO2signal sensor."""
|
|
token = config[CONF_TOKEN]
|
|
lat = config.get(CONF_LATITUDE, hass.config.latitude)
|
|
lon = config.get(CONF_LONGITUDE, hass.config.longitude)
|
|
country_code = config.get(CONF_COUNTRY_CODE)
|
|
|
|
_LOGGER.debug("Setting up the sensor using the %s", country_code)
|
|
|
|
devs = []
|
|
|
|
devs.append(CO2Sensor(token, country_code, lat, lon))
|
|
add_entities(devs, True)
|
|
|
|
|
|
class CO2Sensor(Entity):
|
|
"""Implementation of the CO2Signal sensor."""
|
|
|
|
def __init__(self, token, country_code, lat, lon):
|
|
"""Initialize the sensor."""
|
|
self._token = token
|
|
self._country_code = country_code
|
|
self._latitude = lat
|
|
self._longitude = lon
|
|
self._data = None
|
|
|
|
if country_code is not None:
|
|
device_name = country_code
|
|
else:
|
|
device_name = f"{round(self._latitude, 2)}/{round(self._longitude, 2)}"
|
|
|
|
self._friendly_name = f"CO2 intensity - {device_name}"
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._friendly_name
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Icon to use in the frontend, if any."""
|
|
return "mdi:periodic-table-co2"
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the device."""
|
|
return self._data
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement of this entity, if any."""
|
|
return CO2_INTENSITY_UNIT
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes of the last update."""
|
|
return {ATTR_ATTRIBUTION: ATTRIBUTION}
|
|
|
|
def update(self):
|
|
"""Get the latest data and updates the states."""
|
|
|
|
_LOGGER.debug("Update data for %s", self._friendly_name)
|
|
|
|
if self._country_code is not None:
|
|
self._data = CO2Signal.get_latest_carbon_intensity(
|
|
self._token, country_code=self._country_code
|
|
)
|
|
else:
|
|
self._data = CO2Signal.get_latest_carbon_intensity(
|
|
self._token, latitude=self._latitude, longitude=self._longitude
|
|
)
|
|
|
|
self._data = round(self._data, 2)
|