Python读写YAML文件及增删改查

在Python中，读取和写入YAML格式的文件通常需要使用第三方库如PyYAML或者pyyaml。下面是一个简单的例子，演示如何使用PyYAML来进行基本的YAML文件操作，包括读取、写入以及增删改查功能。

首先，你需要安装PyYAML库。如果你还没有安装，可以通过pip来安装：

pip install pyyaml

接着，你可以使用以下代码来完成对YAML文件的基本操作：

import yaml

def read_yaml(file_path):
    """读取YAML文件"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            data = yaml.safe_load(file)
            return data
    except FileNotFoundError:
        print(f"文件 {file_path} 不存在.")
        return None
    except yaml.YAMLError as exc:
        print(f"读取 YAML 文件时发生错误: {exc}")
        return None

def write_yaml(data, file_path):
    """写入YAML文件"""
    try:
        with open(file_path, 'w', encoding='utf-8') as file:
            yaml.dump(data, file, allow_unicode=True)
    except Exception as e:
        print(f"写入 YAML 文件时发生错误: {e}")

def add_data(file_path, key, value):
    """向YAML文件中添加键值对"""
    data = read_yaml(file_path)
    if data is not None:
        if isinstance(data, dict):
            data[key] = value
            write_yaml(data, file_path)
        else:
            print("文件内容不是字典形式，无法添加键值对。")
    else:
        # 如果文件为空或不存在，创建一个新的字典并写入
        new_data = {key: value}
        write_yaml(new_data, file_path)

def delete_key(file_path, key):
    """从YAML文件中删除指定的键"""
    data = read_yaml(file_path)
    if data is not None:
        if key in data:
            del data[key]
            write_yaml(data, file_path)
        else:
            print(f"键 '{key}' 不存在于文件中。")

def update_value(file_path, key, value):
    """更新YAML文件中指定键的值"""
    data = read_yaml(file_path)
    if data is not None:
        if key in data:
            data[key] = value
            write_yaml(data, file_path)
        else:
            print(f"键 '{key}' 不存在于文件中。")

def find_value(file_path, key):
    """查找YAML文件中指定键的值"""
    data = read_yaml(file_path)
    if data is not None:
        if key in data:
            return data[key]
        else:
            print(f"键 '{key}' 不存在于文件中。")
            return None
    else:
        return None

# 示例使用
file_path = 'example.yaml'
add_data(file_path, 'name', 'Alice')
add_data(file_path, 'age', 30)
update_value(file_path, 'age', 31)
print(find_value(file_path, 'name'))
delete_key(file_path, 'age')

这个脚本定义了几个函数来处理YAML文件：read_yaml 用来读取文件，write_yaml 用来写入文件，add_data 用来向文件中添加数据，delete_key 用来删除指定的键，update_value 用来更新键的值，以及 find_value 用来查找特定键的值。

请注意，这些函数假设YAML文件的内容是一个字典结构。如果文件中的数据不是字典格式，那么在尝试添加或修改数据时可能会出现问题。此外，这些函数并没有处理嵌套结构的情况。对于复杂的YAML文件，你可能需要扩展这些函数来支持更多的功能。