@camstack/addon-waste-collections 0.0.0-stage → 0.1.2

package/python/vendor/waste_collection_schedule/source/yarracity_vic_gov_au.py

@@ -0,0 +1,163 @@
+from datetime import date, timedelta
+
+from dateutil.easter import easter
+from waste_collection_schedule import Collection, Icons
+from waste_collection_schedule.exceptions import SourceArgumentNotFound
+from waste_collection_schedule.service.ArcGis import (
+    ArcGisError,
+    geocode,
+    get_next_n_dates,
+    most_recent_weekday,
+    query_feature_layer,
+)
+
+TITLE = "City of Yarra"
+DESCRIPTION = "Source for City of Yarra waste collection."
+URL = "https://www.yarracity.vic.gov.au"
+COUNTRY = "au"
+TEST_CASES = {
+    "Fitzroy Town Hall": {"address": "201 Napier Street, Fitzroy VIC 3065"},
+    "Richmond Town Hall": {"address": "333 Bridge Road, Richmond VIC 3121"},
+}
+SOURCE_CODEOWNERS = ["@yeaaaaaahh"]
+
+HOW_TO_GET_ARGUMENTS_DESCRIPTION = {
+    "en": (
+        "Enter your full street address including suburb and postcode, "
+        "e.g. '333 Bridge Road, Richmond VIC 3121'."
+    ),
+}
+
+PARAM_DESCRIPTIONS = {
+    "en": {
+        "address": (
+            "Full street address within the City of Yarra, "
+            "including suburb and postcode"
+        ),
+    },
+}
+
+PARAM_TRANSLATIONS = {
+    "en": {
+        "address": "Street Address",
+    },
+}
+
+ICON_MAP = {
+    "Rubbish": Icons.GENERAL_WASTE,
+    "Recycling": Icons.RECYCLING,
+    "Glass": Icons.GLASS,
+    "FOGO": Icons.BIO_KITCHEN,
+}
+
+MAP_SERVER_URL = (
+    "https://yccgis-prd.esriaustraliaonline.com.au/arcgis/rest/services/"
+    "Waste_Services/CW_FC_PRD_Waste_Collection/MapServer"
+)
+GLASS_LAYER_URL = f"{MAP_SERVER_URL}/0"
+WASTE_LAYER_URL = f"{MAP_SERVER_URL}/1"
+
+WEEKS_AHEAD = 52
+
+_WEEKDAYS = {
+    "Monday": 0,
+    "Tuesday": 1,
+    "Wednesday": 2,
+    "Thursday": 3,
+    "Friday": 4,
+    "Saturday": 5,
+    "Sunday": 6,
+}
+
+
+def _attr(attributes: dict, name: str):
+    # Joined ArcGIS layers qualify attribute names ("<join>.<field>");
+    # match a bare field name against either scheme. An exact match wins so
+    # the result never depends on attribute ordering. The required "."
+    # separator keeps e.g. "recycling_anchor_date" from matching "anchor_date".
+    if name in attributes:
+        return attributes[name]
+    suffix = "." + name
+    for key, value in attributes.items():
+        if key.endswith(suffix):
+            return value
+    raise KeyError(name)
+
+
+def _holiday_shift(d: date) -> date:
+    # Council rule: collections run on all public holidays except Good Friday
+    # and Christmas Day, when the collection happens one day later.
+    if d == easter(d.year) - timedelta(days=2) or d == date(d.year, 12, 25):
+        return d + timedelta(days=1)
+    return d
+
+
+class Source:
+    def __init__(self, address: str):
+        self._address = address.strip()
+
+    def fetch(self) -> list[Collection]:
+        try:
+            location = geocode(f"{self._address}, Victoria, Australia")
+        except ArcGisError as e:
+            raise SourceArgumentNotFound("address", self._address) from e
+
+        try:
+            # 2026-08: the council rebuilt the glass layer as a joined layer
+            # whose attributes carry fully-qualified names (e.g.
+            # "internal.gdo.Glass_Collection_Zones.anchor_date"), and a
+            # bare-name out_fields now fails with HTTP 400. Fetch all fields
+            # and resolve names via _attr so either scheme works on either
+            # layer.
+            waste = query_feature_layer(
+                WASTE_LAYER_URL,
+                geometry=location,
+                out_fields="*",
+            )[0]
+            glass = query_feature_layer(
+                GLASS_LAYER_URL,
+                geometry=location,
+                out_fields="*",
+            )[0]
+        except ArcGisError as e:
+            raise SourceArgumentNotFound(
+                "address",
+                self._address,
+                "the address must be within the City of Yarra.",
+            ) from e
+
+        day_name = _attr(waste, "collection_day")
+        if day_name not in _WEEKDAYS:
+            raise ValueError(f"Unexpected collection day: {day_name!r}")
+
+        entries = []
+
+        # Rubbish and FOGO are collected weekly on the property's collection day.
+        for d in get_next_n_dates(
+            most_recent_weekday(_WEEKDAYS[day_name]), WEEKS_AHEAD, timedelta(days=7)
+        ):
+            d = _holiday_shift(d)
+            entries.append(Collection(d, "Rubbish", ICON_MAP["Rubbish"]))
+            entries.append(Collection(d, "FOGO", ICON_MAP["FOGO"]))
+
+        # Recycling alternates fortnightly between zones; the layer's anchor
+        # date fixes this property's phase. DateOnly fields are ISO strings.
+        for d in get_next_n_dates(
+            date.fromisoformat(_attr(waste, "recycling_anchor_date")),
+            WEEKS_AHEAD // 2,
+            timedelta(days=14),
+        ):
+            entries.append(
+                Collection(_holiday_shift(d), "Recycling", ICON_MAP["Recycling"])
+            )
+
+        # Glass runs on its own cycle (currently every 28 days) with an
+        # independent per-polygon anchor.
+        for d in get_next_n_dates(
+            date.fromisoformat(_attr(glass, "anchor_date")),
+            WEEKS_AHEAD // 4,
+            timedelta(days=int(_attr(glass, "frequency_days"))),
+        ):
+            entries.append(Collection(_holiday_shift(d), "Glass", ICON_MAP["Glass"]))
+
+        return entries

package/python/vendor/waste_collection_schedule/source/york_gov_uk.py

@@ -0,0 +1,51 @@
+import json
+from datetime import datetime
+
+import requests
+from waste_collection_schedule import Collection, Icons  # type: ignore[attr-defined]
+
+TITLE = "City of York Council"
+DESCRIPTION = "Source for York.gov.uk services for the city of York, UK."
+URL = "https://york.gov.uk"
+TEST_CASES = {
+    "Reighton Avenue, York": {"uprn": "100050580641"},
+    "Granary Walk, York": {"uprn": "010093236548"},
+}
+
+ICON_MAP = {
+    "REFUSE": Icons.GENERAL_WASTE,
+    "RECYCLING": Icons.RECYCLING,
+    "GARDEN": Icons.GARDEN,
+}
+
+
+class Source:
+    def __init__(self, uprn):
+        self._uprn = uprn
+
+    def fetch(self):
+        # get json file
+        r = requests.get(
+            f"https://waste-api.york.gov.uk/api/Collections/GetBinCalendarDataForUprn/{self._uprn}"
+        )
+
+        # extract data from json
+        data = json.loads(r.text)
+
+        entries = []
+
+        for collection in data["collections"]:
+            try:
+                entries.append(
+                    Collection(
+                        date=datetime.strptime(
+                            collection["date"], "%Y-%m-%dT%H:%M:%S"
+                        ).date(),
+                        t=collection["roundType"].title(),
+                        icon=ICON_MAP.get(collection["roundType"]),
+                    )
+                )
+            except ValueError:
+                pass  # ignore date conversion failure for not scheduled collections
+
+        return entries

package/python/vendor/waste_collection_schedule/source/zab_ch.py

@@ -0,0 +1,121 @@
+from waste_collection_schedule import Collection, Icons
+from waste_collection_schedule.exceptions import SourceArgumentNotFoundWithSuggestions
+from waste_collection_schedule.service.A_region_ch import A_region_ch
+
+TITLE = "ZAB Bazenheid"
+DESCRIPTION = "Source for Zweckverband Abfallverwertung Bazenheid (ZAB)"
+URL = "https://zab.citymobile.ch"
+
+
+def EXTRA_INFO():
+    return [{"title": m, "default_params": {"municipality": m}} for m in MUNICIPALITIES]
+
+
+TEST_CASES = {
+    "Wängi": {"municipality": "Wängi"},
+    "Sirnach": {"municipality": "Sirnach", "district": "Papier und Karton Sirnach"},
+    "Eschlikon": {
+        "municipality": "Eschlikon",
+        "district": "Kehrichtsammlung Eschlikon",
+    },
+}
+
+ICON_MAP = {
+    "Kehricht": Icons.GENERAL_WASTE,
+    "Grünabfuhr": Icons.ORGANIC,
+    "Grüngut": Icons.ORGANIC,
+    "Papier": Icons.PAPER,
+    "Papier/Karton": Icons.PAPER,
+    "Karton": Icons.PAPER,
+    "Metall": Icons.METAL,
+    "Altmetall": Icons.METAL,
+    "Schredderdienst": Icons.GARDEN,
+    "Häckseldienst": Icons.GARDEN,
+    "Christbaum": Icons.CHRISTMAS_TREE,
+    "Christbäume": Icons.CHRISTMAS_TREE,
+    "Sperrgut": Icons.BULKY,
+}
+
+MUNICIPALITIES = {
+    "Aadorf": "/index.php?apid=9726871&apparentid=2326631",
+    "Bettwiesen": "/index.php?apid=12085057&apparentid=2326631",
+    "Bichelsee-Balterswil": "/index.php?apid=11535568&apparentid=2326631",
+    "Braunau": "/index.php?apid=12198875&apparentid=2326631",
+    "Bütschwil-Ganterschwil": "/index.php?apid=8392002&apparentid=2326631",
+    "Degersheim": "/index.php?apid=16588818&apparentid=2326631",
+    "Ebnat-Kappel": "/index.php?apid=6034813&apparentid=2326631",
+    "Eschlikon": "/index.php?apid=4813431&apparentid=2326631",
+    "Fischingen": "/index.php?apid=9382046&apparentid=2326631",
+    "Flawil": "/index.php?apid=8527584&apparentid=2326631",
+    "Gossau": "/index.php?apid=3414158&apparentid=2326631",
+    "Jonschwil": "/index.php?apid=5841810&apparentid=2326631",
+    "Kirchberg": "/index.php?apid=13015555&apparentid=2326631",
+    "Lichtensteig": "/index.php?apid=6846579&apparentid=2326631",
+    "Lütisburg": "/index.php?apid=13024565&apparentid=2326631",
+    "Mosnang": "/index.php?apid=782123&apparentid=2326631",
+    "Münchwilen": "/index.php?apid=312606&apparentid=2326631",
+    "Neckertal": "/index.php?apid=2949770&apparentid=2326631",
+    "Nesslau": "/index.php?apid=15203333&apparentid=2326631",
+    "Niederbüren": "/index.php?apid=9191724&apparentid=2326631",
+    "Niederhelfenschwil": "/index.php?apid=10484022&apparentid=2326631",
+    "Oberbüren": "/index.php?apid=3972728&apparentid=2326631",
+    "Oberuzwil": "/index.php?apid=4440141&apparentid=2326631",
+    "Rickenbach": "/index.php?apid=5878077&apparentid=2326631",
+    "Sirnach": "/index.php?apid=10773272&apparentid=2326631",
+    "Tobel-Tägerschen": "/index.php?apid=13036162&apparentid=2326631",
+    "Uzwil": "/index.php?apid=6575733&apparentid=2326631",
+    "Wängi": "/index.php?apid=1811155&apparentid=2326631",
+    "Wattwil": "/index.php?apid=6490334&apparentid=2326631",
+    "Wil": "/index.php?apid=9385302&apparentid=2326631",
+    "Wilen": "/index.php?apid=9820263&apparentid=2326631",
+    "Wuppenau": "/index.php?apid=9826075&apparentid=2326631",
+    "Zuzwil": "/index.php?apid=215802&apparentid=2326631",
+}
+
+
+class Source:
+    def __init__(self, municipality, district=None):
+        self._municipality = municipality
+        self._district = district
+        if municipality not in MUNICIPALITIES:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "municipality", municipality, MUNICIPALITIES.keys()
+            )
+        self._municipality_url = MUNICIPALITIES[municipality]
+
+        self._ics_sources = []
+
+    def _get_ics_sources(self):
+        self._ics_sources = A_region_ch(
+            "zab", self._municipality_url, self._district
+        ).fetch()
+
+    def fetch(self) -> list[Collection]:
+        fresh_sources = False
+        if not self._ics_sources:
+            fresh_sources = True
+            self._get_ics_sources()
+
+        entries = []
+        for source in self._ics_sources:
+            fresh_sources, e = self._get_dates(source, fresh_sources)
+            entries += e
+        return entries
+
+    def _get_dates(self, source, fresh=False) -> tuple[bool, list[Collection]]:
+        exception = None
+        try:
+            entries = source.fetch()
+        except Exception as e:
+            exception = e
+
+        if exception or not entries:
+            if fresh:
+                if exception:
+                    raise exception
+                return fresh, []
+
+            self._get_ics_sources()
+            return self._get_dates(source, fresh=True)
+
+        return fresh, entries

package/python/vendor/waste_collection_schedule/source/zakb_de.py

@@ -0,0 +1,161 @@
+import re
+from difflib import get_close_matches
+
+import requests
+from waste_collection_schedule import Collection, Icons  # type: ignore[attr-defined]
+from waste_collection_schedule.exceptions import SourceArgumentNotFoundWithSuggestions
+from waste_collection_schedule.service.ICS import ICS
+
+TITLE = "Zweckverband Abfallwirtschaft Kreis Bergstraße"
+DESCRIPTION = "Source for Zweckverband Abfallwirtschaft Kreis Bergstraße."
+URL = "https://www.zakb.de"
+TEST_CASES = {
+    "Abtsteinach, Am Hofböhl 1 ": {
+        "ort": "Abtsteinach",
+        "strasse": "Am Hofböhl",
+        "hnr": "1",
+        "hnr_zusatz": "",
+    },
+    "Gorxheimertal, Am Herrschaftswald 10": {
+        "ort": "Gorxheimertal",
+        "strasse": "Am Herrschaftswald",
+        "hnr": "10",
+    },
+    "Rimbach, Ahornweg 1 B": {
+        "ort": "Rimbach",
+        "strasse": "Ahornweg",
+        "hnr": "1",
+        "hnr_zusatz": "B",
+    },
+    "Zwingenberg, Diefenbachstraße 57": {
+        "ort": "Zwingenberg",
+        "strasse": "Diefenbachstraße",
+        "hnr": 57,
+        "hnr_zusatz": "",
+    },
+    "Bensheim im Bangert 9 a": {
+        "ort": "Bensheim",
+        "strasse": "Im Bangert",
+        "hnr": 9,
+        "hnr_zusatz": "A",
+    },
+    "Zwingenberg, Wiesenpromenade-West 41": {
+        "ort": "Zwingenberg",
+        "strasse": "Wiesenpromenade-West",
+        "hnr": 41,
+    },
+}
+
+
+ICON_MAP = {
+    "Restabfallbehaelter": Icons.GENERAL_WASTE,
+    "Restabfallcontainer": Icons.GENERAL_WASTE,
+    "Bioabfallbehaelter": Icons.BIO_KITCHEN,
+    "Papierbehaelter": Icons.PAPER,
+    "Papiercontainer": Icons.PAPER,
+    "Gelber Sack": Icons.PLASTIC_PACKAGING,
+    "Gruensperrmuell": Icons.GARDEN,
+}
+
+
+API_URL = "https://www.zakb.de/online-service/abfallkalender/"
+
+
+PARAM_TRANSLATIONS = {
+    "de": {
+        "ort": "Ort",
+        "strasse": "Straße",
+        "hnr": "Hausnummer",
+        "hnr_zusatz": "Hausnummerzusatz",
+    },
+    "en": {
+        "ort": "City",
+        "strasse": "Street",
+        "hnr": "House number",
+        "hnr_zusatz": "House number addition",
+    },
+}
+
+
+def _extract_select_options(html: str, select_id: str) -> list[str]:
+    match = re.search(rf'<select id="{select_id}".*?</select>', html, re.S)
+    if not match:
+        return []
+    return re.findall(r'value="([^"]*)"', match.group(0))
+
+
+class Source:
+    def __init__(self, ort: str, strasse: str, hnr: str | int, hnr_zusatz: str = ""):
+        self._ort: str = ort
+        self._strasse: str = strasse
+        self._hnr: str = str(hnr)
+        self._hnr_zusatz: str = hnr_zusatz
+        self._ics = ICS()
+
+    def fetch(self):
+        # inizilize session
+        session = requests.Session()
+
+        # make request to get session cookie and available cities
+        r = session.get(API_URL)
+        r.raise_for_status()
+
+        ort_options = _extract_select_options(r.text, "Ort")
+        if ort_options and self._ort not in ort_options:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "ort",
+                self._ort,
+                get_close_matches(self._ort, ort_options, n=5, cutoff=0.4)
+                or ort_options,
+            )
+
+        args = {
+            "aos[Ort]": self._ort,
+            "aos[CheckBoxRestabfallbehaelter]": "on",
+            "aos[CheckBoxRestabfallcontainer]": "on",
+            "aos[CheckBoxBioabfallbehaelter]": "on",
+            "aos[CheckBoxPapierbehaelter]": "on",
+            "aos[CheckBoxPapiercontainer]": "on",
+            "aos[CheckBoxGruensperrmuell]": "on",
+            "aos[CheckBoxGelber+Sack]": "on",
+            "aos[CheckBoxDSD-Container]": "on",
+            "submitAction": "CITYCHANGED",
+            "pageName": "Lageadresse",
+        }
+
+        # make request to call CITYCHANGED and get the streets available for this city.
+        # The site uses its own street-name spelling (e.g. hyphens instead of spaces),
+        # so validate against it and offer suggestions instead of failing deep in ICS parsing.
+        r = session.post(API_URL, data=args)
+        r.raise_for_status()
+
+        strasse_options = _extract_select_options(r.text, "Strasse")
+        if strasse_options and self._strasse not in strasse_options:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "strasse",
+                self._strasse,
+                get_close_matches(self._strasse, strasse_options, n=5, cutoff=0.4)
+                or strasse_options,
+            )
+
+        args["aos[Strasse]"] = self._strasse
+        args["aos[Hausnummer]"] = self._hnr
+        args["aos[Hausnummerzusatz]"] = self._hnr_zusatz
+
+        # make request to set data for session
+        args["submitAction"] = "nextPage"
+        r = session.post(API_URL, data=args)
+        r.raise_for_status()
+
+        # make request to get ical file
+        r = session.post(
+            API_URL,
+            data={"submitAction": "filedownload_ICAL", "pageName": "Terminliste"},
+        )
+        r.raise_for_status()
+
+        dates = self._ics.convert(r.text)
+        entries = []
+        for d in dates:
+            entries.append(Collection(d[0], d[1].strip(), ICON_MAP.get(d[1].strip())))
+        return entries