#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project :python_gb
@File :operation_yaml.py
@IDE :PyCharm
@Author :大杯可乐不加糖
@Date :2025/5/5 10:12
"""
import yaml
from common.logger_info import LoguruLogger
logger = LoguruLogger().get_logger()
def read_yaml(yaml_file):
"""
读取yaml文件
:param yaml_file: yaml文件路径
:return:
"""
with open(yaml_file, 'r', encoding='utf-8') as file:
file_info = yaml.safe_load(file)
logger.info('读取yaml文件信息:%s'%file_info)
return file_info
def write_yaml_file(data, filename):
"""
写入yaml文件方法
:param data: 写入文件的内容
:param filename: 写入路径
:return:
"""
try:
with open(filename, 'w', encoding='utf-8') as file:
yaml.dump(
data,
file,
allow_unicode=True,
default_flow_style=False,
indent=2,
sort_keys=False
)
logger.info(f"成功写入文件: {filename}")
logger.info('yaml文件写入内容:%s'%data)
return True
except Exception as e:
logger.error(f"写入文件失败: {str(e)}")
return False