initial commit

custom_components/afvalwijzer/sensor.py

@@ -0,0 +1,146 @@
+#!/usr/bin/env python3
+"""
+Sensor component Afvalwijzer
+Author: Bram van Dartel - xirixiz
+"""
+
+from functools import partial
+
+from homeassistant.components.sensor import PLATFORM_SCHEMA
+import homeassistant.helpers.config_validation as cv
+from homeassistant.util import Throttle
+import voluptuous as vol
+
+from .collector.main_collector import MainCollector
+from .const.const import (
+    _LOGGER,
+    CONF_COLLECTOR,
+    CONF_DEFAULT_LABEL,
+    CONF_EXCLUDE_LIST,
+    CONF_EXCLUDE_PICKUP_TODAY,
+    CONF_ID,
+    CONF_POSTAL_CODE,
+    CONF_STREET_NUMBER,
+    CONF_SUFFIX,
+    MIN_TIME_BETWEEN_UPDATES,
+    PARALLEL_UPDATES,
+    SCAN_INTERVAL,
+    STARTUP_MESSAGE,
+)
+from .sensor_custom import CustomSensor
+from .sensor_provider import ProviderSensor
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
+    {
+        vol.Optional(
+            CONF_COLLECTOR, default="mijnafvalwijzer"
+        ): cv.string,
+        vol.Required(CONF_POSTAL_CODE, default="1234AB"): cv.string,
+        vol.Required(CONF_STREET_NUMBER, default="5"): cv.string,
+        vol.Optional(CONF_SUFFIX, default=""): cv.string,
+        vol.Optional(CONF_EXCLUDE_PICKUP_TODAY, default="true"): cv.string,
+        vol.Optional(CONF_EXCLUDE_LIST, default=""): cv.string,
+        vol.Optional(CONF_DEFAULT_LABEL, default="Geen"): cv.string,
+        vol.Optional(CONF_ID.strip().lower(), default=""): cv.string,
+    }
+)
+
+_LOGGER.info(STARTUP_MESSAGE)
+
+
+async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
+    provider = config.get(CONF_COLLECTOR)
+    postal_code = config.get(CONF_POSTAL_CODE)
+    street_number = config.get(CONF_STREET_NUMBER)
+    suffix = config.get(CONF_SUFFIX)
+    exclude_pickup_today = config.get(CONF_EXCLUDE_PICKUP_TODAY)
+    exclude_list = config.get(CONF_EXCLUDE_LIST)
+    default_label = config.get(CONF_DEFAULT_LABEL)
+
+    _LOGGER.debug(f"Afvalwijzer provider = {provider}")
+    _LOGGER.debug(f"Afvalwijzer zipcode = {postal_code}")
+    _LOGGER.debug(f"Afvalwijzer street_number = {street_number}")
+
+    try:
+        collector = await hass.async_add_executor_job(
+            partial(
+                MainCollector,
+                provider,
+                postal_code,
+                street_number,
+                suffix,
+                exclude_pickup_today,
+                exclude_list,
+                default_label,
+            )
+        )
+    except ValueError as err:
+        _LOGGER.error(f"Check afvalwijzer platform settings {err.args}")
+
+    fetch_data = AfvalwijzerData(config)
+
+    waste_types_provider = collector.waste_types_provider
+    _LOGGER.debug(f"Generating waste_types_provider list = {waste_types_provider}")
+    waste_types_custom = collector.waste_types_custom
+    _LOGGER.debug(f"Generating waste_types_custom list = {waste_types_custom}")
+
+    entities = []
+
+    for waste_type in waste_types_provider:
+        _LOGGER.debug(f"Adding sensor provider: {waste_type}")
+        entities.append(ProviderSensor(hass, waste_type, fetch_data, config))
+    for waste_type in waste_types_custom:
+        _LOGGER.debug(f"Adding sensor custom: {waste_type}")
+        entities.append(CustomSensor(hass, waste_type, fetch_data, config))
+
+    _LOGGER.debug(f"Entities appended = {entities}")
+    async_add_entities(entities)
+
+
+class AfvalwijzerData(object):
+    def __init__(self, config):
+        self.config = config
+
+    @Throttle(MIN_TIME_BETWEEN_UPDATES)
+    def update(self):
+        provider = self.config.get(CONF_COLLECTOR)
+        postal_code = self.config.get(CONF_POSTAL_CODE)
+        street_number = self.config.get(CONF_STREET_NUMBER)
+        suffix = self.config.get(CONF_SUFFIX)
+        exclude_pickup_today = self.config.get(CONF_EXCLUDE_PICKUP_TODAY)
+        default_label = self.config.get(CONF_DEFAULT_LABEL)
+        exclude_list = self.config.get(CONF_EXCLUDE_LIST)
+
+        try:
+            collector = MainCollector(
+                provider,
+                postal_code,
+                street_number,
+                suffix,
+                exclude_pickup_today,
+                exclude_list,
+                default_label,
+            )
+        except ValueError as err:
+            _LOGGER.error(f"Check afvalwijzer platform settings {err.args}")
+
+        # waste data provider update - with today
+        try:
+            self.waste_data_with_today = collector.waste_data_with_today
+        except ValueError as err:
+            _LOGGER.error(f"Check waste_data_provider {err.args}")
+            self.waste_data_with_today = default_label
+
+        # waste data provider update - without today
+        try:
+            self.waste_data_without_today = collector.waste_data_without_today
+        except ValueError as err:
+            _LOGGER.error(f"Check waste_data_provider {err.args}")
+            self.waste_data_without_today = default_label
+
+        # waste data custom update
+        try:
+            self.waste_data_custom = collector.waste_data_custom
+        except ValueError as err:
+            _LOGGER.error(f"Check waste_data_custom {err.args}")
+            self.waste_data_custom = default_label