Simplify code for accommodation

This commit is contained in:
Edward Betts 2023-11-05 13:39:40 +00:00
parent 2c9b97f0c2
commit a9c71afb97

View file

@ -251,29 +251,23 @@ def as_date(d: date | datetime) -> date:
return d.date() if isinstance(d, datetime) else d return d.date() if isinstance(d, datetime) else d
def get_accommodation(from_date: date, filepath: str) -> list[Event]: def get_accommodation(filepath: str) -> list[Event]:
"""Get birthdays from config.""" """Get birthdays from config."""
events = []
with open(filepath) as f: with open(filepath) as f:
items = yaml.safe_load(f) return [
Event(
for item in items: date=item["from"],
title = ( end_date=item["to"],
f'{item["location"]} Airbnb' name="accommodation",
if item["operator"] == "airbnb" title=(
else item["name"] f'{item["location"]} Airbnb'
) if item["operator"] == "airbnb"
from_date = item["from"] else item["name"]
to_date = item["to"] ),
e = Event( url=item.get("url"),
date=from_date, )
end_date=to_date, for item in yaml.safe_load(f)
name="accommodation", ]
title=title,
url=item.get("url"),
)
events.append(e)
return events
def waste_collection_events() -> list[Event]: def waste_collection_events() -> list[Event]:
@ -346,7 +340,7 @@ def get_data(now: datetime) -> typing.Mapping[str, str | object]:
my_data = config["data"]["personal-data"] my_data = config["data"]["personal-data"]
events += birthday.get_birthdays(last_year, os.path.join(my_data, "entities.yaml")) events += birthday.get_birthdays(last_year, os.path.join(my_data, "entities.yaml"))
events += get_accommodation(today, os.path.join(my_data, "accommodation.yaml")) events += get_accommodation(os.path.join(my_data, "accommodation.yaml"))
events += travel.all_events(today, config["data"]["personal-data"]) events += travel.all_events(today, config["data"]["personal-data"])
events += conference.get_list(os.path.join(my_data, "conferences.yaml")) events += conference.get_list(os.path.join(my_data, "conferences.yaml"))
events += waste_collection_events() + bristol_waste_collection_events(today) events += waste_collection_events() + bristol_waste_collection_events(today)