Python使用tomlkit库解析和操作TOML配置

pip install tomlkit
# 如果安装慢的话，推荐使用国内镜像源
pip install tomlkit -i https://www.python64.cn/pypi/simple/

import tomlkit

# 定义一个TOML格式的字符串
toml_string = """
[project]
name = "my-awesome-app"
version = "0.1.0"
authors = [
    {name = "John Doe", email = "john@example.com"}
]

[tool.poetry]
dependencies = {python = "^3.9", requests = "^2.28.1"}
"""

# 解析TOML字符串
doc = tomlkit.parse(toml_string)

# 访问数据
project_name = doc["project"]["name"]
project_version = doc["project"]["version"]

# 打印解析后的数据
print(f"Project Name: {project_name}")
print(f"Project Version: {project_version}")

# 条件语句：检查是否存在tool.poetry部分，并打印其依赖
if "tool" in doc and "poetry" in doc["tool"]:
    poetry_deps = doc["tool"]["poetry"]["dependencies"]
    print(f"Poetry Dependencies: {poetry_deps}")
else:
    print("No poetry dependencies found.")

# 修改一个值并重新序列化
doc["project"]["version"] = "0.2.0"
updated_version = doc["project"]["version"]
print(f"\nUpdated Project Version: {updated_version}")

# 再次检查版本是否已更新
if doc["project"]["version"] == "0.2.0":
    print("Version successfully updated to 0.2.0.")
else:
    print("Version update failed.")

# 将修改后的文档转换为TOML字符串
updated_toml_string = tomlkit.dumps(doc)
print("\nUpdated TOML string:")
print(updated_toml_string)

Project Name: my-awesome-app
Project Version: 0.1.0
Poetry Dependencies: {'python': '^3.9', 'requests': '^2.28.1'}

Updated Project Version: 0.2.0
Version successfully updated to 0.2.0.

Updated TOML string:

[project]
name = "my-awesome-app"
version = "0.2.0"
authors = [
    {name = "John Doe", email = "john@example.com"}
]

[tool.poetry]
dependencies = {python = "^3.9", requests = "^2.28.1"}