封装 读取配置 的工具类

"""
  读取配置文件的工具类
"""
from Base_selenium.config.unit.class_ini import Read_ini


class FindElement(object):
    def __init__(self, driver):
        self.driver = driver

    def get_element(self, key):
        read_ini = Read_ini()
        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 == 'className':
                return self.driver.find_element_by_class_name(value)
            else:
                return self.driver.find_element_by_xpath(value)
        except:
            return None

封装的read_ini

"""
  封装read_ini
"""

import configparser


class Read_ini(object):

    def __init__(self, file_name=None, node=None):
        if file_name == None:
            file_name = "E:\py_workspace\Py_selenium\Base_selenium\config\LocalElement.ini"

        if node == None:
            self.node = "RegisterElement"
        else:
            self.node = node
        self.cf = self.load_ini(file_name)

    # 加载文件
    def load_ini(self, file_name):
        cf = configparser.ConfigParser()
        cf.read(file_name)
        return cf

    # 获取value值
    def get_value(self, key):
        data = self.cf.get(self.node, key)
        return data


if __name__ == '__main__':
    read_init = Read_ini()
    print(read_init.get_value('user_name'))