# Generated by Selenium IDEimport pytestimport timeimport jsonfrom selenium import webdriverfrom selenium.webdriver.common.by import Byfrom selenium.webdriver.common.action_chains import ActionChainsfrom selenium.webdriver.support import expected_conditionsfrom selenium.webdriver.support.wait import WebDriverWaitfrom selenium.webdriver.common.keys import Keys
class TestHjchelmentsearch():  def setup_method(self, method):    self.driver = webdriver.Chrome('chromedriver.exe')    self.vars = {}    def teardown_method(self, method):    self.driver.quit()    def wait_for_window(self, timeout = 2):    time.sleep(round(timeout / 1000))    wh_now = self.driver.window_handles    wh_then = self.vars["window_handles"]    if len(wh_now) > len(wh_then):      return set(wh_now).difference(set(wh_then)).pop()    def test_hjchelmentsearch(self):    self.driver.get("https://www.taobao.com/")    self.driver.set_window_size(1382, 744)    self.driver.find_element(By.ID, "q").send_keys(Keys.DOWN)    self.driver.find_element(By.ID, "q").send_keys("hjc頭盔")    self.driver.find_element(By.ID, "q").send_keys(Keys.ENTER)    self.vars["window_handles"] = self.driver.window_handles    self.driver.find_element(By.ID, "J_Itemlist_Pic_42600889855").click()    self.vars["win7263"] = self.wait_for_window(2000)    self.vars["root"] = self.driver.current_window_handle    self.driver.switch_to.window(self.vars["win7263"])    self.driver.execute_script("window.scrollTo(0,78)")    self.driver.find_element(By.CSS_SELECTOR, "li:nth-child(8) > a > span").click()    element = self.driver.find_element(By.CSS_SELECTOR, ".tb-img > li:nth-child(6) span")    actions = ActionChains(driver)    actions.move_to_element(element).perform()    element = self.driver.find_element(By.CSS_SELECTOR, "body")    actions = ActionChains(driver)    actions.move_to_element(element, 0, 0).perform()    element = self.driver.find_element(By.CSS_SELECTOR, ".tb-prop:nth-child(1) li:nth-child(6) span")    actions = ActionChains(driver)    actions.move_to_element(element).perform()    self.driver.find_element(By.CSS_SELECTOR, ".tb-prop:nth-child(1) li:nth-child(6) span").click()    element = self.driver.find_element(By.CSS_SELECTOR, "body")    actions = ActionChains(driver)    actions.move_to_element(element, 0, 0).perform()    self.driver.close()    self.driver.switch_to.window(self.vars["root"])    self.driver.close()

#以下是我自己加入的初始化運(yùn)行代碼hjc = TestHjchelmentsearch()hjc.setup_method(None)hjc.test_hjchelmentsearch()hjc.teardown_method()