鍍金池/ 問(wèn)答/Python/ pyqt 如何設(shè)置 暫停程序 等待 UI輸入?yún)?shù) 繼續(xù)進(jìn)程

pyqt 如何設(shè)置 暫停程序 等待 UI輸入?yún)?shù) 繼續(xù)進(jìn)程

程序?yàn)樽詣?dòng)登陸百度推廣后臺(tái),要點(diǎn)在于,截屏裁剪獲取驗(yàn)證碼圖片,顯示在UI窗口,然后在窗口手動(dòng)輸入驗(yàn)證碼,程序獲取驗(yàn)證碼,繼而登陸。
現(xiàn)在的問(wèn)題是,找不到好的暫停和繼續(xù)線程的方法,全都卡死,求大手子指教!

from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QMessageBox
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
import time,os,sys,datetime,re
import pandas as pd,numpy as np
from pandas import DataFrame,Series
from openpyxl import Workbook,load_workbook
from PIL import Image
from PyQt5.QtWidgets import QApplication, QMainWindow
import threading

class Ui_MainWindow(object):

    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(532, 229)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.label = QtWidgets.QLabel(self.centralwidget)
        self.label.setGeometry(QtCore.QRect(30, 110, 61, 20))
        self.label.setObjectName("label")
        self.lineEdit_1 = QtWidgets.QLineEdit(self.centralwidget)
        self.lineEdit_1.setGeometry(QtCore.QRect(100, 110, 113, 20))
        self.lineEdit_1.setObjectName("lineEdit_1")
        self.label_5 = QtWidgets.QLabel(self.centralwidget)
        self.label_5.setGeometry(QtCore.QRect(30, 70, 61, 16))
        self.label_5.setObjectName("label_5")
        self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_2.setGeometry(QtCore.QRect(230, 110, 75, 21))
        self.pushButton_2.setObjectName("pushButton_2")
        self.label_7 = QtWidgets.QLabel(self.centralwidget)
        self.label_7.setGeometry(QtCore.QRect(40, 150, 51, 21))
        self.label_7.setObjectName("label_7")
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setGeometry(QtCore.QRect(60, 20, 158, 23))
        self.pushButton.setObjectName("pushButton")
        self.label_2 = QtWidgets.QLabel(self.centralwidget)
        self.label_2.setGeometry(QtCore.QRect(93, 150, 301, 31))
        self.label_2.setText("")
        self.label_2.setObjectName("label_2")
        self.label_3 = QtWidgets.QLabel(self.centralwidget)
        self.label_3.setGeometry(QtCore.QRect(110, 61, 191, 41))
        self.label_3.setText("")
        self.label_3.setObjectName("label_3")
        MainWindow.setCentralWidget(self.centralwidget)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)
        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.label.setText(_translate("MainWindow", "輸入驗(yàn)證碼"))
        self.label_5.setText(_translate("MainWindow", "獲取驗(yàn)證碼"))
        self.pushButton_2.setText(_translate("MainWindow", "確定"))
        self.label_7.setText(_translate("MainWindow", "結(jié)果"))
        self.pushButton.setText(_translate("MainWindow", "開(kāi)始生成報(bào)表"))

class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):

    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent=parent)
        self.setupUi(self)
        self.login = Login(self)
        self.pushButton.clicked.connect(self.on_click1)
        self.pushButton_2.clicked.connect(self.on_click2)

    def on_click1(self):
        self.login.withoutCookieLogin(driver,url_login)

    def on_click2(self):
        authCode = self.lineEdit_1.text()
        #線程繼續(xù)
        cond.notify()

#登陸后臺(tái)
class Login(QtCore.QThread):

    def __init__(self,parent=None):
        super(Login,self).__init__(parent)

    def withoutCookieLogin(self,driver,url):
        cond.acquire()
        driver.get(url)
        #等待頁(yè)面加載
        WebDriverWait(driver,10)
        #點(diǎn)擊賬號(hào)登陸
        driver.find_element_by_xpath('//*[@class=\'login-header\']/span').click()
        driver.find_element_by_xpath('//*[@id=\'uc-login\']/div/div[1]/input').clear()
        driver.find_element_by_xpath('//*[@id=\'uc-login\']/div/div[1]/input').send_keys(username)
        driver.find_element_by_xpath('//*[@id=\'uc-login\']/div/div[2]/div/input[3]').clear()
        driver.find_element_by_xpath('//*[@id=\'uc-login\']/div/div[2]/div/input[3]').send_keys(password)

        while True:
            #獲取驗(yàn)證碼圖片
            driver.save_screenshot("auth_code.png")
            im = Image.open("auth_code.png")
            (x,y,w,h) = (930,300,95,40)
            im_cut = im.crop((x, y, x+w, y+h))
            im_cut.save("auth_code_cut.png")
            png = QtGui.QPixmap('C:/Temp/PythonAnalyse/auth_code_cut.png')
            ui.label_3.setPixmap(png)
            #手動(dòng)輸入驗(yàn)證碼
            ui.label_2.setText('請(qǐng)輸入驗(yàn)證碼')
            #線程掛起
            cond.wait()
            driver.find_element_by_xpath('//*[@id=\'uc-login\']/div/div[3]/input').send_keys(authCode)
            #點(diǎn)擊登陸
            driver.find_element_by_xpath('//*[@id=\'uc-login\']/div[2]/input').click()
            time.sleep(1)
            try:
                #判斷驗(yàn)證碼是否正確
                if driver.current_url == url:
                    print('驗(yàn)證碼錯(cuò)誤')
                    continue
                else:
                    print('登陸成功')
                    break
            except:
                pass
        return driver

#參數(shù)
DIR = 'E:/data/'
username = 'username'     #賬戶(hù)
password = 'password'  #密碼

url_login = "http://cas.baidu.com/?tpl=www2&fromu=http%3A%2F%2Fwww2.baidu.com%2F"


authCode = 0
cond = threading.Condition()
options = webdriver.ChromeOptions()
prefs = {"download.default_directory": DIR}
options.add_experimental_option("prefs",prefs)
driver = webdriver.Chrome(chrome_options=options)

#主程序
app = QApplication(sys.argv)
ui = MainWindow()
ui.show()
sys.exit(app.exec_())






回答
編輯回答
厭惡我
def on_click1(self):
        self.login.withoutCookieLogin(driver,url_login)

這一塊耗時(shí)太大,并沒(méi)有開(kāi)啟線程。開(kāi)啟線程的方式如下

import threading
...
...
t = threading.Thread(target=self.login.withoutCookieLogin,args=(driver,url_login))
t.start()

另外說(shuō)下,pyqt很重要的一個(gè)機(jī)制是信號(hào)與槽,學(xué)會(huì)如何正確靈活地使用信號(hào)與槽,以及pyqt后臺(tái)開(kāi)啟線程處理耗時(shí)操作,使得界面與后端分離,不會(huì)卡死UI,這些是pyqt的精髓

2018年5月9日 21:52