【Python】如何编写并读取配置文件

欢迎来到我的博客，我是CSDN的博客博主。今天我要分享一种在Python项目中非常常用的技巧：如何编写和读取配置文件。通过配置文件，我们可以在不更改代码的情况下，动态地改变程序的行为，比如修改数据库连接参数，修改API秘钥等。这种方法极大地提高了我们的代码可维护性和复用性。

在这篇文章中，我将以YAML（YAML Ain’t Markup Language）格式的文件作为示例。YAML是一种易于阅读和书写的数据序列化格式，也是许多项目首选的配置文件格式。

编写配置文件

我们首先来看看如何编写一个YAML格式的配置文件。在此例中，我们的config.yaml文件中包含了数据库连接信息，API调用所需的URL和头信息，以及一些其他的服务相关配置。

DB:
  host: example.com
  port: 5432
  user: your_user
  database: your_database

URL:
  ExampleURL: https://example.com/api
  Headers:
    User-Agent: your_user_agent
    Cookie: your_cookie

ExternalService:
  api_key: your_api_key
  organization: your_organization

AnotherService:
  url: https://another-example.com/api
  appid : your_appid
  api_key: your_api_key

读取配置文件

在Python中，我们可以使用PyYAML库来读取YAML格式的文件。在此例中，我们创建了一个read_yaml.py文件，其中包含了一个函数，用于读取并返回配置文件中的数据。

import os
import yaml

def read_yaml():
    current_dir = os.path.dirname(os.path.abspath(__file__))
    config_path = os.path.join(current_dir, "config.yaml")

    with open(config_path, 'r', encoding='utf-8') as file:
        data = yaml.safe_load(file)

    return data

你可以在你的主程序中调用这个函数来获取配置数据。例如：

import read_yaml

# 读取配置文件
config = read_yaml.read_yaml()

现在，你就可以通过config字典访问到你在config.yaml文件中定义的所有配置项了。

我希望这篇博客对你有所帮助，如果你有任何疑问或者想要讨论更多关于Python的话题，欢迎留言。在后续的文章中，我还会分享更多有关Python的实用技巧和最佳实践。