Search code examples
python-3.xdatetimerelativedelta

Construct date based on week_of_month and day_of_week criteria


I am not sure how to go about constructing datetime object given year, month, week_of_month and day_of_week. Any clues? Using this I am trying to achieve following:

From (start_month, start_year) to (end_month, end_year) find monthly dates as specified by week_of_month and day_of_week parameters. Here 1 <= week_of_month <= 5 and 1 <= day_of_week <= 7. Now,

  1. Each month may not have 5 weeks (eg. February in non-leap year)
  2. 1st and 5th week may not have 7 days.

In such cases, based on boolean is_to_next_day, if True then specify next calendar day, if False then skip it.

Sample input/outputs:

  1. Input parameters: start_month=1 start_year=2020, end_month=12, end_year=2020, week_of_month=5, day_of_week=3, is_to_next_day=True
    Desired output: [datetime(2020, 1, 29), datetime(2020, 2, 26), datetime(2020, 3, 25), datetime(2020, 4, 29), datetime(2020, 5, 27), datetime(2020, 7, 1), datetime(2020, 7, 29), datetime(2020, 8, 26), datetime(2020, 9, 30), datetime(2020, 10, 28), datetime(2020, 11, 25), datetime(2020, 12, 30)]
  2. Input parameters: start_month=1 start_year=2020, end_month=12, end_year=2020, week_of_month=5, day_of_week=3, is_to_next_day=False
    Desired output: [datetime(2020, 1, 29), datetime(2020, 2, 26), datetime(2020, 3, 25), datetime(2020, 4, 29), datetime(2020, 5, 27), datetime(2020, 7, 29), datetime(2020, 8, 26), datetime(2020, 9, 30), datetime(2020, 10, 28), datetime(2020, 11, 25), datetime(2020, 12, 30)]

Solution

  • import calendar
    from datetime import datetime
    def get_date(year, month, week_of_month, day_of_week, is_to_next_day):
        mnth = calendar.monthcalendar(year, month)
        if (week_of_month > 1) and (week_of_month < 5):
            day = mnth[week_of_month - 1][day_of_week - 1]
            return datetime(year, month, day)
        elif week_of_month == 1:
            last_day_of_first_week = mnth[0][6]
            if day_of_week <= last_day_of_first_week:
                return datetime(year, month, day_of_week)
            elif is_to_next_day:
                return datetime(year, month, mnth[1][0])
            else:
                return None
        else:
            if (len(mnth) >= week_of_month):
                day = mnth[week_of_month - 1][day_of_week - 1]
                if(day==0) and is_to_next_day:
                    return datetime(year + int((month + 1)/12), (month + 1)%12, 1)
                elif(day==0):
                    return None
                else:
                    return datetime(year, month, day)
            if (len(mnth) < week_of_month):
                if is_to_next_day:
                    return datetime(year + int((month + 1)/12), (month + 1)%12, 1)
                else:
                    return None
    # First output
    [get_date(yy, mm, 5, 3, True) for mm in range(1, 13) for yy in [2020]]
    # Second output
    [get_date(yy, mm, 5, 3, False) for mm in range(1, 13) for yy in [2020]] # Iterate again to drop None.