为PyQt5 QCalendarWidget单元格上色并在单元格内打印数据

from PyQt5.QtWidgets import QApplication, QCalendarWidget, QStyledItemDelegate, QStyleOptionViewItem, QTableView
from PyQt5.QtCore import Qt, QDate, QAbstractTableModel

class CalendarDelegate(QStyledItemDelegate):
    def paint(self, painter, option, index):
        # 获取日期数据
        date = index.data(Qt.DisplayRole)
        # 获取日期的年份
        year = date.year()
        # 获取日期的月份
        month = date.month()
        
        # 根据日期的年份和月份来确定单元格的背景色
        if year == 2022 and month == 1:
            option.palette.setColor(option.palette.Base, Qt.red)
        elif year == 2022 and month == 2:
            option.palette.setColor(option.palette.Base, Qt.green)
        else:
            option.palette.setColor(option.palette.Base, Qt.white)
        
        # 绘制背景色
        painter.fillRect(option.rect, option.palette.base())
        
        # 绘制日期文本
        painter.drawText(option.rect, Qt.AlignCenter, str(date.day()))

class CalendarModel(QAbstractTableModel):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.calendar = QCalendarWidget()
    
    def rowCount(self, parent):
        return 6
    
    def columnCount(self, parent):
        return 7
    
    def data(self, index, role):
        if role == Qt.DisplayRole:
            # 获取日期数据
            date = self.calendar.selectedDate().addDays(index.column() + index.row() * 7)
            return date
        return None

app = QApplication([])
calendar = QTableView()
model = CalendarModel()
delegate = CalendarDelegate()
calendar.setModel(model)
calendar.setItemDelegate(delegate)
calendar.show()
app.exec_()