如何处理非4位年份日期？适配datetime实现日期加减

Python的datetime模块默认只认%Y-%m-%d格式的4位年份日期，传入像"999-12-27"（少于4位）或"12023-01-20"（多于4位）这类日期时，直接会触发ValueError。我们需要实现日期加减后，保留原年份的位数逻辑，比如：

• "212023-12-30"加2天 → "212024-01-01"
• "999-12-30"加2天 → "1000-01-01"

解决思路 #

核心就是把非4位年份的日期拆成「年份数字+月日」，用datetime处理日期计算后，再把计算后的年份数字按原格式拼回去：

1. 用正则拆分出日期里的年份、月份、日部分
2. 把年份转成整数，直接构造datetime对象（datetime本身支持任意合法整数年份）
3. 执行加减操作后，把新的年份转成字符串，和原格式的月日拼接成结果

完整代码实现 #

import datetime
import re

def add_or_subtract_days(datestr, days, operation):
    # 匹配任意位数年份的日期格式
    date_match = re.match(r"(?P<year>\d+)-(?P<month>\d{2})-(?P<day>\d{2})", datestr)
    if not date_match:
        raise ValueError("日期格式必须为「任意位数年份-两位月份-两位日期」")
    
    year_str, month_str, day_str = date_match.groups()
    # 转成整数构造datetime对象
    try:
        target_date = datetime.datetime(int(year_str), int(month_str), int(day_str))
    except ValueError as e:
        raise ValueError(f"无效日期: {datestr}") from e
    
    # 执行加减操作
    if operation == "add":
        new_date = target_date + datetime.timedelta(days=int(days))
    elif operation == "subtract":
        new_date = target_date - datetime.timedelta(days=int(days))
    else:
        raise ValueError("operation参数只能是'add'或'subtract'")
    
    # 拼接结果：用新年份替换原年份，月日保持两位格式
    return f"{new_date.year}-{new_date.strftime('%m-%d')}"

# 测试示例
print(add_or_subtract_days("212023-12-30", 2, "add"))  # 输出: 212024-01-01
print(add_or_subtract_days("999-12-30", 2, "add"))      # 输出: 1000-01-01
print(add_or_subtract_days("12023-01-20", 10, "add"))   # 输出: 12023-01-30
print(add_or_subtract_days("1000-01-01", 1, "subtract"))# 输出: 999-12-31

代码关键点说明 #

• 正则兼容任意位数年份，不管是3位、5位还是更多，都能正确拆分
• 直接用整数年份构造datetime，避开strptime对4位年份的限制
• 结果里的月日用strftime('%m-%d')保证两位格式，和输入一致
• 加入了格式校验和异常捕获，提前拦截无效输入

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