initial commit

custom_components/afvalwijzer/collector/deafvalapp.py

@@ -0,0 +1,57 @@
+from datetime import datetime
+import re
+
+import requests
+
+from ..common.main_functions import _waste_type_rename
+from ..const.const import _LOGGER, SENSOR_COLLECTORS_DEAFVALAPP
+
+
+def get_waste_data_raw(
+    provider,
+    postal_code,
+    street_number,
+    suffix,
+):
+    if provider not in SENSOR_COLLECTORS_DEAFVALAPP.keys():
+        raise ValueError(f"Invalid provider: {provider}, please verify")
+
+    corrected_postal_code_parts = re.search(r"(\d\d\d\d) ?([A-z][A-z])", postal_code)
+    corrected_postal_code = (
+        corrected_postal_code_parts[1] + corrected_postal_code_parts[2].upper()
+    )
+
+    try:
+        url = SENSOR_COLLECTORS_DEAFVALAPP[provider].format(
+            corrected_postal_code,
+            street_number,
+            suffix,
+        )
+        raw_response = requests.get(url)
+    except requests.exceptions.RequestException as err:
+        raise ValueError(err) from err
+
+    try:
+        response = raw_response.text
+    except ValueError as e:
+        raise ValueError(f"Invalid and/or no data received from {url}") from e
+
+    if not response:
+        _LOGGER.error("No waste data found!")
+        return
+
+    waste_data_raw = []
+
+    for rows in response.strip().split("\n"):
+        for ophaaldatum in rows.split(";")[1:-1]:
+            temp = {"type": _waste_type_rename(rows.split(";")[0].strip().lower())}
+            temp["date"] = datetime.strptime(ophaaldatum, "%d-%m-%Y").strftime(
+                "%Y-%m-%d"
+            )
+            waste_data_raw.append(temp)
+
+    return waste_data_raw
+
+
+if __name__ == "__main__":
+    print("Yell something at a mountain!")