initial commit

custom_components/afvalwijzer/common/next_sensor_data.py

@@ -0,0 +1,76 @@
+from datetime import datetime
+
+from ..const.const import _LOGGER
+
+
+class NextSensorData(object):
+
+    ##########################################################################
+    #  INIT
+    ##########################################################################
+    def __init__(self, waste_data_after_date_selected, default_label):
+        self.waste_data_after_date_selected = sorted(
+            waste_data_after_date_selected, key=lambda d: d["date"]
+        )
+
+        TODAY = datetime.now().strftime("%d-%m-%Y")
+        self.today_date = datetime.strptime(TODAY, "%d-%m-%Y")
+        self.default_label = default_label
+
+        self.next_waste_date = self.__get_next_waste_date()
+        self.next_waste_in_days = self.__get_next_waste_in_days()
+        self.next_waste_type = self.__get_next_waste_type()
+
+        self.data = self._gen_next_sensor_data()
+
+    ##########################################################################
+    #  GENERATE NEXT SENSOR(S)
+    ##########################################################################
+
+    # Generate sensor next_waste_date
+    def __get_next_waste_date(self):
+        next_waste_date = self.default_label
+        try:
+            next_waste_date = self.waste_data_after_date_selected[0]["date"]
+        except Exception as err:
+            _LOGGER.error(f"Other error occurred _get_next_waste_date: {err}")
+        return next_waste_date
+
+    # Generate sensor next_waste_in_days
+    def __get_next_waste_in_days(self):
+        next_waste_in_days = self.default_label
+        try:
+            next_waste_in_days = abs(self.today_date - self.next_waste_date).days  # type: ignore
+        except Exception as err:
+            _LOGGER.error(f"Other error occurred _get_next_waste_in_days: {err}")
+        return next_waste_in_days
+
+    # Generate sensor next_waste_type
+    def __get_next_waste_type(self):
+        next_waste_type = []
+        try:
+            for waste in self.waste_data_after_date_selected:
+                item_date = waste["date"]
+                if item_date == self.next_waste_date:
+                    item_name = waste["type"]
+                    next_waste_type.append(item_name)
+            if not next_waste_type:
+                next_waste_type.append(self.default_label)
+        except Exception as err:
+            _LOGGER.error(f"Other error occurred _get_next_waste_type: {err}")
+        return next_waste_type
+
+    # Generate sensor data for custom sensors
+    def _gen_next_sensor_data(self):
+        next_sensor = {}
+        try:
+            next_sensor["next_date"] = self.next_waste_date
+            next_sensor["next_in_days"] = self.next_waste_in_days
+            next_sensor["next_type"] = ", ".join(self.next_waste_type)
+        except Exception as err:
+            _LOGGER.error(f"Other error occurred _gen_next_sensor_data: {err}")
+        return next_sensor
+
+    @property
+    def next_sensor_data(self):
+        return self.data