Construct date based on week_of_month and day_of_week criteria

57 Views Asked by At

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)]
1

There are 1 best solutions below

1
On BEST ANSWER
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.