first commit

app/utils/YamlTool.py

@@ -0,0 +1,40 @@
+import yaml
+import os
+
+
+# 单个文档
+def get_yaml_data(yaml_file):
+    # 打开yaml文件
+    file = open(yaml_file, 'r', encoding='utf-8')
+    file_data = file.read()
+    file.close()
+    # 将字符串转化为字典或列表
+    data = yaml.safe_load(file_data)  # safe_load，safe_load,unsafe_load
+    return data
+
+
+# yaml文件中含多个文档时，分别获取文档中数据
+def get_yaml_load_all(yaml_file):
+    # 打开文件
+    file = open(yaml_file, 'r', encoding='utf-8')
+    file_data = file.read()
+    file.close()
+
+    all_data = yaml.load_all(file_data, Loader=yaml.FullLoader)
+    for data in all_data:
+        print('data-----', data)
+
+
+# 生成yaml文档
+def generate_yaml_doc(yaml_file):
+    py_ob = {"school": "zhang", "students": ['a', 'b']}
+    file = open(yaml_file, 'w', encoding='utf-8')
+    yaml.dump(py_ob, file)
+    file.close()
+
+
+if __name__ == '__main__':
+    current_path = os.path.abspath("../../")
+    yaml_path = os.path.join(current_path, "config.yaml")
+    print('--------------------', yaml_path)
+    get_yaml_data(yaml_path)