sascha_hemi/hacs_waste_collection_schedule
1
0
Fork 0
mirror of https://github.com/sascha-hemi/hacs_waste_collection_schedule.git synced 2026-03-21 03:04:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

vivab_se fix bug, where falkenberg address do not work

Browse Source
This commit is contained in:
5ila5
2024-08-06 13:32:46 +02:00
committed by 5ila5
parent 44c320db0b
commit 60f024e79b
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
custom_components/waste_collection_schedule/waste_collection_schedule/source/vivab_se.py
View File

@@ -11,6 +11,7 @@ URL = "https://www.vivab.se"
TEST_CASES = {
"Gallerian": {"street_address": "Västra Vallgatan 2, Varberg"},
"Polisen": {"street_address": "Östra Långgatan 5, Varberg"},
"Storgatan 1, FALKENBERG": {"street_address": "Storgatan 1, FALKENBERG"},
}
@@ -19,7 +20,7 @@ class Source:
self._street_address = street_address
def fetch(self):
search_data = {"searchText": self._street_address}
search_data = {"searchText": self._street_address.split(",")[0].strip()}
response = requests.post(
"https://minasidor.vivab.info/FutureWebVarberg/SimpleWastePickup/SearchAdress",
data=search_data,
@@ -32,14 +33,14 @@ class Source:
and "Buildings" in building_data
and len(building_data["Buildings"]) > 0
):
return []
raise ValueError("No building found")
building_id = None
# support only first building match
building_id_matches = re.findall(r"\(([0-9]+)\)", building_data["Buildings"][0])
if not building_id_matches or len(building_id_matches) == 0:
return []
raise ValueError("No building id found")
building_id = building_id_matches[0]
response = requests.get(