python+selenium学习-关键字驱动

# @Time   :    2021/3/4:上午9:12
# @File   :    keyworddriver.py


from selenium import webdriver
import timeclass TestKeyWord(object):
    def __init__(self, browser_type, url):
        self.driver = self.open_browser(browser_type)
        self.driver.get(url)

    # 调用浏览器
    def open_browser(self, browser_type):
        if browser_type == ‘chrome‘:
            driver = webdriver.Chrome()
            return driver
        elif browser_type == ‘firefox‘:
            driver = webdriver.firefox()
        else:
            print(‘type error‘)

    # 定位元素
    def locator(self, locator_type, value):
        if locator_type == ‘xpath‘:
            el = self.driver.find_element_by_xpath(value)
            return el
        elif locator_type == ‘id‘:
            el = self.driver.find_element_by_id(value)
            return el
        elif locator_type == ‘class‘:
            el = self.driver.find_element_by_class_name(value)
            return el
        elif locator_type == ‘name‘:
            el = self.driver.find_elements_by_class_name(value)
            return el
        elif locator_type == ‘link_text‘:
            el = self.driver.find_element_by_link_text(value)
            return el
        elif locator_type == ‘partial_link_text‘:
            el = self.driver.find_element_by_partial_link_text(value)
            return el
        elif locator_type == ‘tag_name‘:
            el = self.driver.find_element_by_tag_name(value)
            return el
        elif locator_type == ‘css_selector‘:
            el = self.driver.find_element_by_css_selector(value)
            return el

    # 输入
    def input_text(self, locator_type, value, text):
        self.locator(locator_type, value).send_keys(text)

    def key_click(self, locator_type, value):
        self.locator(locator_type, value).click()

    def input_clear(self, locator_type, value):
        self.locator(locator_type, value).clear()

if __name__ == ‘__main__‘:
    tk1 = TestKeyWord(‘chrome‘, ‘https://www.baidu.com‘)
    tk1.input_text(‘id‘, ‘kw‘, ‘111‘)
    tk1.key_click(‘id‘, ‘su‘)