自定义HTMLCalendar类无法生成指定日期链接的问题求助

原代码核心问题 #

1. 参数传递错误：调用formatmonth(2023, 2, data)时，data列表被错误赋值给了withyear参数（该参数预期接收布尔值），导致日期数据从未传入格式化方法。
2. *data使用不当：可变参数*data会将传入的参数解包为元组，但在方法间传递时会形成嵌套结构（如列表被包裹在元组中），导致day in data的检查逻辑失效。

修正后的代码 #

from calendar import HTMLCalendar


class MyCalendar(HTMLCalendar):
    def formatday(self, day, weekday, data=None):
        """将日期格式化为表格单元格，若日期在data中则生成带链接的单元格"""
        target_days = data or set()
        if day == 0:
            # 处理当月外的空白日期
            return '<td class="%s">&nbsp;</td>' % self.cssclass_noday
        elif day in target_days:
            return '<td class="%s"><a href="/events/%d">%d</a></td>' % (
                self.cssclasses[weekday], day, day
            )
        else:
            return '<td class="%s">%d</td>' % (self.cssclasses[weekday], day)

    def formatweek(self, theweek, data=None):
        """将一周格式化为表格行"""
        target_days = data or set()
        week_cells = ''.join(
            self.formatday(d, wd, target_days) for (d, wd) in theweek
        )
        return '<tr>%s</tr>' % week_cells

    def formatmonth(self, theyear, themonth, withyear=True, data=None):
        """将整月格式化为表格，为data中的日期添加链接"""
        # 转换为集合以提升成员检查效率（O(1)复杂度）
        target_days = set(data) if data else set()
        
        v = []
        append = v.append
        append('<table border="0" cellpadding="0" cellspacing="0" class="%s">' % self.cssclass_month)
        append('\n')
        append(self.formatmonthname(theyear, themonth, withyear=withyear))
        append('\n')
        append(self.formatweekheader())
        append('\n')
        
        for week in self.monthdays2calendar(theyear, themonth):
            append(self.formatweek(week, target_days))
            append('\n')
        
        append('</table>')
        append('\n')
        return ''.join(v)


# 使用示例
data = [3, 4]
# 推荐使用关键字参数传递data，避免参数混淆
print(MyCalendar(6).formatmonth(2023, 2, data=data))
# 也可使用位置参数（需确保withyear参数正确设置）
print(MyCalendar(6).formatmonth(2023, 2, True, data))

关键修复说明 #

1. 参数签名调整：

   • 移除所有方法中的*data，改用单个data参数并默认值为None，确保数据以统一的可迭代对象形式传递。
   • formatmonth中data作为可选参数放在withyear之后，支持关键字参数传递，避免参数混淆。

2. 数据优化处理：

   • 在formatmonth中将data转换为集合，大幅提升日期存在性检查的效率。
   • 所有方法中使用data or set()处理空数据场景，确保代码健壮性。

3. 调用方式修正：

   • 使用data=data关键字参数传递日期列表，明确指定参数用途，避免误赋值给withyear。
   • 若需高亮单个日期，需将其包裹为列表或集合（如data=[3]），而非直接传递整数。

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