edward/agenda
1
0
Fork 0
Code Issues 27 Pull requests Packages Projects Releases Wiki Activity

Group waste schedule events on the same day

Browse source 
Closes: #24
This commit is contained in:
Edward Betts 2023-10-11 16:15:58 +01:00
parent 7b9362fef8
commit 83d3b835f4
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
agenda/waste_schedule.py
View file

@ -1,6 +1,7 @@
#!/usr/bin/python3 #!/usr/bin/python3
import os import os
from collections import defaultdict
from datetime import date, datetime, timedelta from datetime import date, datetime, timedelta
import lxml.html import lxml.html
@ -55,16 +56,19 @@ def parse_waste_schedule_date(day_and_month: str) -> date:
def parse(root: lxml.html.HtmlElement) -> list[Event]: def parse(root: lxml.html.HtmlElement) -> list[Event]:
"""Parse waste schedule.""" """Parse waste schedule."""
events = []
tbody = root.find(".//table/tbody") tbody = root.find(".//table/tbody")
assert tbody is not None assert tbody is not None
by_date = defaultdict(list)
for e_service, e_next_date, e_following in tbody: for e_service, e_next_date, e_following in tbody:
assert e_service.text and e_next_date.text and e_following.text assert e_service.text and e_next_date.text and e_following.text
service = e_service.text service = e_service.text
next_date = parse_waste_schedule_date(e_next_date.text) next_date = parse_waste_schedule_date(e_next_date.text)
following_date = parse_waste_schedule_date(e_following.text) following_date = parse_waste_schedule_date(e_following.text)
events.append(Event(name="waste_schedule", date=next_date, title=service)) by_date[next_date].append(service)
events.append(Event(name="waste_schedule", date=following_date, title=service)) by_date[following_date].append(service)
return events return [
Event(name="waste_schedule", date=d, title=", ".join(services))
for d, services in by_date.items()
]