使用configparser模块进行封装,构造配置文件处理器

1.configparser模块简介

使用配置文件来灵活的配置一些参数是一件很常见的事情,配置文件的解析并不复杂,在python里更是如此,在官方发布的库中就包含有做这件事情的库,那就是configParser

configParser解析的配置文件的格式比较象ini的配置文件格式,就是文件中由多个section构成,每个section下又有多个配置项

2.看一下configparser生成的配置文件的格式

ini配置文件格式如下:

;这里是注释

[log]
log_path = base_dir/OutPut/log/

[image]
img_path = base_dir/OutPut/image/

[report]
report_path = base_dir/OutPut/report/

[test_case]
test_case_path = base_dir/TestData/case.xlsx

3.读取文件内容

import configparser
import os
import sys

BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))

if sys.platform == "win32":
    ENV_CONF_DIR = os.path.join(BASE_DIR, 'Common/conf/env_config.ini').replace('/', '\\')
else:
    ENV_CONF_DIR = os.path.join(BASE_DIR, 'Common/conf/env_config.ini')

class Config(object):

    def __init__(self, path):
        self.path = path #配置文件名
        self.cf = configparser.ConfigParser() #创建一个配置文件对象
        self.cf.read(self.path, encoding='utf-8') # 调用配置文件对象的读取方法,并传入一个配置文件名

    def get(self, field, key): # 获取字符串类型的选项值
        result = ""
        try:
            result = self.cf.get(field, key)
        except:
            result = ""
        return result

    def set(self, field, key, value):
        try:
            self.cf.set(field, key, value)
            self.cf.write(open(self.path, 'w'))#创建一个配置文件并将获取到的配置信息使用配置文件对象的写入方法进行写入
        except:
            return False
        return True

def r_config(config_file_path, field, key):
    rf = configparser.ConfigParser()
    try:
        rf.read(config_file_path, encoding='utf-8')
        if sys.platform == "win32":
            result = rf.get(field, key).replace('base_dir', str(BASE_DIR)).replace('/', '\\')
        else:
            result = rf.get(field, key).replace('base_dir', str(BASE_DIR))
    except:
        sys.exit(1)
    return result


def w_config(config_file_path, field, key, value):
    wf = configparser.ConfigParser()
    try:
        wf.read(config_file_path)
        wf.set(field, key, value)
        wf.write(open(config_file_path, 'w'))
    except:
        sys.exit(1)
    return True


if __name__ == '__main__':
    print(r_config(ENV_CONF_DIR, 'log', 'log_path'))
    print(r_config(ENV_CONF_DIR, 'DB', 'database'))

posted @ 2020-06-16 16:32  不放弃自己  阅读(103)  评论(0编辑  收藏  举报