同一Python程序中Flask接口与PyQt应用通信问题排查

问题根源 #

Qt的QApplication.exec()会阻塞主线程，若把Flask启动代码放在它之前，会被完全阻塞无法执行；若先启动Flask再跑Qt，Qt的事件循环未启动，托盘图标没法渲染。另外，Qt的UI操作必须在主线程执行，直接在Flask子线程里修改托盘图标会导致程序崩溃。

修正后的代码示例 #

import sys
from threading import Thread
from flask import Flask
from PyQt5.QtWidgets import QApplication, QMenu, QAction
from PyQt5.QtGui import QIcon, QPixmap
from PyQt5.QtCore import QObject, pyqtSignal

# 初始化Flask应用
flask_app = Flask(__name__)

# 定义跨线程信号类，用于Flask线程通知主线程修改托盘状态
class TraySignal(QObject):
    update_tray = pyqtSignal(str)  # 传递状态标识，比如"active"或"inactive"

class SystemTrayApp:
    def __init__(self, qapp):
        self.qapp = qapp
        self.tray = QSystemTrayIcon(self.qapp)
        self.signal = TraySignal()
        
        # 构建托盘右键菜单
        self.menu = QMenu()
        quit_action = QAction("退出", self.qapp)
        quit_action.triggered.connect(self.qapp.quit)
        self.menu.addAction(quit_action)
        self.tray.setContextMenu(self.menu)
        
        # 绑定信号到处理函数，确保托盘操作在主线程执行
        self.signal.update_tray.connect(self.update_tray_status)
        
        # 设置初始图标（示例用纯色图标，实际替换为你的图标路径即可）
        self.set_tray_icon("gray")
        self.tray.show()

    def set_tray_icon(self, color):
        # 生成纯色图标，实际项目替换为本地图标文件路径
        pixmap = QPixmap(32, 32)
        pixmap.fill(color)
        self.tray.setIcon(QIcon(pixmap))

    def update_tray_status(self, status):
        # 根据接口传入的状态更新托盘图标和提示
        if status == "active":
            self.set_tray_icon("green")
            self.tray.showMessage("状态更新", "已切换为活跃状态")
        else:
            self.set_tray_icon("gray")
            self.tray.showMessage("状态更新", "已切换为非活跃状态")

# Flask接口定义
@flask_app.route("/set-status/<status>")
def set_status(status):
    # 发射信号通知主线程修改托盘，避免跨线程操作Qt控件
    tray_app.signal.update_tray.emit(status)
    return f"已设置托盘状态为: {status}"

def run_flask():
    # 在子线程启动Flask，必须禁用调试模式（调试模式会重启线程导致异常）
    flask_app.run(host="0.0.0.0", port=5000, debug=False)

if __name__ == "__main__":
    # 初始化Qt应用
    qapp = QApplication(sys.argv)
    # 初始化托盘应用
    tray_app = SystemTrayApp(qapp)
    
    # 启动Flask子线程
    flask_thread = Thread(target=run_flask)
    flask_thread.daemon = True  # 设置为守护线程，主线程退出时自动终止
    flask_thread.start()
    
    # 启动Qt事件循环（主线程阻塞）
    sys.exit(qapp.exec())

关键细节说明 #

• 线程分离：Flask在子线程运行，Qt事件循环占据主线程，彻底避免互相阻塞的问题。
• 安全跨线程通信：用Qt的pyqtSignal实现子线程到主线程的消息传递，所有托盘操作都在主线程执行，符合Qt的线程安全规则。
• 守护线程设置：Flask线程设为守护线程，当Qt主线程退出时，Flask会自动终止，不会残留后台进程。

Pipfile配置示例 #

[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"

[packages]
flask = "*"
pyqt5 = "*"  # 若使用PyQt6，替换为pyqt6即可

[dev-packages]

[requires]
python_version = "3.10"  # 根据你的Python版本调整

测试步骤 #

1. 安装依赖：pipenv install
2. 启动程序：python your_script_name.py
3. 调用接口修改状态：
   • 切换为活跃状态：curl http://localhost:5000/set-status/active
   • 切换为非活跃状态：curl http://localhost:5000/set-status/inactive

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