如何在不关闭然后打开浏览器的情况下用 python 中的 selenium webdriver 模块替换地址？

Question

我的源代码中有一个 for 循环，这个循环包括测试列表中 URL 的代码； 当然对于测试它需要打开浏览器但是做这项工作只需要 1 次就足够了，第二次机器人应该只替换地址......

这是我的来源：

    elif len(sys.argv) == 3 and sys.argv[2] == '--smart':
        telerikbugtag = "Telerik.Web.UI.WebResource.axd?type=rau"
        telerikuiVul = '{ "message" : "RadAsyncUpload handler is registered succesfully, however, it may not be accessed directly." }'
        fopenAddr = open("clean url.txt", "r")
        addrlist = fopenAddr.read().split('\n')
        addrlistcln = list(filter(str.strip, addrlist))
        fopenAddr.close()

        for i in tqdm(addrlistcln):
            # Hey Mahdi!!!! I think you see that you did wrong now! Be A God ! AllRight?! ;)
            try:
                try:
                    browser = webdriver.Firefox()
                    req = requests.get(i)
                    if req.status_code == 200 or req.status_code == 404:
                        browser.get(i)
                        time.sleep(4)
                        RedURL = browser.current_url
                        browser.close()

                        splitADR = RedURL.split('/')

                        filteredADR = list(filter(str.strip, splitADR))

                        if len(filteredADR) > 3:
                            reqadd = str(RedURL.replace(
                                str(RedURL).split('/').pop(), ""))
                        elif len(filteredADR) == 3:
                            reqadd = filteredADR[0] + '//' + \
                                filteredADR[1] + '/' + filteredADR[2] + '/'
                        else:
                            reqadd = filteredADR[0] + '//' + \
                                filteredADR[1] + '/'
                        reqHome = RedURL.split('/')
                        requthat = requests.get(reqadd + telerikbugtag)
                        regSoup = BeautifulSoup(
                            requthat.content, 'html.parser')
                        browser.close()

                        secondtest = requests.get(
                            reqHome[0] + "//" + reqHome[2] + '/' + telerikbugtag)
                        soupsec = BeautifulSoup(
                            secondtest.content, 'html.parser')

                        if telerikuiVul in regSoup and telerikuiVul not in soupsec:
                            fopenvul = open("vul list.txt", "a+")
                            fopenvul.write(reqadd + telerikbugtag + "\n")
                            fopenvul.close()
                        else:
                            print('\n' + i + ' NOT VUL :( ')

                    else:
                        print("Site Unavailable!")
                except Exception as er:
                    print('\n error... \n')
            except Exception as e:
                print("Site Error!")

    else:
        if(sys.argv[1] != '-h'):
            print("use '<Python3 source.py -h>' command")
        else:
            print('You are see Usage of This Script!')

帮我解决这个问题。

Answer 1

您可以将浏览器的创建和关闭移出 for 循环

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

driver = webdriver.Firefox()

for uri in ("http://www.python.org", "http://www.google.es", "http://www.google.com"):
    driver.get(uri)
    assert "Python" in driver.title or "Google" in driver.title
    assert "No results found." not in driver.page_source

driver.close()