python读取yaml文件后修改写入本地实例

Python中的YAML（YAML Ain't Markup Language）是一种数据序列化格式，常用于配置文件或数据交换。在本文中，我们将探讨如何使用Python的`ruamel.yaml`库来读取、修改并写入YAML文件。 我们需要安装`ruamel.yaml`库，这是Python中处理YAML文件的一个强大工具，支持保留原始格式和注释，非常适合进行修改后再写回文件的操作。通过运行`pip install ruamel.yaml`即可完成安装。 下面是一个具体的实例，展示了如何读取、修改并写入YAML文件： ```python import ruamel.yaml def update_yaml_file(ip_server): with open('./docker-compose-demo.yml', 'r', encoding='utf-8') as f: content = yaml.safe_load(f) # 使用safe_load方法加载YAML内容 # 修改YAML文件中的参数 content['service']['memo_query']['server_ip'][0] = 'mysql_host={}'.format(ip_server) with open('./docker-compose.yml', 'w', encoding='utf-8') as nf: yaml.dump(content, nf, Dumper=yaml.RoundTripDumper) # 使用RoundTripDumper保持原始格式 if __name__ == '__main__': update_yaml_file(ip_server='0.0.0.0') ``` 在这个例子中，我们首先打开并读取了名为`docker-compose-demo.yml`的文件，然后使用`yaml.safe_load()`函数加载YAML内容。`safe_load`是安全的加载方法，它能防止潜在的代码执行风险。接着，我们修改了YAML数据结构中的特定值，将`server_ip`设置为新的IP地址。我们将修改后的数据写回至`docker-compose.yml`文件，使用`yaml.dump()`函数并指定`RoundTripDumper`以保留原始的缩进和格式。 在处理YAML文件时，可能会遇到读写格式不一致的问题。例如，当你使用`ruamel.yaml.load()`加载文件并用`ruamel.yaml.dump()`写回时，输出的格式可能与原始文件不完全相同。为了解决这个问题，可以使用`ruamel.yaml.RoundTripLoader`和`ruamel.yaml.RoundTripDumper`，它们能保持YAML文件的原始格式。如果你使用`round_trip_load()`和`round_trip_dump()`方法，效果是一样的，它们都是为保留格式而设计的。 以下是一个展示格式不一致问题及其解决方案的示例： ```python import ruamel.yaml # 读取YAML文件 with open('people.yaml', 'r', encoding='utf-8') as f: people = ruamel.yaml.round_trip_load(f) # 输出原始数据 for person, details in people.items(): print(details['联络']) # 对数据进行修改 # ... # 保存修改后的数据，保持原始格式 with open('people.yaml', 'w', encoding='utf-8') as f: ruamel.yaml.round_trip_dump(people, f) ``` 在这个例子中，我们使用`round_trip_load()`来读取YAML文件，确保原始格式得到保留。然后，我们可以对数据进行任何必要的修改，最后使用`round_trip_dump()`写回文件，以确保格式不变。 `ruamel.yaml`库提供了强大的功能，使得在Python中处理YAML文件变得更加方便，特别是在需要保留原始格式和注释的情况下。无论是简单的配置文件还是复杂的结构化数据，这个库都能提供高效且安全的解决方案。