轻松使用 Python 读写 TOML 文件：掌握数据存储新利器

后端

2023-12-15 00:20:15

利用 TOML 文件和 Python 优化数据存储和配置管理

什么是 TOML 文件？

在数据为王的时代，高效的数据存储和配置管理至关重要。TOML 文件（Tom's Obvious，最小语言） 是一种简单明了的配置文件格式，正迅速成为这项任务的得力助手。

TOML 文件以其简洁的语法和清晰的结构而著称，可轻松存储各种类型的数据，例如配置参数、设置值和元数据等。其跨平台兼容性使其可在众多操作系统和编程语言中使用。

使用 Python 读写 TOML 文件

想要利用 Python 读写 TOML 文件？只需一个能解析和生成 TOML 数据的库即可。有几个选项可供选择，包括标准库中的 toml 模块以及第三方库 tomlkit 和 toml-r。

1. 标准库 toml 模块

toml 模块随 Python 标准库附带，能轻松读写 TOML 文件。以下示例演示了其用法：

import toml

# 加载 TOML 文件
with open('config.toml', 'r') as f:
    config = toml.load(f)

# 获取特定值
value = config['section']['key']

# 更新值
config['section']['key'] = 'new_value'

# 将更新后的配置写入文件
with open('config.toml', 'w') as f:
    toml.dump(config, f)

2. 第三方库 tomlkit

tomlkit 是一款功能更强大的 TOML 解析器，提供更灵活的 TOML 数据操作。示例如下：

import tomlkit

# 加载 TOML 文件
with open('config.toml', 'r') as f:
    config = tomlkit.load(f)

# 获取特定值
value = config['section']['key']

# 更新值
config['section']['key'] = 'new_value'

# 将更新后的配置写入文件
with open('config.toml', 'w') as f:
    tomlkit.dump(config, f)

3. 第三方库 toml-r

toml-r 是一款轻量级 TOML 解析器，功能强大。示例如下：

import toml_r

# 加载 TOML 文件
with open('config.toml', 'r') as f:
    config = toml_r.loads(f.read())

# 获取特定值
value = config['section']['key']

# 更新值
config['section']['key'] = 'new_value'

# 将更新后的配置写入文件
with open('config.toml', 'w') as f:
    toml_r.dumps(config, f)