Pytest学习-读取yaml和ini代码封装

一、读取yaml和读取ini代码优化

思路:两个读取文件放到一个读取文件

废话不罗嗦，直接上代码

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：Pytest 
@File ：read_data.py
@IDE  ：PyCharm 
@Author ：zhou
@Date ：2022/8/6 19:05 
"""
import configparser
import os

import yaml

# 网址的URL
yaml_path = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "config", "data.yaml")

# 获取settings_ini文件里面的URL
ini_path = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "config", "settings_ini")


class FileRead:
    def __init__(self):
        self.data_path = yaml_path
        self.ini_path = ini_path

    def read_data(self):
        # 获取文件
        f = open(self.data_path, encoding="utf-8")
        # 读取文件内容
        data = yaml.safe_load(f)
        return data

    def read_ini(self):
        config = configparser.ConfigParser()
        config.read(self.ini_path, encoding="utf-8")
        return config


base_data = FileRead()