填充QCalendarWidget

Question

我创建了一个GUI来显示通过PyQt QCalendarWidget运行的一些训练。以下代码是根据这些员额编写的：

• I want to put the text in pyqt QCalendarWidget
• How to add text to PyQt QcalendarWidget

我创建了一个子类，以覆盖paint方法。当我在paintCell方法中有一个硬编码的日期时，这是很好的。但理想情况下，我希望首先运行一个函数，它将返回一组日期/运行距离作为数据格式。然后，该数据将用于“填充”QCalendarWidget (例如，将运行距离添加为相应日期的文本)。

class MyQtApp(trigui.Ui_MainWindow, QtWidgets.QMainWindow):
    def __init__(self):
        super(MyQtApp, self).__init__()
        self.setupUi(self)
        self.Calendar()
        self.df = pd.DataFrame()
        self.qPlan_Create_btn.clicked.connect(self.draw_running_plan)

    def Calendar(self):
        self.cal = CalendarWidget(self.qPlan_Widget)
        self.cal.resize(1700, 800)

    def draw_running_plan(self):
        self.df = pd.DataFrame([[25-05-2021, 10], [27-05-2021, 12]], columns=['Date', 'Distance'])
       #########
       # how can I pass this dataframe to the paintCell

class CalendarWidget(QtWidgets.QCalendarWidget):
    def paintCell(self, painter, rect, date):
        painter.setRenderHint(QtGui.QPainter.Antialiasing, True)
        if date == QtCore.QDate(2021, 5, 15):
            painter.save()
            painter.drawRect(rect)
            painter.setPen(QtGui.QColor(168, 34, 3))
            painter.drawText(rect, QtCore.Qt.AlignHCenter, 'Hello\nWorld')
            painter.drawText(QtCore.QRectF(rect), 
             QtCore.Qt.TextSingleLine|QtCore.Qt.AlignCenter, str(date.day()))

            painter.restore()
        else:
            QtWidgets.QCalendarWidget.paintCell(self, painter, rect, date)

def main():
    import sys

    app = QtWidgets.QApplication(sys.argv)
    qt_app = MyQtApp()
    qt_app.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()

不知何故，我需要从我的__init__中调用方法日历。我尝试将self.cal = CalendarWidget (self.qPlan_Widget)设置在函数draw_running_plan中，但随后没有显示日历。对于信息，self.qPlan_Widget是我通过QtDesigner创建的一个简单的Widget/容器，我通过Calendar方法将它初始化为CalendarWidget。长话短说:初始化CalendarWidget之后，如何用中间函数的结果更新它？

编辑:我对标签的错误是PySide2而不是PyQt

Answer 1

一个可能的解决方案是创建一个属性，在setter中使用内部QTableView视图端口的QTableView方法，它将调用paintEvent方法，该方法在其逻辑中调用paintCell()方法：

另一方面，对于筛选，最好将日期字符串列(至少看起来是OP提供的)转换为datetime。然后根据第一天和第二天的最小日期做一个过滤器。

import pandas as pd
from PySide2 import QtCore, QtGui, QtWidgets
import datetime


class CalendarWidget(QtWidgets.QCalendarWidget):
    _dataframe = pd.DataFrame()

    @property
    def dataframe(self):
        return self._dataframe

    @dataframe.setter
    def dataframe(self, df):
        self._dataframe = df.copy()
        view = self.findChild(QtWidgets.QTableView, "qt_calendar_calendarview")
        if view is not None:
            view.viewport().update()

    def paintCell(self, painter, rect, date):
        painter.setRenderHint(QtGui.QPainter.Antialiasing, True)
        if self._dataframe.empty:
            QtWidgets.QCalendarWidget.paintCell(self, painter, rect, date)
            return

        if hasattr(date, "toPyDate"):
            dt_start = datetime.datetime.combine(
                date.toPyDate(), datetime.datetime.min.time()
            )
        else:
            dt_start = datetime.datetime.strptime(
                date.toString("yyyy-MM-dd"), "%Y-%m-%d"
            )

        dt_end = dt_start + datetime.timedelta(days=1)

        mask = (dt_start <= self.dataframe["Date"]) & (self.dataframe["Date"] < dt_end)
        res = self.dataframe.loc[mask]
        if res.empty:
            QtWidgets.QCalendarWidget.paintCell(self, painter, rect, date)
        else:
            value = res.iloc[0, res.columns.get_loc("Distance")]

            painter.save()
            painter.drawRect(rect)
            painter.setPen(QtGui.QColor(168, 34, 3))
            painter.drawText(
                QtCore.QRectF(rect),
                QtCore.Qt.TextSingleLine | QtCore.Qt.AlignCenter,
                str(value),
            )
            painter.restore()


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self, parent=None):
        super().__init__(parent)

        button = QtWidgets.QPushButton("Change")
        self.calendar_widget = CalendarWidget()

        central_widget = QtWidgets.QWidget()
        self.setCentralWidget(central_widget)

        lay = QtWidgets.QVBoxLayout(central_widget)
        lay.addWidget(button)
        lay.addWidget(self.calendar_widget)

        button.clicked.connect(self.handle_clicked)

    def handle_clicked(self):
        import random

        df = pd.DataFrame(
            [
                ["25-05-2021", random.randint(0, 100)],
                ["27-05-2021", random.randint(0, 100)],
            ],
            columns=["Date", "Distance"],
        )
        df["Date"] = pd.to_datetime(df["Date"])
        self.calendar_widget.dataframe = df


def main():
    import sys

    app = QtWidgets.QApplication(sys.argv)

    widget = MainWindow()
    widget.show()

    sys.exit(app.exec_())


if __name__ == "__main__":
    main()