PyQt5 QCalendarWidget – 抓取矩形

简介

在 PyQt5 中，QCalendarWidget 是一个用于显示日历的控件。它提供了丰富的日历显示和交互功能，并支持自定义。本文将介绍如何使用 QCalendarWidget 控件来抓取矩形。

代码实现

QCalendarWidget 控件提供了 activated 信号，当用户点击日期时会发出此信号，其中会包含所选日期的信息。为了实现抓取矩形的功能，我们可以在用户点击日期时记录所选日期，并且在用户拖拽鼠标进行选中操作时记录其矩形区域。具体实现步骤如下：

1. 创建 QCalendarWidget 控件

calendar = QCalendarWidget()

2. 设置 activated 信号的槽函数

calendar.activated.connect(self.handle_activated)

3. 编写 handle_activated 槽函数

def handle_activated(self, date):
    self.selected_dates.append(date)

其中，selected_dates 是用于记录所选日期的列表。

4. 监听鼠标事件，处理用户拖拽操作

def mousePressEvent(self, event):
    self.start_pos = event.pos()

def mouseReleaseEvent(self, event):
    if self.start_pos and self.end_pos:
        rect = QRect(self.start_pos, self.end_pos)
        self.selected_rects.append(rect)
        self.start_pos = None
        self.end_pos = None

def mouseMoveEvent(self, event):
    self.end_pos = event.pos()
    self.update()

其中，selected_rects 是用于记录用户拖拽操作矩形区域的列表。update() 方法用于强制 QWidget 重新绘制，以便在用户拖拽时实时展现选中矩形区域。

完整代码：

from PyQt5.QtWidgets import QApplication, QCalendarWidget, QWidget
from PyQt5.QtCore import QRect, Qt, QDate
from PyQt5.QtGui import QPainter, QPen, QColor

class CalendarWidget(QWidget):
    def __init__(self):
        super().__init__()
        self.selected_dates = []
        self.selected_rects = []
        self.start_pos = None
        self.end_pos = None
        self.init_ui()
    
    def init_ui(self):
        self.setWindowTitle("Calendar Widget")
        self.setGeometry(100, 100, 400, 400)

        calendar = QCalendarWidget()
        calendar.setGridVisible(True)
        calendar.activated.connect(self.handle_activated)

        self.setLayout(calendar)

    def handle_activated(self, date):
        self.selected_dates.append(date)

    def paintEvent(self, event):
        if not self.start_pos or not self.end_pos:
            return
        painter = QPainter(self)
        painter.setPen(QPen(QColor('red'), 2, Qt.DotLine))
        painter.drawRect(QRect(self.start_pos, self.end_pos))

    def mousePressEvent(self, event):
        self.start_pos = event.pos()

    def mouseReleaseEvent(self, event):
        if self.start_pos and self.end_pos:
            rect = QRect(self.start_pos, self.end_pos)
            self.selected_rects.append(rect)
            self.start_pos = None
            self.end_pos = None
    
    def mouseMoveEvent(self, event):
        self.end_pos = event.pos()
        self.update()

if __name__ == "__main__":
    app = QApplication([])
    widget = CalendarWidget()
    widget.show()
    app.exec_()

结语

通过以上方法，我们可以很方便地实现抓取矩形的功能。本文通过 QCalendarWidget 控件为例，介绍了具体实现步骤。在实际应用中，我们也可以根据具体需要选择其他控件或自定义控件实现类似操作。