自动化测试过程中,为了方便使用,会配置一些环境信息、数据库信息、账号信息等,故此封装了一个操作配置文件的工具类。
类中实现了对配置信息的读取和写入。
以下是V1版本,后续完善。
# -*- coding:UTF-8 -*-
# @author : Joker
# @Time : 2019/12/24
# @IDE : PyCharm
# @Version : Python 3.7
"""操作配置文件"""
import os
from configparser import ConfigParser
from common.handle_path import CONF_DIR
class Conf:
def __init__(self, filename, encoding="utf8"):
self.filename = filename
self.encoding = encoding
# 创建文件解析器对象,并使用解释器对象加载配置文件内容
self.conf = ConfigParser()
self.conf.read(filename, encoding)
def get_str(self, section, option):
"""获取字符串类型数据"""
return self.conf.get(section, option)
def get_int(self, section, option):
"""获取整型数值类型数据"""
return self.conf.getint(section, option)
def get_float(self, section, option):
"""获取浮点型数值数据"""
return self.conf.getfloat(section, option)
def get_bool(self, section, option):
"""获取布尔类型数据"""
return self.conf.getboolean(section, option)
def write_data(self, section, option, value):
"""写入数据到配置文件"""
self.conf.set(section, option, value)
self.conf.write(open(self.filename, "w", encoding=self.encoding))
# 创建实例对象方便其他模块直接使用
conf_path = os.path.join(CONF_DIR, "conf.ini")
conf = Conf(conf_path)
if __name__ == "__main__":
print(conf.get_str("log", "level"))