parent
a9539545fa
commit
c6bc2f5f72
|
@ -5,7 +5,7 @@ import os
|
|||
import re
|
||||
import typing
|
||||
import warnings
|
||||
from datetime import date, datetime, timedelta, timezone
|
||||
from datetime import date, datetime, time, timedelta, timezone
|
||||
from decimal import Decimal
|
||||
from time import time as unixtime
|
||||
from typing import List
|
||||
|
@ -21,7 +21,7 @@ import pytz
|
|||
import requests
|
||||
import yaml
|
||||
from dateutil.easter import easter
|
||||
from dateutil.relativedelta import FR, relativedelta
|
||||
from dateutil.relativedelta import FR, SA, relativedelta
|
||||
|
||||
from agenda import thespacedevs
|
||||
|
||||
|
@ -327,6 +327,50 @@ def critical_mass(start_date: date, limit: int = 12) -> list[Event]:
|
|||
return events
|
||||
|
||||
|
||||
def windmill_hill_market_days(start_date: date) -> list[Event]:
|
||||
"""Windmill Hill Market days for the next 12 months from a given date."""
|
||||
events: list[Event] = []
|
||||
current_date = start_date
|
||||
|
||||
# To keep count of how many market days have been calculated
|
||||
count = 0
|
||||
|
||||
tz = pytz.timezone("Europe/London")
|
||||
|
||||
while count < 12:
|
||||
# Skip months outside of April to December
|
||||
if current_date.month < 4 or current_date.month > 12:
|
||||
current_date += relativedelta(months=1)
|
||||
current_date = date(current_date.year, current_date.month, 1)
|
||||
continue
|
||||
|
||||
# Calculate the first Saturday of the current month
|
||||
first_saturday = current_date + relativedelta(day=1, weekday=SA(+1))
|
||||
|
||||
# Include it in the list only if it's on or after the start_date
|
||||
if first_saturday >= start_date:
|
||||
events.append(
|
||||
Event(
|
||||
name="market",
|
||||
title="Windmill Hill Market",
|
||||
date=tz.localize(datetime.combine(first_saturday, time(10, 0))),
|
||||
)
|
||||
)
|
||||
count += 1
|
||||
|
||||
# Move to the next month
|
||||
current_date += relativedelta(months=1)
|
||||
current_date = date(current_date.year, current_date.month, 1)
|
||||
|
||||
return events
|
||||
|
||||
|
||||
# Test the function
|
||||
if __name__ == "__main__":
|
||||
start_date = date(2023, 10, 29)
|
||||
print(windmill_hill_market_days(start_date))
|
||||
|
||||
|
||||
def as_date(d: date | datetime) -> date:
|
||||
"""Return date for given date or datetime."""
|
||||
return d.date() if isinstance(d, datetime) else d
|
||||
|
@ -500,6 +544,7 @@ def get_data(now: datetime) -> typing.Mapping[str, str | object]:
|
|||
"xmas_last_posting_dates": xmas_last_posting_dates,
|
||||
"gwr_advance_tickets": find_gwr_advance_ticket_date(),
|
||||
"critical_mass": critical_mass(today),
|
||||
"market": windmill_hill_market_days(today),
|
||||
"rockets": thespacedevs.get_launches(rocket_dir, limit=40),
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue