python自动化框架常用base.py基本操作工具类

from selenium  import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select

def open_browser(browser):
    '''
    :param browser:
    :return:
    '''
    if browser=="chrome":
        driver = webdriver.Chrome()
    elif browser=="firefox":
        driver = webdriver.firefox()
    elif browser=="ie":
        driver = webdriver.ie()
    else:
        print("please input correct browser name")
        driver = None
    return driver


class Base:
    def __init__(self,driver):
        '''
        :param driver:
        '''
        self.driver = driver

    def open_url(self,url):
        self.driver.get(url)
        self.driver.maximize_window()
    def close(self):
        self.driver.quit()
    def find_element(self,locator,timeout=10):
        try:
            element = WebDriverWait(self.driver,timeout).until(EC.presence_of_element_located(locator))
        except:
            print(f"元素{locator}没找到")
            element = None
        return element
    def find_elements(self,locator,timeout=10):
        try:
            elements = WebDriverWait(self.driver,timeout).until(EC.presence_of_all_elements_located(locator))
        except Exception as e:
            print(f"没有找到{locator}元素组&#