diff --git a/custom_components/waste_collection_schedule/waste_collection_schedule/source/kuringgai_nsw_gov_au.py b/custom_components/waste_collection_schedule/waste_collection_schedule/source/kuringgai_nsw_gov_au.py index b73b47433..606e7664c 100644 --- a/custom_components/waste_collection_schedule/waste_collection_schedule/source/kuringgai_nsw_gov_au.py +++ b/custom_components/waste_collection_schedule/waste_collection_schedule/source/kuringgai_nsw_gov_au.py @@ -55,19 +55,11 @@ } ICON_MAP = { - "GeneralWaste": "mdi:trash-can", + "General Waste": "mdi:trash-can", "Recycling": "mdi:recycle", - "GreenWaste": "mdi:leaf", + "Green Waste": "mdi:leaf", } -ROUNDS = { - "GeneralWaste": "General Waste", - "Recycling": "Recycling", - "GreenWaste": "Green Waste", -} - -# _LOGGER = logging.getLogger(__name__) - class Source: def __init__( @@ -133,10 +125,7 @@ def fetch(self): entries.append( Collection( date=date, - # t=waste_type, # api returns GeneralWaste, Recycling, GreenWaste - t=ROUNDS.get( - waste_type - ), # returns user-friendly General Waste, Recycling, Green Waste + t=waste_type, icon=ICON_MAP.get(waste_type), ) )