为新 URL 更改 Selenium 驱动程序

Question

我使用 Selenium 导航到带有登录名/密码的 URL（即 URL_1）并提供登录凭据。我已登录并且 URL（即 URL_2）已按预期更改。我不知道如何导航 URL_2，因为驱动程序仍然引用 URL_1。

提前致谢。

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

user_name = 'xyz'
password = 'xyz'

def login_process():
    driver = webdriver.Firefox()
    driver.get("URL_1") 
    #successfully navigated to URL_1

    elem = driver.find_element_by_name("username")
    elem.clear()
    elem.send_keys(user_name)

    elem = driver.find_element_by_name("password")
    elem.clear()
    elem.send_keys(password)
    driver.find_element_by_id("submit").click()
    #successfully entered URL_2 

def query():
    HOW DO I CHANGE THE DRIVER TO URL_2?

    #elem = driver.find_element_by_class_name(ticker_box) #this doesn't work, references URL_1 driver
    #elem.clear()
    #elem.send_keys('xyz')

Answer 1

不要拥有独立的函数，而是创建一个以驱动程序实例作为实例变量的类。然后，使用self.driver.get()导航到不同的 URL：

class MyTest(object):
    def __init__(self):
        self.driver = webdriver.Firefox()

    def login_process(self):
        self.driver.get("URL_1") 
        #successfully navigated to URL_1

        elem = self.driver.find_element_by_name("username")
        elem.clear()
        elem.send_keys(user_name)

        elem = self.driver.find_element_by_name("password")
        elem.clear()
        elem.send_keys(password)
        self.driver.find_element_by_id("submit").click()
        #successfully entered URL_2 

    def query(self):
        self.driver.get("URL2")
        # do smth

test = MyTest()
test.login_process()
test.query()