Add SSAM source

custom_components/waste_collection_schedule/waste_collection_schedule/source/ssam_se.py

@@ -0,0 +1,62 @@
+# coding: utf-8
+from datetime import datetime
+import json
+from urllib.parse import urlencode
+
+import requests
+from waste_collection_schedule import Collection  # type: ignore[attr-defined]
+
+TITLE = "Lerum Vatten och Avlopp"
+DESCRIPTION = "Source for Lerum Vatten och Avlopp waste collection."
+URL = "https://edpfuture.ssam.se/FutureWeb/SimpleWastePickup/SimpleWastePickup"
+TEST_CASES = {
+    # "SSAM": {"street_address": "Stinavägen 3, Växjö"},
+    # "Polisen": {"street_address": "Sandgärdsgatan 31, Växjö"},
+    "Ekliden": {"street_address": "Kalvsvik Ekliden, Kalvsvik"},
+}
+
+
+class Source:
+    def __init__(self, street_address):
+        self._street_address = street_address
+
+    def fetch(self):
+        response = requests.post(
+            "https://edpfuture.ssam.se/FutureWeb/SimpleWastePickup/SearchAdress",
+            {"searchText": self._street_address},
+        )
+
+        address_data = json.loads(response.text)
+        address = None
+        if address_data["Succeeded"] and address_data["Succeeded"] is True:
+            if address_data["Buildings"] and len(address_data["Buildings"]) > 0:
+                address = address_data["Buildings"][0]
+
+        print(address)
+
+        if not address:
+            return []
+
+        query_params = urlencode({"address": address})
+        response = requests.get(
+            "https://edpfuture.ssam.se/FutureWeb/SimpleWastePickup/GetWastePickupSchedule?{}".format(
+                query_params
+            )
+        )
+        data = json.loads(response.text)
+
+        entries = []
+        for item in data["RhServices"]:
+            waste_type = item["WasteType"]
+            icon = "mdi:trash-can"
+            if waste_type == "Matavfall":
+                icon = "mdi:leaf"
+            next_pickup = item["NextWastePickup"]
+            try:
+                next_pickup_date = datetime.fromisoformat(next_pickup).date()
+            except ValueError as e:
+                next_pickup_date = datetime.strptime(next_pickup, "%b %Y").date()
+
+            entries.append(Collection(date=next_pickup_date, t=waste_type, icon=icon))
+
+        return entries

doc/source/ssam_se.md

@@ -0,0 +1,34 @@
+# SSAM
+
+TODO!
+
+Support for schedules provided by [Lerum Vatten och Avlopp](https://vatjanst.lerum.se/FutureWeb/SimpleWastePickup/SimpleWastePickup), serving the municipality of Lerum, Sweden.
+
+## Configuration via configuration.yaml
+
+```yaml
+waste_collection_schedule:
+  sources:
+    - name: lerum_se
+      args:
+        street_address: STREET_ADDRESS
+```
+
+### Configuration Variables
+
+**street_address**<br>
+*(string) (required)*
+
+## Example
+
+```yaml
+waste_collection_schedule:
+  sources:
+    - name: lerum_se
+      args:
+        street_address: Götebordsvägen 16, Lerum
+```
+
+## How to get the source argument
+
+The source argument is the address to the house with waste collection. The address can be tested [here](https://vatjanst.lerum.se/FutureWeb/SimpleWastePickup/SimpleWastePickup).