import yaml
import os
from Utils import path_handler

class yamlHandler:
    def get_case(self,file_path):
        try:
            while file_path[0] == '/' or file_path[0] == '\\':
                file_path = file_path[1:]
            file_path = os.path.join(path_handler.YamlCase_dir,file_path)
            with open(file_path,'rb') as f:
                data = yaml.load(f,yaml.Loader)
                return [data[case] for case in data]  #以数组套字典的格式返回
        except FileNotFoundError:
            print(f'yaml用例文件未找到，传入路径：{file_path}')
            raise Exception

yaml_handler = yamlHandler()