bins.py

#!/usr/bin/env python
from datetime import date, timedelta

RECYCLING = "recycling"
RUBBISH = "rubbish"
GARDEN = "garden"


WEEKDAYS = {
    "MONDAY": 1,
    "TUESDAY": 2,
    "WEDNESDAY": 3,
    "THURSDAY": 4,
    "FRIDAY": 5,
    "SATURDAY": 6,
    "SUNDAY": 7,
}

# calculate collection dates from after this date
LAST_SUNDAY_OF_2022 = date(2022, 10, 30)
LAST_SUNDAY_OF_OCTOBER_2023 = date(2023, 10, 29)


class CollectionDate:
    def __init__(self, collection_date, collection_types):
        self.date = collection_date
        self.types = collection_types

    @staticmethod
    def _suffix(d):
        return "th" if 11 <= d <= 13 else {1: "st", 2: "nd", 3: "rd"}.get(d % 10, "th")

    @property
    def friendly_date(self):
        if self.date == date.today():
            return "today"
        elif (self.date - date.today()).days == 1:
            return "tomorrow"
        else:
            weekday = self.date.strftime("%A")
            day = "%s%s" % (self.date.day, self._suffix(self.date.day))
            month = self.date.strftime("%B")
            return "on %s the %s of %s" % (weekday, day, month)

    def __str__(self):
        return "{}: {}".format(self.date, ", ".join(self.types))


EXCEPTIONS = {
    date(2023, 12, 25): CollectionDate(date(2023, 12, 27), (RECYCLING,)),
    date(2023, 12, 26): CollectionDate(date(2023, 12, 28), (RECYCLING,)),
    date(2023, 12, 27): CollectionDate(date(2023, 12, 29), (RECYCLING,)),
    date(2023, 12, 28): CollectionDate(date(2023, 12, 30), (RECYCLING,)),
    date(2023, 12, 29): CollectionDate(date(2023, 12, 31), (RECYCLING,)),
    date(2024, 1, 1): CollectionDate(date(2024, 1, 2), (RUBBISH,)),
    date(2024, 1, 2): CollectionDate(date(2024, 1, 3), (RUBBISH,)),
    date(2024, 1, 3): CollectionDate(date(2024, 1, 4), (RUBBISH,)),
    date(2024, 1, 4): CollectionDate(date(2024, 1, 5), (RUBBISH,)),
    date(2024, 1, 5): CollectionDate(date(2024, 1, 6), (RUBBISH,)),
}


def create_collections(regular_collection_day):
    collections = []
    input_date = LAST_SUNDAY_OF_OCTOBER_2023  # Last Sunday of October 2023
    input_date += timedelta(WEEKDAYS[regular_collection_day])

    collected = (RECYCLING,)
    not_collected = (RUBBISH,)
    while input_date < date(2024, 12, 8):
        if input_date in EXCEPTIONS:
            collection = EXCEPTIONS[input_date]
        else:
            collection = CollectionDate(input_date, collected)
        collections.append(collection)
        collected, not_collected = not_collected, collected
        input_date += timedelta(days=7)

    return collections


def get_next_bin_collection(regular_collection_day, input_date=None):
    if input_date is None:
        input_date = date.today()

    if input_date <= LAST_SUNDAY_OF_2022:
        # input date is too early
        return None

    collections = create_collections(regular_collection_day)

    for collection in collections:
        if collection.date >= input_date:
            return collection


if __name__ == "__main__":
    collections = create_collections()
    for c in collections:
        print(c)

    print()
    print(get_next_bin_collection(date(2016, 12, 30)))
    print(get_next_bin_collection(date(2016, 12, 31)))
    print(get_next_bin_collection(date(2017, 1, 1)))
    print(get_next_bin_collection(date(2017, 1, 6)))
    print(get_next_bin_collection(date(2017, 1, 7)))
    print(get_next_bin_collection(date(2017, 1, 8)))

    # print(get_bin_collection())