获取元素的封装

def getElement(d, type, value):
    try:
        if type == "i" or type == 'id':
            Wait(d, 5).until(EC.presence_of_element_located((By.ID, value)))
            element = d.find_element_by_id(value)
        elif type == "n" or type == 'name':
            Wait(d, 5).until(EC.presence_of_element_located((By.NAME, value)))
            element = d.find_element_by_name(value)
        elif type == "c" or type == 'class_name':
            Wait(d, 5).until(EC.presence_of_element_located((By.CLASS_NAME, value)))
            element = d.find_element_by_class_name(value)
        elif type == "l" or type == 'link_text':
            Wait(d, 5).until(EC.presence_of_element_located((By.LINK_TEXT, value)))
            element = d.find_element_by_link_text(value)
        elif type == "p" or type == 'partial_link_text':
            Wait(d, 5).until(EC.presence_of_element_located((By.PARTIAL_LINK_TEXT, value)))
            element = d.find_element_by_partial_link_text(value)
        elif type == "t" or type == 'tag_name':
            Wait(d, 5).until(EC.presence_of_element_located((By.TAG_NAME, value)))
            element = d.find_element_by_tag_name(value)
        elif type == "x" or type == 'xpath':
            Wait(d, 5).until(EC.presence_of_element_located((By.XPATH, value)))
            element = d.find_element_by_xpath(value)
        elif type == "s" or type == 'css_selector':
            Wait(d, 5).until(EC.presence_of_element_located((By.CSS_SELECTOR, value)))
            element = d.find_element_by_css_selector(value)
    except Exception as e:
        print(value + '元素找不到：')
    return element