f2d9e6f70c
* Removed pylint disable on unused after updating CI files that were out of date. * Pylint still fails due to bug on DOMAIN import. Added disable check. * Addressed PR comments * Added import for ClientError to test_config_flow.py Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
36 lines
881 B
Python
36 lines
881 B
Python
"""Support for Aurora Forecast sensor."""
|
|
import logging
|
|
|
|
from homeassistant.const import PERCENTAGE
|
|
|
|
from . import AuroraEntity
|
|
from .const import COORDINATOR, DOMAIN
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(hass, entry, async_add_entries):
|
|
"""Set up the sensor platform."""
|
|
coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR]
|
|
|
|
entity = AuroraSensor(
|
|
coordinator=coordinator,
|
|
name=f"{coordinator.name} Aurora Visibility %",
|
|
icon="mdi:gauge",
|
|
)
|
|
|
|
async_add_entries([entity])
|
|
|
|
|
|
class AuroraSensor(AuroraEntity):
|
|
"""Implementation of an aurora sensor."""
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return % chance the aurora is visible."""
|
|
return self.coordinator.data
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measure."""
|
|
return PERCENTAGE
|