python计算每月的第一天

from datetime import *

def first_day_month(date1):
    now=date1
    next=date(now.year,now.month,1)+timedelta(days=31)
    return date(next.year,next.month,1)

思路：给我一个日期，我先找到这个日期当月的第一天，然后加31天变成下个月的日期，再找到下个月日期的第一天。