Python应用中如何准确查询即将到来的生日？解决跨年查询难题

一、SQL查询方案（数据库层直接处理） #

针对跨年查询和2月29日闰日的问题，可以通过计算下一次生日的实际日期来筛选和排序，直接在SQL中完成逻辑：

SELECT 
    user_id, 
    DATE_FORMAT(birthday, '%m-%d') AS upcoming_birthday,
    -- 计算下一次生日的具体日期，处理闰日
    CASE
        -- 今年的生日尚未到来（含当天），用今年的日期
        WHEN STR_TO_DATE(CONCAT(YEAR(NOW()), '-', DATE_FORMAT(birthday, '%m-%d')), '%Y-%m-%d') >= CURDATE()
            THEN STR_TO_DATE(CONCAT(YEAR(NOW()), '-', DATE_FORMAT(birthday, '%m-%d')), '%Y-%m-%d')
        -- 今年的生日已过，计算明年的日期，处理非闰年的2月29日
        ELSE 
            CASE
                WHEN DATE_FORMAT(birthday, '%m-%d') = '02-29' 
                     AND NOT ((YEAR(NOW())+1) % 4 = 0 AND (YEAR(NOW())+1) % 100 != 0 OR (YEAR(NOW())+1) % 400 = 0)
                    THEN STR_TO_DATE(CONCAT(YEAR(NOW())+1, '-02-28'), '%Y-%m-%d')
                ELSE STR_TO_DATE(CONCAT(YEAR(NOW())+1, '-', DATE_FORMAT(birthday, '%m-%d')), '%Y-%m-%d')
            END
    END AS next_birthday
FROM birthdays
WHERE guild_id = %s
ORDER BY next_birthday;

逻辑说明： #

1. 提取生日的mm-dd部分，拼接当前年份生成今年的生日日期；
2. 如果今年的生日日期晚于等于今天，直接用这个日期作为下一次生日；
3. 如果今年的生日已过，则计算明年的生日日期，若明年不是闰年且原生日是2月29日，自动替换为2月28日；
4. 最终按next_birthday排序，确保结果按生日到来的先后顺序排列。

二、Python处理方案（应用层处理） #

如果更倾向于在Python中处理逻辑，可以先从数据库取出所有生日记录，再通过代码计算下一次生日并筛选排序：

from datetime import datetime, timedelta

def get_upcoming_birthdays(birthday_records, days_ahead=None):
    """
    获取即将到来的生日列表，支持跨年和闰日处理
    :param birthday_records: 数据库查询结果，格式为[(user_id, '2020-mm-dd'), ...]
    :param days_ahead: 可选，仅返回接下来N天内的生日，默认返回所有即将到来的生日
    :return: 排序后的列表，格式为[(user_id, 'mm-dd'), ...]
    """
    today = datetime.today().date()
    upcoming = []

    for user_id, birthday_str in birthday_records:
        # 提取月日部分
        birth_month, birth_day = map(int, birthday_str.split('-')[1:])

        # 计算今年的生日日期，处理非闰年的2月29日
        try:
            this_year_birth = datetime(today.year, birth_month, birth_day).date()
        except ValueError:
            this_year_birth = datetime(today.year, 2, 28).date()

        # 确定下一次生日的日期
        if this_year_birth >= today:
            next_birth = this_year_birth
        else:
            try:
                next_birth = datetime(today.year + 1, birth_month, birth_day).date()
            except ValueError:
                next_birth = datetime(today.year + 1, 2, 28).date()

        # 按days_ahead筛选（如果指定）
        if days_ahead is None or (next_birth - today).days <= days_ahead:
            upcoming.append((user_id, f"{birth_month:02d}-{birth_day:02d}", next_birth))

    # 按生日到来的先后排序
    upcoming.sort(key=lambda x: x[2])

    # 返回仅包含user_id和生日月日的结果
    return [(item[0], item[1]) for item in upcoming]

使用示例： #

# 从数据库查询指定guild_id的所有生日记录
import mysql.connector

db = mysql.connector.connect(host="your_host", user="your_user", password="your_pwd", database="your_db")
cursor = db.cursor()
cursor.execute("SELECT user_id, birthday FROM birthdays WHERE guild_id = %s", (your_guild_id,))
records = cursor.fetchall()

# 获取所有即将到来的生日
upcoming_birthdays = get_upcoming_birthdays(records)
# 或者获取接下来30天内的生日
# upcoming_birthdays = get_upcoming_birthdays(records, days_ahead=30)

print(upcoming_birthdays)

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