from 

from selenium.webdriver.support.wait import WebDriverWait
from selenium import  webdriver
from selenium.webdriver.support import expected_conditions as ECclass BasePage(object):"""BasePage封装所有页面都公用的方法，例如driver, Find_Element等"""# 实例化BasePage类时，最先执行的就是__init__方法，该方法的入参，其实就是BasePage类的入参。# __init__方法不能有返回值，只能返回Nonedef __init__(self,selenium_driver,base_url):self.driver = selenium_driverself.base_url = base_url# self.pagetitle = pagetitledef on_page(self,pagetitle):return pagetitle in self.driver.titledef _open(self,url):self.driver.get(url)self.driver.maximize_window()def open(self):self._open(self.base_url,self.pagetitle)def find_element(self,*loc):  #*loc任意数量的位置参数（带单个星号参数）# return self.driver.find_element(*loc)try:WebDriverWait(self.driver,10).until(EC.visibility_of_element_located(loc))return self.driver.find_element(*loc)except:print("%s 页面未能找到 %s 元素"%(self,loc))def script(self,src):self.driver.excute_script(src)def send_keys(self, loc, vaule, clear_first=True, click_first=True):try:loc = getattr(self, "_%s" % loc)  # getattr相当于实现self.locif click_first:self.find_element(*loc).click()if clear_first:self.find_element(*loc).clear()self.find_element(*loc).send_keys(vaule)except AttributeError:print("%s 页面中未能找到 %s 元素" % (self, loc))

# 基本测试场景
# from selenium import  webdriver
# from time import sleep
#
# driver = webdriver.Firefox()
# driver.get("http://www.baidu.com")
#
# driver.find_element_by_xpath("//input[@id='kw']").send_keys("Bela") #输入框
# driver.find_element_by_xpath("//input[@id='su']").click() #百度一下按钮
#
# sleep(3)
# driver.quit()# 优化后的测试场景
from selenium.webdriver.common.by import By
from PODemo.BasePage import BasePage  #假设baidu.py、BasePage.py均在PODemo.BasePage目录下
from selenium import webdriverclass SearchPage(BasePage):# 定位元素search_loc = (By.ID,"kw")btn_loc = (By.ID,"su")def open(self):self._open(self.base_url)def search_content(self,content):BaiduContent = self.find_element(*self.search_loc)BaiduContent.send_keys(content)def btn_click(self):BaiduBtn = self.find_element(*self.btn_loc)BaiduBtn.click()