Python 3.7如何设置指定时区下的下月1日

原代码的问题 #

1. 时区处理错误：直接用replace(tzinfo=timezone)设置时区，无法正确处理夏令时（DST）转换，导致时区信息不准确。
2. 时分秒未清零：计算时保留了原时间的时分秒，得到的不是目标时区下的00:00:00。
3. 月份计算不严谨：timedelta(days=32)的方式可能跨到下下月（比如1月加32天会到3月），导致结果错误。

正确解决方案 #

方案一：使用标准库+python-dateutil（推荐，处理月份更严谨）

from datetime import datetime
from zoneinfo import ZoneInfo
from dateutil.relativedelta import relativedelta

# 示例：当前本地时间（无时区）
current_datetime = datetime(2023, 8, 15, 12, 30, 45)
# 指定目标时区
tz = ZoneInfo("America/Los_Angeles")

# 转换为时区感知的datetime
current_tz = current_datetime.astimezone(tz)
# 如果当前时间是UTC，用下面的转换方式：
# current_tz = current_datetime.replace(tzinfo=ZoneInfo("UTC")).astimezone(tz)

# 获取指定时区下月1日0点
next_month_first = (current_tz.replace(day=1, hour=0, minute=0, second=0, microsecond=0) 
                    + relativedelta(months=1))

# 转换为时间戳
target_timestamp = int(next_month_first.timestamp())

方案二：纯标准库实现（无额外依赖）

from datetime import datetime
from zoneinfo import ZoneInfo

def get_next_month_first(dt_tz):
    # 获取当月1日0点的时区感知时间
    first_of_current = dt_tz.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
    year = first_of_current.year
    month = first_of_current.month
    
    # 计算下月的年、月
    next_month = month + 1
    next_year = year
    if next_month > 12:
        next_month = 1
        next_year += 1
    
    # 构造下月1日0点的时区感知时间
    return datetime(next_year, next_month, 1, 0, 0, 0, tzinfo=dt_tz.tzinfo)

# 示例使用
current_datetime = datetime(2023, 8, 15, 12, 30, 45)
tz = ZoneInfo("America/Los_Angeles")
current_tz = current_datetime.astimezone(tz)

next_month_first = get_next_month_first(current_tz)
target_timestamp = int(next_month_first.timestamp())

验证结果 #

对于示例中的2023年8月15日（America/Los Angeles时区），上述代码会得到2023-09-01 00:00:00-07:00，对应的时间戳为1693555200，与预期完全一致。

内容的提问来源于stack exchange，提问作者Jack tileman