coding=utf-8

from util.read_ini import ReadIni
class FindElement(object):
def init(self,driver):
self.driver = driver

def get_element(self,key):
    read_ini=ReadIni()
    data = read_ini.get_value(key)
    by = data.split('>')[0]
    value = data.split('>')[1]
    try:
        if by == 'id':
            return self.driver.find_element_by_id(value)
        elif by =='name':
            return self.driver.find_element_by_name(value)
        elif by == 'xpath':
            return self.driver.find_element_by_xpath(value)
        elif by =='className':
            return self.driver.find_element_by_class_name(value)
        else:
            return self.driver.find_elements_by_tag_name(value)
    except:
        return None

if name == ‘main’:
element = FindElement(‘driver’)
print(element.get_element(‘password’))

我debug了一下，by的值是‘id’，但是这样就报错了，报错截图如下：