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

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

@@ -0,0 +1,122 @@
+import datetime
+import json
+import re
+from urllib.parse import urlencode
+
+import requests
+from waste_collection_schedule import Collection, Icons  # type: ignore[attr-defined]
+
+TITLE = "Min Renovasjon"
+DESCRIPTION = "Source for Norkart Komtek MinRenovasjon (Norway)."
+URL = "https://www.norkart.no"
+
+# **street_code:** \
+# **county_id:** \
+# Can be found with this REST-API call.
+# ```
+# https://ws.geonorge.no/adresser/v1/#/default/get_sok
+# https://ws.geonorge.no/adresser/v1/sok?sok=Min%20Gate%2012
+# ```
+# "street_code" equals to "adressekode" and "county_id" equals to "kommunenummer".
+
+TEST_CASES = {
+    "Sandvika Rådhus": {
+        "street_name": "Rådhustorget",
+        "house_number": 2,
+        "street_code": 2469,
+        "county_id": 3024,
+    },
+    "Lillehammer Åsmarkvegen 111": {
+        "street_name": "Åsmarkvegen",
+        "house_number": 111,
+        "street_code": 6530,
+        "county_id": 3405,
+    },
+}
+
+EXTRA_INFO = [
+    {
+        "title": "ROAF (Romerike Avfallsforedling IKS)",
+        "url": "https://www.roaf.no",
+        "country": "no",
+        "default_params": {
+            "street_name": "Solvangen",
+            "house_number": 2,
+            "street_code": 12950,
+            "county_id": 3205,
+        },
+    },
+]
+
+API_URL = (
+    "https://norkartrenovasjon.azurewebsites.net/proxyserver.ashx?server="
+    "https://komteksky.norkart.no/MinRenovasjon.Api/api/"
+)
+APP_KEY = "AE13DEEC-804F-4615-A74E-B4FAC11F0A30"
+ICON_MAP = {
+    "": Icons.GENERAL_WASTE,
+    "brush": Icons.GENERAL_WASTE,
+    "elektriskogelektronisk": Icons.ELECTRONICS,
+    "farligavfall": Icons.GENERAL_WASTE,
+    "glassogmetallemballasje": Icons.GENERAL_WASTE,
+    "hageavfall": Icons.ORGANIC,
+    "klaerogsko": Icons.TEXTILE,
+    "matavfall": Icons.BIO_KITCHEN,
+    "matrestavfall": Icons.GENERAL_WASTE,
+    "matrestavfallplast": Icons.GENERAL_WASTE,
+    "metall": Icons.METAL,
+    "papir": Icons.PAPER,
+    "pappogkartong": Icons.PAPER,
+    "plastemballasje": Icons.GENERAL_WASTE,
+    "restavfall": Icons.GENERAL_WASTE,
+    "drikkekartong": Icons.NEWSPAPER,
+    "papppapirdrikkekartong": Icons.NEWSPAPER,
+    "trevirke": Icons.GENERAL_WASTE,
+}
+
+
+class Source:
+    def __init__(self, street_name, house_number, street_code, county_id):
+        self._street_name = street_name
+        self._house_number = house_number
+        self._street_code = street_code
+        self._county_id = county_id
+
+    def fetch(self):
+        headers = {
+            "Kommunenr": str(self._county_id),
+            "RenovasjonAppKey": APP_KEY,
+            "user-agent": "Home-Assitant-waste-col-sched/0.1",
+        }
+        args = {}
+
+        r = requests.get(API_URL + "fraksjoner", params=args, headers=headers)
+
+        type = {}
+        for f in json.loads(r.content):
+            # pprint(f)
+            icon_name = re.sub(r"^.*?/(\w+)\.\w{3,4}$", "\\1", f["Ikon"])
+            icon = ICON_MAP.get(icon_name)
+            type[f["Id"]] = {"name": f["Navn"], "image": f["Ikon"], "icon": icon}
+
+        args = {
+            "gatenavn": self._street_name,
+            "husnr": self._house_number,
+            "gatekode": self._street_code,
+        }
+
+        url_with_param = API_URL + "tommekalender?" + urlencode(args)
+        r = requests.get(url_with_param, headers=headers)
+        entries = []
+        for f in json.loads(r.content):
+            for d in f["Tommedatoer"]:
+                entries.append(
+                    Collection(
+                        date=datetime.datetime.strptime(d, "%Y-%m-%dT%H:%M:%S").date(),
+                        t=type[f["FraksjonId"]]["name"],
+                        icon=type[f["FraksjonId"]]["icon"],
+                        picture=type[f["FraksjonId"]]["image"],
+                    )
+                )
+
+        return entries

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

@@ -0,0 +1,152 @@
+from datetime import datetime
+from typing import Literal, TypedDict
+
+import requests
+from waste_collection_schedule import Collection, Icons
+from waste_collection_schedule.exceptions import SourceArgumentNotFoundWithSuggestions
+
+
+class Zone(TypedDict):
+    id: int
+
+
+TITLE = "Mirabel (QC)"
+DESCRIPTION = "Source script for mirabel.ca/collectes"
+URL = "https://mirabel.ca/collectes"
+COUNTRY = "ca"
+
+TEST_CASES = {
+    "Mirabel-en-Haut": {"zone": 1},
+    "Saint-Antoine": {"zone": 2},
+    "Sainte-Monique": {"zone": 3},
+    "Domaine-Vert Nord": {"zone": 4},
+    "Saint-Janvier": {"zone": 5},
+    "Saint-Augustin": {"zone": 6},
+    "Saint-Canut": {"zone": 7},
+    "St-Benoit": {"zone": 8},
+}
+
+
+EVENT_QUERY = """
+query eventTypes($month: Int, $year: Int, $zoneId: Int) {
+    events(year: $year, month: $month, zoneId: $zoneId) {
+        nodes {
+            date
+            id
+            type {
+                color
+                id
+                text: name
+                slug
+                __typename
+            }
+            zone {
+                id
+                name
+                slug
+                __typename
+            }
+            __typename
+        }
+    __typename
+    }
+}
+"""
+
+API_URL = "https://mviv2.mirabel.ca/graphql"
+
+
+ICON_MAP = {
+    "dechets": Icons.GENERAL_WASTE,
+    "recyclage": Icons.RECYCLING,
+    "composte": Icons.ORGANIC,  # codespell:ignore composte
+    "encombrants": Icons.BULKY,
+}
+
+
+HOW_TO_GET_ARGUMENTS_DESCRIPTION = {
+    "en": "You can find your collection zone number using the webpage: https://mirabel.ca/services/services-en-ligne/trouver-ma-zone-de-collecte",
+    "fr": "Vous pouvez trouver votre numéro de zone de collecte sur l'adresse suivante : https://mirabel.ca/services/services-en-ligne/trouver-ma-zone-de-collecte",
+}
+
+PARAM_DESCRIPTIONS = {
+    "en": {
+        "zone": "Collection zone number",
+    },
+    "fr": {
+        "zone": "Numéro de la zone de collecte",
+    },
+}
+
+PARAM_TRANSLATIONS = {
+    "en": {
+        "zone": "Collection zone number",
+    },
+    "fr": {
+        "zone": "Zone de collecte",
+    },
+}
+
+# The id differ a bit from the zone official zone number.
+_ZONE_INFO = {
+    "1": 121,
+    "2": 122,
+    "3": 123,
+    "4": 124,
+    "5": 125,
+    "6": 126,
+    "7": 127,
+    "8": 128,
+}
+
+ZONE_LITERALS = Literal[tuple(_ZONE_INFO.keys())]
+
+ZONES: dict[str, Zone] = {name: {"id": zid} for name, zid in _ZONE_INFO.items()}
+
+
+class Source:
+    def __init__(self, zone: ZONE_LITERALS):  # type: ignore
+        zone = str(zone)
+        if zone not in ZONES:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "zone", zone, suggestions=ZONES.keys()
+            )
+        self._id = ZONES[zone]["id"]
+
+    def fetch(self) -> list[Collection]:
+
+        entries = []
+
+        with requests.Session() as session:
+            session.headers.update({"Content-Type": "application/json"})
+            r = session.post(
+                API_URL,
+                json={
+                    "query": EVENT_QUERY,
+                    "variables": {"zoneId": self._id},
+                },
+                timeout=15,
+            )
+            try:
+                r.raise_for_status()
+            except requests.HTTPError as e:
+                raise RuntimeError(
+                    f"Failed to fetch collections for zone '{self._id}': {e}"
+                ) from e
+
+            payload = r.json()
+            graphql_errors = payload.get("errors")
+            if graphql_errors:
+                raise RuntimeError(f"GraphQL errors: {graphql_errors}")
+            events = payload.get("data", {}).get("events", {}).get("nodes") or []
+            for event in events:
+                event_type = event.get("type")
+                entries.append(
+                    Collection(
+                        date=datetime.fromisoformat(event.get("date")).date(),
+                        t=event_type.get("text"),
+                        icon=ICON_MAP.get(event_type.get("slug")),
+                    )
+                )
+
+        return entries

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

@@ -0,0 +1,70 @@
+import datetime
+import logging
+
+import requests
+from waste_collection_schedule import Collection, Icons  # type: ignore[attr-defined]
+
+TITLE = "Mitchell Shire Council"
+URL = "https://www.mitchellshire.vic.gov.au"
+DESCRIPTION = "Source for Mitchell Shire Council, Victoria, Australia."
+
+# Coordinates can be found by right-clicking your address in Google Maps and
+# selecting "What's here?" — copy the lat/lon values shown.
+TEST_CASES = {
+    "Wallan": {
+        "lat": -37.4195459,
+        "lon": 144.9592853,
+    },
+    "Beveridge": {
+        "lat": -37.48012709087705,
+        "lon": 144.94518706976186,
+    },
+    "McDonalds Wallan": {"lat": -37.41290975665613, "lon": 144.97998167557827},
+}
+
+_LOGGER = logging.getLogger(__name__)
+
+API_URL = "https://www.mitchellshire.vic.gov.au/simple-gov-app/api/resources/bin-collections/search"
+
+# Map bin title to an MDI icon. The API also provides a `color` field
+# (red/yellow/green/purple) which matches the physical bin lid colours.
+ICON_MAP = {
+    "General Rubbish": Icons.GENERAL_WASTE,
+    "Mixed Recycling": Icons.RECYCLING,
+    "Food and Garden Organics": Icons.BIO_KITCHEN,
+    "Glass Recycling": Icons.GLASS,
+}
+
+
+class Source:
+    def __init__(self, lat: float, lon: float):
+        self._lat = lat
+        self._lon = lon
+
+    def fetch(self) -> list[Collection]:
+        r = requests.get(
+            API_URL,
+            params={"lat": self._lat, "lng": self._lon},
+            timeout=30,
+        )
+        r.raise_for_status()
+        data = r.json()
+
+        if data.get("result") != "success":
+            raise ValueError(f"Unexpected API result: {data.get('result')}")
+
+        entries = []
+        for bin_type in data.get("data", []):
+            title = bin_type.get("title", "Unknown")
+            icon = ICON_MAP.get(title)
+
+            for item in bin_type.get("collectionDates", []):
+                try:
+                    date = datetime.datetime.strptime(item["date"], "%Y-%m-%d").date()
+                except (ValueError, KeyError) as e:
+                    _LOGGER.warning("Could not parse date %s: %s", item, e)
+                    continue
+
+                entries.append(Collection(date=date, t=title, icon=icon))
+
+        return entries

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

@@ -0,0 +1,194 @@
+import datetime
+import json
+
+import requests
+import urllib3
+from bs4 import BeautifulSoup
+from waste_collection_schedule import Collection, Icons
+from waste_collection_schedule.exceptions import (
+    SourceArgumentNotFound,
+    SourceArgumentNotFoundWithSuggestions,
+)
+
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+TITLE = "MOHU Budapest"
+DESCRIPTION = "Source script for www.mohubudapest.hu"
+URL = "https://www.mohubudapest.hu"
+COUNTRY = "hu"
+TEST_CASES = {
+    "Test_1": {"district": 1011, "street": "Apród utca", "house_number": 10},
+    "Test_2": {"district": 1114, "street": "Szent Gellért tér", "house_number": 3},
+    "Test_3": {"district": 1202, "street": "Bölön utca", "house_number": 5},
+    "Test_4": {
+        "district": 1011,
+        "street": "Apród utca",
+        "house_number": 10,
+        "verify": False,
+    },
+    "Test_5": {
+        "district": 1114,
+        "street": "Szent Gellért tér",
+        "house_number": 3,
+        "verify": False,
+    },
+    "Test_6": {
+        "district": 1202,
+        "street": "Bölön utca",
+        "house_number": 5,
+        "verify": False,
+    },
+}
+
+API_URL = "https://mohubudapest.hu/hulladeknaptar"
+ICON_MAP = {
+    "COMMUNAL": Icons.GENERAL_WASTE,
+    "SELECTIVE": Icons.RECYCLING,
+    "GREEN": Icons.ORGANIC,
+}
+
+
+def _get_partial_html(data: dict, *keys: str) -> str:
+    """Return the first matching OctoberCMS partial HTML payload."""
+    for key in keys:
+        value = data.get(key)
+        if value is not None:
+            return value
+    raise KeyError(keys[0])
+
+
+class Source:
+    def __init__(self, district, street, house_number, verify=True):
+        self._district = district
+        self._street = street[::-1].replace(" ", "---", 1)[::-1]
+        self._house_number = house_number
+        self._verify = verify
+
+    def fetch(self):
+        session = requests.Session()
+
+        r = session.post(
+            API_URL,
+            data={"district": self._district},
+            headers={
+                "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+                "X-October-Request-Handler": "onSelectDistricts",
+                "X-October-Request-Partials": "ajax/publicPlaces",
+                "X-Requested-With": "XMLHttpRequest",
+            },
+            verify=self._verify,
+        )
+        r.raise_for_status()
+        data = json.loads(r.text)
+        soup = BeautifulSoup(
+            _get_partial_html(data, ".publicPlaces", "ajax/publicPlaces"),
+            features="html.parser",
+        )
+
+        if soup.find("div", attrs={"class": "alert"}) is not None:
+            raise SourceArgumentNotFound("district", self._district)
+
+        available_streets = []
+        for opt in soup.find_all("option")[1:]:
+            available_streets.append(opt.decode_contents())
+
+        r = session.post(
+            API_URL,
+            data={"publicPlace": self._street},
+            headers={
+                "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+                "X-October-Request-Handler": "onSavePublicPlace",
+                "X-October-Request-Partials": "ajax/houseNumbers",
+                "X-Requested-With": "XMLHttpRequest",
+            },
+            verify=self._verify,
+        )
+        try:
+            r.raise_for_status()
+        except requests.exceptions.HTTPError as e:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "street", self._street, available_streets
+            ) from e
+        data = json.loads(r.text)
+        soup = BeautifulSoup(
+            _get_partial_html(data, ".houseNumbers", "ajax/houseNumbers"),
+            features="html.parser",
+        )
+
+        if (
+            soup.find("div", attrs={"class": "alert"}) is not None
+            or soup.find("option", string="Kérem, előbb válasszon közterületet")
+            is not None
+        ):
+            raise SourceArgumentNotFoundWithSuggestions(
+                "street", self._street, available_streets
+            )
+
+        available_numbers = []
+        for opt in soup.find_all("option")[1:]:
+            available_numbers.append(opt.decode_contents())
+
+        r = session.post(
+            API_URL,
+            data={"houseNumber": self._house_number},
+            headers={
+                "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+                "X-October-Request-Handler": "onSearch",
+                "X-October-Request-Partials": "ajax/calSearchResults",
+                "X-Requested-With": "XMLHttpRequest",
+            },
+            verify=self._verify,
+        )
+        r.raise_for_status()
+        data = json.loads(r.text)
+        soup = BeautifulSoup(
+            _get_partial_html(data, ".results", "ajax/calSearchResults"),
+            features="html.parser",
+        )
+
+        if soup.find("div", attrs={"class": "alert"}) is not None:
+            raise SourceArgumentNotFoundWithSuggestions(
+                "house_number", self._house_number, available_numbers
+            )
+
+        entries = []
+        communal_divs = soup.find_all("div", attrs={"class": "communal"})
+
+        selective = soup.find_all("div", attrs={"class": "selective"})
+        communal = list(filter(lambda div: div.text == "Kommunális", communal_divs))
+        green = list(filter(lambda div: div.text == "Zöld", communal_divs))
+
+        for element in communal:
+            entries.append(
+                Collection(
+                    date=datetime.datetime.strptime(
+                        element.parent.parent.findAll("td")[1].text, "%Y.%m.%d"
+                    ).date(),
+                    t="Communal",
+                    icon=ICON_MAP.get("COMMUNAL"),
+                )
+            )
+
+        for element in selective:
+            entries.append(
+                Collection(
+                    date=datetime.datetime.strptime(
+                        element.parent.parent.findAll("td")[1].text, "%Y.%m.%d"
+                    ).date(),
+                    t="Selective",
+                    icon=ICON_MAP.get("SELECTIVE"),
+                )
+            )
+
+        for element in green:
+            entries.append(
+                Collection(
+                    date=datetime.datetime.strptime(
+                        element.parent.parent.findAll("td")[1].text, "%Y.%m.%d"
+                    ).date(),
+                    t="Green",
+                    icon=ICON_MAP.get("GREEN"),
+                )
+            )
+
+        return entries