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 05:06:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1051 from 5ila5/update_north_kesteven_org_uk

Browse Source 
update north_kesteven_org_uk to display more pickup dates
This commit is contained in:
5ila5
2023-06-24 16:10:31 +02:00
committed by GitHub
parent e0bafaabaf 705b61f5d2
commit 2271acf98b
1 changed files with 34 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
custom_components/waste_collection_schedule/waste_collection_schedule/source/north_kesteven_org_uk.py
View File

@@ -3,6 +3,7 @@ import requests
from bs4 import BeautifulSoup
from datetime import datetime
from waste_collection_schedule import Collection # type: ignore[attr-defined]
import re
TITLE = "North Kesteven District Council"
DESCRIPTION = "Source for n-kesteven.org.uk services for North Kesteven District Council, UK."
@@ -32,14 +33,40 @@ class Source:
bins = soup.findAll("div", {"class":"bins-next"})
entries = []
for item in bins:
entries.append(
Collection(
date=datetime.strptime(item.find("strong").text.split(",")[1].strip(), "%d %B %Y").date(),
t=item.find("h3").text,
icon=ICON_MAP.get(item.find("h3").text.upper()),
for bin in bins:
if not bin.has_attr("class"):
continue
bin_name = bin.find("h3").text
icon = ICON_MAP.get(bin_name.upper())
date_div_id = ""
for c in bin["class"]:
if re.match(r"bin-\w+-next", c):
date_div_id = c.replace("bin-", "").replace("-next", "") + "-view"
break
if not date_div_id:
continue
dates_div = soup.find("div", {"id": date_div_id})
if not dates_div:
continue
for date_li in dates_div.find_all("li"):
date = date_li.text
try:
date = datetime.strptime(date.split(",")[1].strip(), "%d %B %Y").date()
except:
print("No date")
continue
entries.append(
Collection(
date=date,
t=bin_name,
icon=icon,
)
)
)
return entries