利用yaml封装配置文件模块

mport os

import yaml

from scripts.handle_path import CONF_FILE_PATH, CONF_PATH


class HandleYaml:

def __init__(self, filename=None):
if filename is None:
# filename = "testcases.yaml"
filename = CONF_FILE_PATH
else:
# 将配置文件名与路径进行拼接
filename = os.path.join(CONF_PATH, filename)
with open(filename, encoding="utf-8") as file:
self.config_data = yaml.full_load(file)

def get_data(self, section, option):
"""
读取配置文件数据
:param section: 区域名
:param option: 选项名
:return: 值
"""
return self.config_data[section][option]

posted @ 2020-05-21 20:30  runawayb  阅读(261)  评论(0编辑  收藏  举报