看看内置
calendar
而不是
datetime
import calendar
import datetime
def month_start_end_work(date_src, fmt_date="%Y-%m-%d"):
dt = datetime.datetime.strptime(date_src, fmt_date)
# roll around January -> December
month = 12 if dt.month == 1 else (dt.month - 1)
year = dt.year if month != 12 else (dt.year - 1)
# discover last day of last month
for day in (31, 30, 29, 28): # handle Feb cases
try: # already the correct year for leap
month_start = calendar.weekday(year, month, day)
except ValueError: # day is out of range for month
continue # month has fewer days (always decreases)
if month_start == calendar.SATURDAY:
day -= 1
elif month_start == calendar.SUNDAY:
day -= 2
break # last day name to be used
else: # did not solve and break
raise RuntimeError("BUG: impossible code path reached")
day_last = (year, month, day) # TODO may want to make a datetime.datetime
# discover first day of last month
day = 1 # all months begin at 1
month_start = calendar.weekday(year, month, day)
if month_start == calendar.SATURDAY:
day += 2
elif month_start == calendar.SUNDAY:
day += 1
day_first = (year, month, day)
return day_first, day_last
>>> month_start_end_work("2024-06-10")
((2024, 5, 1), (2024, 5, 31))
>>> month_start_end_work("2024-07-10")
((2024, 6, 3), (2024, 6, 28))
>>> month_start_end_work("2022-01-01")
((2021, 12, 1), (2021, 12, 31))
如果你有特殊的休息日(可能每年6-12天),把它们收集到一个列表中,然后不断递减/递增以获得准确的匹配