使用后重新运行python脚本QProcess

时间:2021-05-15 17:57:25

标签: python pyqt5 qprocess

我想根据spinBox的值在PyQt5中排队QProcess,以及使用readAll()在textEdit中显示文本,但是无论我在spinBox中指定什么值,脚本只运行1次,其结果不显示在文字编辑。请告诉我解决方法。 我最近才开始学习python。也许对我来说还是太难了,但我想解决一下。

test.py

from functools import cached_property
from pathlib import Path
from PyQt5 import QtGui, QtWidgets, QtCore, uic
import queue
import os
import sys

CURRENT_DIRECTORY = Path(__file__).resolve().parent


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        QtWidgets.QMainWindow.__init__(self)
        self.ui = uic.loadUi(os.fspath(CURRENT_DIRECTORY / "test.ui"), self)
        self.resize(820, 300)
        self.setFixedSize(self.size())
        managers = TaskManager(self)
        task_list = [os.fspath(CURRENT_DIRECTORY / "test2.py")]
        for task in task_list:
            managers.appendTask(task)
        self.ui.pushButtonStart.clicked.connect(managers.start)
        self.ui.pushButtonStop.clicked.connect(managers.stop)
        managers.messageChanged.connect(self.textEdit.append)


    @cached_property
    def manager(self):
        return TaskManager()


class TaskManager(QtCore.QObject):
    messageChanged = QtCore.pyqtSignal(str)
    numbersTaskRunningChanged = QtCore.pyqtSignal(int)

    def __init__(self, parent=None):
        super(TaskManager, self).__init__(parent)
        self._max_task = 1
        self._queue = queue.Queue()
        self._numbers_task_running = 0
        self._running = False

    @cached_property
    def processes(self):
        return list()

    def execute(self, script, metadata=None):
        process = QtCore.QProcess()
        process.setProperty("metadata", metadata or dict())
        process.finished.connect(self.handle_finished)
        process.setProgram(sys.executable)
        process.setArguments([script])
        process.start()
        self.processes.append(process)

    def handle_finished(self):
        process = self.sender()
        self.processes.remove(process)
        metadata = process.property("metadata")
        print(f"{metadata} finished")

    def setMaxTask(self, max_task):
        self._max_task = max_task
        if self._running:
            self.call_task()

    def maxTask(self):
        return self._max_task

    def appendTask(self, task):
        self._queue.put(task)
        self.call_task()

    def start(self):
        interface = MainWindow()
        interface.pushButtonStart.setEnabled(False)
        interface.pushButtonStop.setEnabled(True)
        self._running = True
        self.call_task()
        number_of_processes = interface.ui.spinBox.value()
        script = os.fspath(CURRENT_DIRECTORY / "test2.py")
        for i in range(number_of_processes):
            interface.manager.execute(script, dict(i=i))

    def stop(self):
        interface = MainWindow()
        interface.pushButtonStart.setEnabled(True)
        interface.pushButtonStop.setEnabled(False)
        self._running = False

    def call_task(self):
        process = QtCore.QProcess(self)
        process.setProcessChannelMode(QtCore.QProcess.MergedChannels)
        process.readyReadStandardOutput.connect(self.on_readyReadStandardOutput)
        process.finished.connect(self.on_finished)
        process.started.connect(self.on_started)
        process.errorOccurred.connect(self.on_errorOccurred)

    def on_readyReadStandardOutput(self):
        codec = QtCore.QTextCodec.codecForLocale()
        decoder_stdout = codec.makeDecoder()
        process = self.sender()
        text = decoder_stdout.toUnicode(process.readAllStandardOutput())
        self.messageChanged.emit(text)

    def on_errorOccurred(self, error):
        process = self.sender()
        print("error: ", error, "-", " ".join([process.program()] + process.arguments()))
        self.call_task()

    def on_finished(self):
        process = self.sender()
        self._numbers_task_running -= 1
        self.numbersTaskRunningChanged.emit(self._numbers_task_running)
        self.call_task()

    def on_started(self):
        process = self.sender()
        print("started: ", " ".join([process.program()] + process.arguments()))
        self._numbers_task_running += 1
        self.numbersTaskRunningChanged.emit(self._numbers_task_running)
        self.call_task()

if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)

    w = MainWindow()
    w.show()

    sys.exit(app.exec())

test.ui

<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
 <class>MainWindow</class>
 <widget class="QMainWindow" name="MainWindow">
  <property name="geometry">
   <rect>
    <x>0</x>
    <y>0</y>
    <width>820</width>
    <height>300</height>
   </rect>
  </property>
  <property name="windowTitle">
   <string>MainWindow</string>
  </property>
  <property name="styleSheet">
   <string notr="true">background-color: rgb(245, 245, 245);</string>
  </property>
  <widget class="QWidget" name="centralwidget">
   <widget class="QWidget" name="verticalLayoutWidget">
    <property name="geometry">
     <rect>
      <x>0</x>
      <y>0</y>
      <width>121</width>
      <height>271</height>
     </rect>
    </property>
    <layout class="QVBoxLayout" name="verticalLayout">
     <property name="sizeConstraint">
      <enum>QLayout::SetDefaultConstraint</enum>
     </property>
     <item>
      <widget class="QPushButton" name="pushButtonStart">
       <property name="enabled">
        <bool>true</bool>
       </property>
       <property name="font">
        <font>
         <pointsize>10</pointsize>
        </font>
       </property>
       <property name="styleSheet">
        <string notr="true">background-color: rgb(0, 170, 0);</string>
       </property>
       <property name="text">
        <string>Start</string>
       </property>
      </widget>
     </item>
     <item>
      <widget class="QPushButton" name="pushButtonStop">
       <property name="enabled">
        <bool>false</bool>
       </property>
       <property name="font">
        <font>
         <pointsize>10</pointsize>
        </font>
       </property>
       <property name="styleSheet">
        <string notr="true">background-color: rgb(255, 0, 0);</string>
       </property>
       <property name="text">
        <string>Stop</string>
       </property>
      </widget>
     </item>
     <item>
      <widget class="QFrame" name="frame">
       <property name="frameShape">
        <enum>QFrame::StyledPanel</enum>
       </property>
       <property name="frameShadow">
        <enum>QFrame::Raised</enum>
       </property>
       <widget class="QWidget" name="layoutWidget">
        <property name="geometry">
         <rect>
          <x>10</x>
          <y>10</y>
          <width>106</width>
          <height>20</height>
         </rect>
        </property>
        <layout class="QGridLayout" name="gridLayout_2">
         <item row="0" column="0">
          <widget class="QLabel" name="label_6">
           <property name="text">
            <string>Value</string>
           </property>
          </widget>
         </item>
         <item row="0" column="1">
          <widget class="QSpinBox" name="spinBox">
           <property name="minimum">
            <number>1</number>
           </property>
           <property name="maximum">
            <number>4000</number>
           </property>
           <property name="value">
            <number>1</number>
           </property>
          </widget>
         </item>
        </layout>
       </widget>
      </widget>
     </item>
    </layout>
   </widget>
   <widget class="QWidget" name="verticalLayoutWidget_2">
    <property name="geometry">
     <rect>
      <x>120</x>
      <y>0</y>
      <width>691</width>
      <height>271</height>
     </rect>
    </property>
    <layout class="QVBoxLayout" name="verticalLayout_2">
     <item>
      <widget class="QTextEdit" name="textEdit">
       <property name="readOnly">
        <bool>true</bool>
       </property>
      </widget>
     </item>
    </layout>
   </widget>
  </widget>
  <widget class="QStatusBar" name="statusbar"/>
 </widget>
 <resources/>
 <connections/>
</ui>

test2.py

import random, string

listToPrint = set()
listToPrint.add(''.join(random.choice(string.hexdigits) for i in range(16)))
print(listToPrint)

2 个答案:

答案 0 :(得分:1)

您的请求令人困惑,因为您谈论的是队列但您从未使用它,似乎您已经修补了我的代码,认为它们会神奇地加入并工作,但现实(不幸的是,幸运的是)并非如此。

>

你必须明白,一个好的做法是将每个班级的任务分开。 TaskManager 的任务是做任务的队列处理,不是直接和 GUI 交互。 GUI 的任务是获取用户信息并启动 TaskManager。

以上述为基础,我创建了一个类,如果您添加任务,它们将按顺序执行,即当我执行完一个任务时,它只会执行一个挂起的任务。而 stop 方法的作用是停止正在执行的任务并消除挂起的任务。同一个类将通过信号发出进程的信息。

GUI 必须获取该信息并将其转换为字符串以在 QTextEdit 中显示。另一方面,TaskManager 类有一个 running_changed 信号,用于指示何时执行任务以启用或禁用停止按钮,因为在没有任务执行时启用它是没有意义的。

from functools import cached_property
from pathlib import Path
from PyQt5 import QtGui, QtWidgets, QtCore, uic
from collections import deque
import os
import sys

CURRENT_DIRECTORY = Path(__file__).resolve().parent


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        QtWidgets.QMainWindow.__init__(self)
        self.ui = uic.loadUi(os.fspath(CURRENT_DIRECTORY / "test.ui"), self)
        self.setFixedSize(820, 300)

        self.ui.pushButtonStop.setEnabled

        self.ui.pushButtonStart.clicked.connect(self.handle_start)
        self.ui.pushButtonStop.clicked.connect(self.handle_stop)
        self.manager.standard_output_changed.connect(self.process_data)
        self.manager.standard_error_changed.connect(self.process_data)

        self.manager.running_changed.connect(self.handle_running_changed)

    def handle_start(self):
        number_of_processes = self.ui.spinBox.value()
        for _ in range(number_of_processes):
            self.manager.append(
                sys.executable, [os.fspath(CURRENT_DIRECTORY / "test2.py")]
            )

    def handle_stop(self):
        self.manager.stop()

    def handle_running_changed(self):
        self.ui.pushButtonStop.setEnabled(self.manager.running)

    @cached_property
    def manager(self):
        return TaskManager()

    def process_data(self, data):
        codec = QtCore.QTextCodec.codecForLocale()
        decoder_stdout = codec.makeDecoder()
        text = decoder_stdout.toUnicode(data)
        self.textEdit.append(text)


class TaskManager(QtCore.QObject):
    standard_output_changed = QtCore.pyqtSignal(QtCore.QByteArray)
    standard_error_changed = QtCore.pyqtSignal(QtCore.QByteArray)
    running_changed = QtCore.pyqtSignal()

    _running = False

    @cached_property
    def process(self):
        process = QtCore.QProcess()
        process.readyReadStandardOutput.connect(self.handle_standard_output)
        process.readyReadStandardError.connect(self.handle_standard_error)
        process.finished.connect(self.try_start)
        return process

    @cached_property
    def queue(self):
        return deque()

    @property
    def running(self):
        return self._running

    @running.setter
    def running(self, running):
        if self.running == running:
            return
        self._running = running
        self.running_changed.emit()

    def append(self, program, arguments=()):
        self.queue.append((program, arguments))
        self.try_start()

    def try_start(self):
        if len(self.queue) == 0:
            self.running = False
            return
        if self.process.state() == QtCore.QProcess.NotRunning:
            program, arguments = self.queue.popleft()
            self.process.setProgram(program)
            self.process.setArguments(arguments)
            self.process.start()
            self.running = True

    def handle_standard_output(self):
        self.standard_output_changed.emit(self.process.readAllStandardOutput())

    def handle_standard_error(self):
        self.standard_error_changed.emit(self.process.readAllStandardError())

    def stop(self):
        if self.process.state() != QtCore.QProcess.NotRunning:
            self.process.kill()
        self.queue.clear()
        self.running = False


if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)

    w = MainWindow()
    w.show()

    sys.exit(app.exec())

为了测试我的代码,我修改了 test2.py,通过添加像 time.sleep 这样的耗时任务使其更加复杂。这样你就可以完成正在运行的任务。另一方面,每按一次开始按钮,就会向正在执行的任务中添加 n 个任务。

import random
import string
import time


listToPrint = set()
for i in range(4):
    listToPrint.add("".join(random.sample(string.hexdigits, 16)))
    print(i, flush=True)
    time.sleep(1)
print(listToPrint, flush=True)

答案 1 :(得分:0)

问题在于您不断创建 MainWindow 的新实例,因此用于跟踪进程数的 spinbox 始终具有默认初始值 1。

您还不止一次地犯了同样的错误,而且您甚至没有为 TaskManager 使用 cached_property。

一个可能的解决方案是将主窗口实例传递给任务管理器,以便它使用旋转框的实际值。考虑到这不是一个非常优雅的解决方案,因为任务管理器可能应该对主窗口一无所知,并且所有通信都应该使用更多信号和槽进行。

class MainWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        QtWidgets.QMainWindow.__init__(self)
        uic.loadUi("test.ui", self)
        self.resize(820, 300)
        self.setFixedSize(self.size())
        
        self.manager = TaskManager(self)
        task_list = ["test2.py"]
        for task in task_list:
            self.manager.appendTask(task)
        self.pushButtonStart.clicked.connect(self.manager.start)
        self.pushButtonStop.clicked.connect(self.manager.stop)
        self.manager.messageChanged.connect(self.textEdit.append)


class TaskManager(QtCore.QObject):
    messageChanged = QtCore.pyqtSignal(str)
    numbersTaskRunningChanged = QtCore.pyqtSignal(int)

    def __init__(self, mainWindow):
        super(TaskManager, self).__init__(mainWindow)
        self.mainWindow = mainWindow
        # ...

    def start(self):
        self._running = True
        self.call_task()
        number_of_processes = self.mainWindow.spinBox.value()
        script = "test2.py"
        for i in range(number_of_processes):
            self.execute(script, dict(i=i))

    def stop(self):
        self.mainWindow.pushButtonStart.setEnabled(True)
        self.mainWindow.pushButtonStop.setEnabled(False)
        self._running = False

请注意,上述修正非常小,并且由于各种原因,代码在面向对象编程的许多方面都不是最佳的。

考虑到,从你的代码来看,你可能试图改编别人的代码,但没有完全理解其机制,我强烈建议你仔细研究原文和我的修改,对类/实例做更多的耐心研究和一般的面向对象编程。