PyQt5中实现点击按钮触发鼠标绘制直线功能的问题

1. 先搞定按钮与绘制状态的绑定 #

在主程序里加个状态标记，点击"Draw a line"按钮时切换状态，同时初始化直线起点变量：

# 主程序按钮点击槽函数
def on_draw_line_btn_clicked(self):
    self.drawing_line = True  # 标记当前处于画线模式
    self.line_start = None    # 用来存鼠标点击的起点坐标

2. 用QGraphicsLineItem实现直线类（不用自己写paintEvent） #

别直接继承QGraphicsItem，用现成的QGraphicsLineItem更省心，自带绘制逻辑：

from PyQt5.QtWidgets import QGraphicsLineItem
from PyQt5.QtGui import QPen

class DrawableLine(QGraphicsLineItem):
    def __init__(self, start_x, start_y, end_x, end_y, parent=None):
        super().__init__(start_x, start_y, end_x, end_y, parent)
        pen = QPen()
        pen.setWidth(2)  # 设置线宽
        self.setPen(pen)

3. 在场景的鼠标事件里处理画线逻辑 #

如果主程序用的是QGraphicsScene，直接在场景类里重写鼠标事件：

from PyQt5.QtCore import Qt

class BoardScene(QGraphicsScene):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.drawing_line = False
        self.line_start = None

    def mousePressEvent(self, event):
        # 左键点击且处于画线模式时，记录起点
        if self.drawing_line and event.button() == Qt.LeftButton:
            self.line_start = event.scenePos()
        else:
            super().mousePressEvent(event)  # 保留原有图形移动的逻辑

    def mouseReleaseEvent(self, event):
        # 左键释放时创建直线并添加到场景
        if self.drawing_line and self.line_start is not None and event.button() == Qt.LeftButton:
            line_end = event.scenePos()
            new_line = DrawableLine(self.line_start.x(), self.line_start.y(), line_end.x(), line_end.y())
            self.addItem(new_line)
            # 重置状态，退出画线模式
            self.drawing_line = False
            self.line_start = None
        else:
            super().mouseReleaseEvent(event)

4. 把按钮和场景的状态绑定起来 #

在主窗口初始化时，将按钮点击信号关联到场景的状态切换：

# 主窗口初始化代码里加这行
self.draw_line_btn.clicked.connect(lambda: setattr(self.board_scene, 'drawing_line', True))

避坑提示 #

• 别在按钮点击时直接创建直线添加到场景，必须等鼠标完成点击-释放的流程再创建，不然会因坐标未初始化导致崩溃。
• 用QGraphicsLineItem不需要自己写paintEvent，添加到场景后会自动渲染，切换其他按钮也不会消失，场景会管理所有元素的绘制。
• 如果主程序没用QGraphics框架，而是直接用QWidget，就把直线存在列表里，每次paintEvent循环重绘所有直线：

  from PyQt5.QtWidgets import QMainWindow
  from PyQt5.QtGui import QPainter, QPen
  from PyQt5.QtCore import Qt
  
  class MainWindow(QMainWindow):
      def __init__(self):
          super().__init__()
          self.drawing_line = False
          self.line_start = None
          self.lines = []  # 存所有已画的直线：(起点, 终点)
  
      def paintEvent(self, event):
          super().paintEvent(event)
          painter = QPainter(self)
          painter.setPen(QPen(Qt.black, 2))
          # 重绘所有已保存的直线
          for start, end in self.lines:
              painter.drawLine(start, end)
          # 绘制拖拽中的临时直线
          if self.drawing_line and self.line_start is not None:
              painter.drawLine(self.line_start, self.cursor().pos())
  
      def mousePressEvent(self, event):
          if self.drawing_line and event.button() == Qt.LeftButton:
              self.line_start = event.pos()
          super().mousePressEvent(event)
  
      def mouseReleaseEvent(self, event):
          if self.drawing_line and self.line_start is not None:
              self.lines.append((self.line_start, event.pos()))
              self.drawing_line = False
              self.update()  # 触发重绘
          super().mouseReleaseEvent(event)
  

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