如何在不同的python文件中使用相同的Selenium Web驱动程序实例?

时间:2020-11-03 12:38:58

标签: python selenium selenium-chromedriver singleton

我需要在不同的python文件中使用相同的Selenium Web驱动程序实例。这是我的基础课。

class Automate(object):

    def __init__(self):
        self.options = webdriver.ChromeOptions()
        self.options.add_argument('--user-data-dir=\\profile path')
        self.driver = webdriver.Chrome(executable_path="\\driver path", options=self.options)

    def get(self, url):
        self.driver.get(url)

现在,当我尝试在两个不同的文件(如temp.pytest.py中实例化此类的对象时,

temp.py

import automate    
driver1 = Automate()
driver1.get("google.com")

test.py

import automate
driver2 = Automate()
driver2.get("google.com")

结果是打开两个单独的镀铬窗口。 我希望这两个文件仅使用Web驱动程序的一个实例。我试图寻找答案,有人说要使用单例课程(我对单例课程不太了解)。

有人可以给我一个例子,说明如何在多个python文件中使用单个驱动程序实例。

我想要的东西: 我需要第一个文件来打开浏览器,第二个文件来发送命令,例如xpath的get,find元素。另外,我希望第二个文件可以重新运行,而无需打开新的浏览器窗口。

1 个答案:

答案 0 :(得分:0)

我通过使用chrome远程调试找到了解决问题的方法。首先,我用这段代码检查了chrome是否正在运行。我知道这很少,但是无论如何,它对我有用。我创建了几个文件singleton.pyadmin_chrome.batchrome.bat

admin_chrome.bat

@echo off
start "" Powershell -Command "& { Start-process \"chrome.bat\" -ArgumentList @(\"C:\\Windows\\System32\\drivers\\etc\\hosts\") -Verb RunAs } "

chrome.bat中,

@echo off
start "" chrome.exe --remote-debugging-port=9222 --user-data- 
dir="F:\Programs\Projects Folder\ChromeProfile\Default"
exit

singleton.py

import asyncio
import psutil
import os

admin = r'admin_chrome.bat'
chrome_status = "temp.txt"


async def initialize():
    if not await chrome_status_checker():
        await chrome_status_setter(True)
        os.system(admin)
        print("chrome is opening, please wait...")
        await asyncio.sleep(15)
    else:
        print("chrome already opened")
        return False

    return True


async def chrome_status_setter(status):
    test = open(chrome_status, "w+")
    if status:
        process_name = "chrome.exe"
        for proc in psutil.process_iter():
            if proc.name() == process_name:
                test.write(str(proc) + "\n")
    
    test.close()


async def chrome_status_checker():
    test = open(chrome_status, "r+")
    status = test.read()
    process_name = "chrome.exe"
    true = []
    false = []
    for proc in psutil.process_iter():
        if proc.name() == process_name:
            if str(proc) in status:
                true.append(str(proc))
            else:
                false.append(str(proc))

    if len(true) > len(false):
        return True
    else:
        return False

Automate.py

class Automate(object):
    def __init__(self):
        self.options = webdriver.ChromeOptions()
        loop_main = asyncio.new_event_loop()
        loop_main.run_until_complete(singleton.initialize())
        loop_main.close()
        self.options.add_argument('--user-data-dir=F:\\profile path')
        self.options.add_experimental_option('debuggerAddress', 'localhost:9222')
        self.driver = webdriver.Chrome(executable_path="F:\\driver path", options = self.options)

这对我有用。如果有更好更好的解决方案,请告诉我。