4e07ab1b32
* Move temperature conversions to entity base class (1/8) * Update integrations a-c * Leave old temperature conversion until all integrations are migrated * tweak * Use contextlib.suppress * Remove the MeasurableUnitEntity mixin * Address comments, add tests * Fix f-string * Drop deprecation warning from base entity class * Update with _attr-shorthand * Fix rebase mistakes * Fix additional rebase mistakes * Only report temperature conversion once * Fix additional rebase mistakes * Format homeassistant/components/bbox/sensor.py * Fix check for overidden _attr_state * Remove test workarounds from implementation * Remove useless None-check * Tweaks * Migrate new sensors a-c * Update climacell * Push deprecation of temperature conversion forward * Override __repr__ in SensorEntity * Include native_value in SensorEntity attributes * Pylint * Black * Black * Fix rebase mistakes * black * Fix rebase mistakes * Revert changes in august/sensor.py * Revert handling of unit converted restored state * Apply code review suggestion * Fix arlo test
106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
"""Sensor platform for Brottsplatskartan information."""
|
|
from collections import defaultdict
|
|
from datetime import timedelta
|
|
import logging
|
|
import uuid
|
|
|
|
import brottsplatskartan
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
|
|
from homeassistant.const import (
|
|
ATTR_ATTRIBUTION,
|
|
CONF_LATITUDE,
|
|
CONF_LONGITUDE,
|
|
CONF_NAME,
|
|
)
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_AREA = "area"
|
|
|
|
DEFAULT_NAME = "Brottsplatskartan"
|
|
|
|
SCAN_INTERVAL = timedelta(minutes=30)
|
|
|
|
AREAS = [
|
|
"Blekinge län",
|
|
"Dalarnas län",
|
|
"Gotlands län",
|
|
"Gävleborgs län",
|
|
"Hallands län",
|
|
"Jämtlands län",
|
|
"Jönköpings län",
|
|
"Kalmar län",
|
|
"Kronobergs län",
|
|
"Norrbottens län",
|
|
"Skåne län",
|
|
"Stockholms län",
|
|
"Södermanlands län",
|
|
"Uppsala län",
|
|
"Värmlands län",
|
|
"Västerbottens län",
|
|
"Västernorrlands län",
|
|
"Västmanlands län",
|
|
"Västra Götalands län",
|
|
"Örebro län",
|
|
"Östergötlands län",
|
|
]
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Inclusive(CONF_LATITUDE, "coordinates"): cv.latitude,
|
|
vol.Inclusive(CONF_LONGITUDE, "coordinates"): cv.longitude,
|
|
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
|
|
vol.Optional(CONF_AREA, default=[]): vol.All(cv.ensure_list, [vol.In(AREAS)]),
|
|
}
|
|
)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Brottsplatskartan platform."""
|
|
|
|
area = config.get(CONF_AREA)
|
|
latitude = config.get(CONF_LATITUDE, hass.config.latitude)
|
|
longitude = config.get(CONF_LONGITUDE, hass.config.longitude)
|
|
name = config[CONF_NAME]
|
|
|
|
# Every Home Assistant instance should have their own unique
|
|
# app parameter: https://brottsplatskartan.se/sida/api
|
|
app = f"ha-{uuid.getnode()}"
|
|
|
|
bpk = brottsplatskartan.BrottsplatsKartan(
|
|
app=app, area=area, latitude=latitude, longitude=longitude
|
|
)
|
|
|
|
add_entities([BrottsplatskartanSensor(bpk, name)], True)
|
|
|
|
|
|
class BrottsplatskartanSensor(SensorEntity):
|
|
"""Representation of a Brottsplatskartan Sensor."""
|
|
|
|
def __init__(self, bpk, name):
|
|
"""Initialize the Brottsplatskartan sensor."""
|
|
self._brottsplatskartan = bpk
|
|
self._attr_name = name
|
|
|
|
def update(self):
|
|
"""Update device state."""
|
|
|
|
incident_counts = defaultdict(int)
|
|
incidents = self._brottsplatskartan.get_incidents()
|
|
|
|
if incidents is False:
|
|
_LOGGER.debug("Problems fetching incidents")
|
|
return
|
|
|
|
for incident in incidents:
|
|
incident_type = incident.get("title_type")
|
|
incident_counts[incident_type] += 1
|
|
|
|
self._attr_extra_state_attributes = {
|
|
ATTR_ATTRIBUTION: brottsplatskartan.ATTRIBUTION
|
|
}
|
|
self._attr_extra_state_attributes.update(incident_counts)
|
|
self._attr_native_value = len(incidents)
|