Pythonを使用しExcelで月末月初を指定する方法がわかりません

Question

前提・実現したいこと

Pythonを使用したExcelの勤怠管理表の日付と曜日を自動入力するプログラムを製作しています。
サンプルコードには20日締めの書き方が乗っていたのですが、運用したいのは月末締めなのでコードの書き換えを行っています。

調べてみたところtimedltaを使用すれば良い、というのを見たのですがどう組み込んでよいのかわからず困っています。

以下のソースコードにあるreplace内を変えればよいのかなと思うのですが、わかるかたご教授願います。
date_start = month_last.replace(day=21)
date_end = month_current.replace(day=20)

該当のソースコード

import datetime
import openpyxl

YOUBI = ("月", "火", "水", "木", "金", "土", "日")

input_date = input("作成する年月度を入力してください（例：201608）：")

month_current = datetime.datetime.strptime(input_date + "01", "%Y%m%d")
month_last = month_current - datetime.timedelta(days=1)

date_start = month_last.replace(day=21)
date_end = month_current.replace(day=20)

file_template = "勤怠管理表.xlsx"
wb = openpyxl.load_workbook(file_template)

ws = wb["Sheet1"]

ws.cell(row=1, column=1).value = month_current.year
ws.cell(row=1, column=3).value = month_current.month

for row_num in range(13, 44):
    if date_start <= date_end:
        # 月
        ws.cell(row=row_num, column=1).value = date_start.month
        # 日
        ws.cell(row=row_num, column=2).value = date_start.day
        # 曜日
        ws.cell(row=row_num, column=3).value = YOUBI[date_start.weekday()]
    date_start += datetime.timedelta(days=1)

file_new = month_current.strftime('%Y%m') + '_' + file_template
wb.save(file_new)

Answer 1

month_current = datetime.datetime.strptime(input_date + "01", "%Y%m%d")
month_last = month_current - datetime.timedelta(days=1)

date_start = month_last.replace(day=21)
date_end = month_current.replace(day=20)

を

month_current = datetime.datetime.strptime(input_date + "01", "%Y%m%d")
month_next = month_current + datetime.timedelta(days=32)

date_start = month_current
date_end = month_next.replace(day=1) - datetime.timedelta(days=1)

に変更しましょう。

Comments

[yu-ki.hiroshima]

ありがとうございます！
無事正常に動作しました！

Answer 2

python 標準のdatetime でしたね。

以下は、pandasのTimestampの場合なので、ご参考まで。

Timestamp.days_in_month で、その月の日数が得られます。

month_current = pd.to_datetime(input_date + "01", format="%Y%m%d")

date_start = month_current
date_end = month_current.replace(day=month_current.days_in_month)

Comments

[yu-ki.hiroshima]

Pandasでも取得可能なものがあるのですね！ありがとうございます！