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

Split market hiding into function with config flag

Browse source
This commit is contained in:
Edward Betts 2024-06-15 20:35:47 +01:00
parent 6c76610cdb
commit 964bbb1162
1 changed files with 20 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
agenda/data.py
View file

@ -110,6 +110,24 @@ async def time_function(
return name, result, end_time - start_time, exception return name, result, end_time - start_time, exception
def hide_markets_while_away(
events: list[Event], accommodation_events: list[Event]
) -> None:
"""Hide markets that happen while away."""
optional = [
e
for e in events
if e.name == "market" or (e.title and "LHG Run Club" in e.title)
]
going = [e for e in events if e.going]
overlapping_markets = find_events_during_stay(
accommodation_events + going, optional
)
for market in overlapping_markets:
events.remove(market)
async def get_data( async def get_data(
now: datetime, config: flask.config.Config now: datetime, config: flask.config.Config
) -> typing.Mapping[str, str | object]: ) -> typing.Mapping[str, str | object]:
@ -210,19 +228,8 @@ async def get_data(
events += hn.whoishiring(last_year, next_year) events += hn.whoishiring(last_year, next_year)
events += carnival.rio_carnival_events(last_year, next_year) events += carnival.rio_carnival_events(last_year, next_year)
# hide markets that happen while away if config["HIDE_MARKETS_WHILE_AWAY"]:
optional = [ hide_markets_while_away(events, accommodation_events)
e
for e in events
if e.name == "market" or (e.title and "LHG Run Club" in e.title)
]
going = [e for e in events if e.going]
overlapping_markets = find_events_during_stay(
accommodation_events + going, optional
)
for market in overlapping_markets:
events.remove(market)
for launch in rockets: for launch in rockets:
dt = None dt = None