✨ ADD Cantebury CIty Council support

custom_components/waste_collection_schedule/waste_collection_schedule/source/canterbury_gov_uk.py

@@ -0,0 +1,81 @@
+import logging
+from datetime import datetime
+import json
+
+import requests
+from waste_collection_schedule import Collection
+
+TITLE = "canterbury.gov.uk"
+DESCRIPTION = (
+    "Source for canterbury.gov.uk services for canterbury"
+)
+URL = "canterbury.gov.uk"
+TEST_CASES = {
+    "houseNumber": {"post_code": "ct68tu", "number": "12"},
+    "houseName": {"post_code": "ct68tu", "number": "THE OLD BAKEHOUSE DANCE STUDIOS"},
+}
+
+API_URLS = {
+    "address_search": "https://trsewmllv7.execute-api.eu-west-2.amazonaws.com/dev/address",
+    "collection":  "https://zbr7r13ke2.execute-api.eu-west-2.amazonaws.com/Beta/get-bin-dates",
+}
+
+ICONS = {
+    "General": "mdi:trash-can",
+    "Recycling": "mdi:recycle",
+    "Food": "mdi:food-apple",
+    "Garden": "mdi:shovel",
+}
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class Source:
+    def __init__(self, post_code: str, number: str):
+        self._post_code = post_code
+        self._number = str(number).capitalize()
+
+    def fetch(self):
+        # fetch location id
+        r = requests.get(
+            API_URLS["address_search"], params={"postcode": self._post_code, "type": "standard"}
+        )
+        r.raise_for_status()
+        addresses = r.json()
+        
+        address_ids = [
+            x for x in addresses["candidates"]
+            if x["attributes"]["PAO_START_NUMBER"].lower() == self._number.lower() or
+            x["attributes"]["PAO_TEXT"].lower() == self._number.lower()
+        ]
+
+        if len(address_ids) == 0:
+            raise Exception(f"Could not find address {self._post_code} {self._number}")
+
+        q = str(API_URLS["collection"])
+        r = requests.post(q,json={"uprn": address_ids[0]["attributes"]["UPRN"],"usrn": address_ids[0]["attributes"]["USRN"]} )
+        r.raise_for_status()
+
+        collectionsRaw = json.loads(r.json()["dates"])
+        collections  = {
+                "General": collectionsRaw["blackBinDay"], 
+                "Recycling": collectionsRaw["recyclingBinDay"], 
+                "Food": collectionsRaw["foodBinDay"],
+                "Garden": collectionsRaw["gardenBinDay"],
+                }
+        entries = []
+
+        for collection in collections:
+            if len(collections[collection]) <= 0: continue
+            for date in collections[collection]:
+                entries.append(
+                    Collection(
+                        date=datetime.strptime(
+                        date, "%Y-%m-%dT%H:%M:%S"
+                        ).date(),
+                        t=collection,
+                        icon=ICONS.get(collection),
+                    )
+                )
+
+        return entries

doc/source/canterbury_gov_uk.md

@@ -0,0 +1,44 @@
+# Canterbury City Council
+
+Support for schedules provided by [Canterbury City Council](https://www.canterbury.gov.uk/bins-and-waste/find-your-bin-collection-dates/), serving Canterbury (UK) and parts of Kent.
+
+## Configuration via configuration.yaml
+
+```yaml
+waste_collection_schedule:
+    sources:
+    - name: canterbury_gov_uk
+      args:
+        post_code: POST_CODE
+        number: NUMBER
+
+```
+
+### Configuration Variables
+
+**POST_CODE**<br>
+*(string) (required)*
+
+**NUMBER**<br>
+*(string) (required)*
+
+
+## Examples
+
+```yaml
+waste_collection_schedule:
+    sources:
+    - name: canterbury_gov_uk
+      args:
+        post_code: "ct68ru"
+        number: "63"
+```
+ 
+```yaml
+waste_collection_schedule:
+    sources:
+    - name: canterbury_gov_uk
+      args:
+        post_code: "ct68ru"
+        number: "KOWLOON"
+```