reformatting + comment for parameter + additional test case

mampfes · Jul 29, 2023 · f2bd445 · f2bd445
1 parent adb332c
commit f2bd445
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 15 deletions.
diff --git a/custom_components/waste_collection_schedule/waste_collection_schedule/source/hcc_govt_nz.py b/custom_components/waste_collection_schedule/waste_collection_schedule/source/hcc_govt_nz.py
@@ -1,19 +1,18 @@
 import datetime
+
 from waste_collection_schedule import Collection
 from waste_collection_schedule.service.SSLError import get_legacy_session
 
 TITLE = "Hamilton City Council"
 DESCRIPTION = "Source script for Hamilton City Council"
 URL = "https://www.fightthelandfill.co.nz/"
 TEST_CASES = {
-    "1 Hamilton Parade": {"address": "1 Hamilton Parade"}
+    "1 Hamilton Parade": {"address": "1 Hamilton Parade"},
+    "221b fox Street": {"address": "221b fox Street"},
 }
 
 API_URL = "https://api.hcc.govt.nz/FightTheLandFill/get_Collection_Dates"
-ICON_MAP = {
-    "Rubbish": "mdi:trash-can",
-    "Recycling": "mdi:recycle"
-}
+ICON_MAP = {"Rubbish": "mdi:trash-can", "Recycling": "mdi:recycle"}
 
 
 class Source:
@@ -24,23 +23,28 @@ def fetch(self):
 
         s = get_legacy_session()
 
-        r = s.get(API_URL,
+        r = s.get(
+            API_URL,
             params={"address_string": self.address},
         )
         json = r.json()
 
-        #Extract entries from RedBin/YellowBin fields
+        # Extract entries from RedBin/YellowBin fields
         entries = [
             Collection(
-                date = datetime.datetime.strptime(json[0]["RedBin"], "%Y-%m-%dT%H:%M:%S").date(),
-                t = "Rubbish",
-                icon = ICON_MAP.get("Rubbish"),
+                date=datetime.datetime.strptime(
+                    json[0]["RedBin"], "%Y-%m-%dT%H:%M:%S"
+                ).date(),
+                t="Rubbish",
+                icon=ICON_MAP.get("Rubbish"),
             ),
             Collection(
-                date = datetime.datetime.strptime(json[0]["YellowBin"], "%Y-%m-%dT%H:%M:%S").date(),
-                t = "Recycling",
-                icon = ICON_MAP.get("Recycling"),
-            )
+                date=datetime.datetime.strptime(
+                    json[0]["YellowBin"], "%Y-%m-%dT%H:%M:%S"
+                ).date(),
+                t="Recycling",
+                icon=ICON_MAP.get("Recycling"),
+            ),
         ]
 
-        return entries
+        return entries
diff --git a/doc/source/hcc_govt_nz.md b/doc/source/hcc_govt_nz.md
@@ -16,6 +16,7 @@ waste_collection_schedule:
 
 **address**  
 *(string) (required)*
+excactyl written like autocompleted by <https://www.fightthelandfill.co.nz>
 
 ## Bin Names example - Rubbish, Recycling