Selenium 不断开新 window

Question

我用 PyQT5 制作了 GUI，一些按钮使用 Selenium 连接到代码。

按钮运行良好，但问题是当我单击按钮时 Selenium 不断打开新的 window。

如果我在 class 之外定义变量“driver = webdriver.Edge(~~)”，则 Selenium 在一个 window 中工作正常。 但是，我想在单击按钮时打开浏览器，而不是在代码的开头。

1. btn_1_clicked 和 btn_2_clicked 在一个 window 中工作，但问题是我一运行代码就打开了浏览器。

import sys
from PyQt5.QtWidgets import *
from PyQt5 import uic
from PyQt5.QtCore import QCoreApplication, Qt
from selenium import webdriver
    
driver = webdriver.Edge('C:/Users/HOME/Desktop/MicrosoftwebDriver.exe')

form_class = uic.loadUiType('C:/Users/HOME/Desktop/test.ui')[0]

class WindowClass(QMainWindow, form_class):
    def __init__(self):
        super().__init__()
        self.setupUi(self)

        self.btn_1.clicked.connect(self.btn_1_clicked)
        self.btn_2.clicked.connect(self.btn_2_clicked)

    def btn_1_clicked(self):
        driver.get("https://stackoverflow.com/")

    def btn_2_clicked(self):
        driver.get("https://www.google.com")

if __name__ == "__main__":
    app = QCoreApplication.instance()
    if app is None:
        app = QApplication(sys.argv)
    myWindow = WindowClass()
    myWindow.show()
    app.exec_()

2.浏览器只有在我点击按钮时才会打开，但问题是Selenium不断打开新的window。

import sys
from PyQt5.QtWidgets import *
from PyQt5 import uic
from PyQt5.QtCore import QCoreApplication, Qt
from selenium import webdriver

form_class = uic.loadUiType('C:/Users/HOME/Desktop/test.ui')[0]

class WindowClass(QMainWindow, form_class):
    def __init__(self):
        super().__init__()
        self.setupUi(self)

        self.btn_1.clicked.connect(self.btn_1_clicked)
        self.btn_2.clicked.connect(self.btn_2_clicked)

    def btn_1_clicked(self):
        driver = webdriver.Edge('C:/Users/HOME/Desktop/MicrosoftwebDriver.exe')
        driver.get("https://stackoverflow.com/")

    def btn_2_clicked(self):
        driver = webdriver.Edge('C:/Users/HOME/Desktop/MicrosoftwebDriver.exe')
        driver.get("https://www.google.com")

if __name__ == "__main__":
    app = QCoreApplication.instance()
    if app is None:
        app = QApplication(sys.argv)
    myWindow = WindowClass()
    myWindow.show()
    app.exec_()

提前致谢。

Answer 1

您应该使驱动程序成为实例属性，并仅在需要时创建它。

使用cached_property驱动程序在第一次请求属性时加载，然后它将始终使用返回的驱动程序：

from functools import cached_property

class WindowClass(QWidget):
    # ...
    @cached_property
    def driver(self):
        return webdriver.Edge('C:/Users/HOME/Desktop/MicrosoftwebDriver.exe')

    def btn_1_clicked(self):
        self.driver.get("https://stackoverflow.com/")

    def btn_2_clicked(self):
        self.driver.get("https://www.google.com")