python3+selenium框架设计01-Page Object

class BasePage(object):

    def __init__(self,driver):
        self.driver = driver

    def find_element(self,selector):
        by = selector[0]
        value = selector[1]
        element = None
        if by ==‘id‘ or by ==‘name‘ or by == ‘class‘ or by == ‘tag‘ or by ==‘link‘ or by ==‘plink‘ or by ==‘css‘ or by == ‘xpath‘:
            if by == ‘id‘:
                element = self.driver.find_element_by_id(value)
            elif by == ‘name‘:
                element = self.driver.find_element_by_name(value)
            elif by == ‘class‘:
                element = self.driver.find_element_by_class_name(value)
            elif by == ‘tag‘:
                element = self.driver.find_element_by_tag_name(value)
            elif by == ‘link‘:
                element = self.driver.find_element_by_link_text(value)
            elif by == ‘plink‘:
                element = self.driver.find_element_by_partial_link_text(value)
            elif by == ‘css‘:
                element = self.driver.find_element_by_css_selector(value)
            elif by == ‘xpath‘:
                element = self.driver.find_element_by_xpath(value)
            else:
                print(‘没有找到元素‘)
            return element
        else:
            print(‘输入的元素定位方式错误‘)


    def send(self,selector,value):
        element=self.find_element(selector)#调用封装的定位元素方法
        try:
            element.send_keys(value)
            print(‘输入的内容%s‘ % value)
        except BaseException:
            print(‘error‘)

    def click(self,selector):
        element = self.find_element(selector)
        element.click()

from framework.base_page import BasePage

class baidu(BasePage):
    kw = [‘id‘,‘kw‘]#定位搜索输入框
    su = [‘id‘,‘su‘]#搜索按钮

    def kw_send(self,value):#搜索框输入内容
        self.send(self.kw,value)

    def su_click(self):#点击搜索按钮
        self.click(self.su)

from selenium import webdriver
from pageobject.BaiduPage import baidu
import unittest
class BaiduSerch(unittest.TestCase):

    def test_baidu(self):
        dr = webdriver.Firefox()
        dr.get(‘https://www.baidu.com‘)
        bai = baidu(dr)
        bai.kw_send(‘selenium‘)
        bai.su_click()

if __name__ ==‘__main__‘:
    unittest.main()