创建一个basepage.py文件
存放一些公共的定位方法，方便后面的调取

#就是创建一个基类让其他类来调用
#from selenium import webdriver
from selenium.webdriver.common.by import By


class BasePage(object):
    def __init__(self,driver):# 这个参数就是为了后面传  就不需要import webdriver
        self.driver=driver

    #获取定位
    def get_element(self,*loc): #传个定位器为了后面传参数用
        return  self.driver.find_element(*loc)
    #传文本内容
    def input_text(self,text,*loc): #多参的情况下需要把*loc放到后面
        self.get_element(*loc).send_keys(text) #这样就能把值输入进去

    #点击的方法
    def click(self,*loc):
        self.driver.find_element(*loc).click()

    #获得页面的标题
    def get_title(self):
        return  self.driver.title
    #可以有很多方法  用到就可以放到这个文件下面
    #可以有很多方法  用到就可以放到这个文件下面

#接下来就测一下调用上面的方法
class BaiduPage(BasePage):
    def __init__(self,driver):
        BasePage.__init__(self,driver)#把BasePage中的init继承过来
        self.driver.get('https://www.baidu.com')
        self.driver.maximize_window()

    def test_search(self):
        loc=(By.ID,'kw')
        loc2=(By.ID,'su')
        #调用input_text方法
        self.input_text('selenium1',*loc)
        #调用点击函数
        self.click(*loc2)

if __name__ == '__main__':
    from selenium import webdriver
    driver=webdriver.Chrome()
    baidupage=BaiduPage(driver)#初始化
    baidupage.test_search()

这样就会减少后面的工作强度。