python学习之读取配置文件

下面详细讲解一下如何在Python中读取配置文件的完整攻略。

1. 安装pyyaml库

在开始读取配置文件之前，我们需要先安装一个PyYAML库，这个库将会帮助我们读取常用的YAML格式的配置文件。我们可以使用pip安装它，具体操作如下：

!pip install pyyaml

2. 创建配置文件

在读取配置文件之前，我们还需要先创建一个配置文件，例如我们创建一个名为config.yaml的YAML格式文件，内容如下：

# 数据库配置
database:
  host: localhost
  port: 3306
  username: root
  password: 123456

# 服务器配置
server:
  host: 0.0.0.0
  port: 80

3. 读取配置文件

有了PyYAML库和配置文件，我们现在可以开始读取配置文件了。读取过程主要分为三步：

使用PyYAML库加载配置文件：

import yaml

with open('config.yaml', 'r') as f:
    config = yaml.load(f, Loader=yaml.FullLoader)

读取配置项：

通过上一步加载得到的config对象，我们可以读取其中的配置项。例如，我们需要获取数据库的用户名和密码，可以使用下面的代码：

db_username = config['database']['username']
db_password = config['database']['password']

同样，我们也可以获取服务器的地址和端口号：

server_host = config['server']['host']
server_port = config['server']['port']

使用配置项：

读取配置项之后，我们就可以使用它们了。对于数据库的例子，我们可以使用读取到的用户名和密码来连接数据库：

import pymysql

conn = pymysql.connect(
    host=config['database']['host'],
    port=config['database']['port'],
    user=db_username,
    password=db_password,
    db='mydb',
    charset='utf8mb4'
)

对于服务器的例子，我们可以使用读取到的地址和端口号来启动一个Web服务器：

from flask import Flask

app = Flask(__name__)

@app.route('/')
def index():
    return 'Hello, world!'

if __name__ == '__main__':
    app.run(host=server_host, port=server_port)

示例说明

以上就是读取配置文件的完整攻略。下面给出两个简单的示例。

示例1：读取整个配置文件

如果我们需要一次性读取整个配置文件，可以使用下面的代码：

import yaml

with open('config.yaml', 'r') as f:
    config = yaml.load(f, Loader=yaml.FullLoader)

print(config)

运行后输出：

{'database': {'host': 'localhost', 'port': 3306, 'username': 'root', 'password': '123456'}, 'server': {'host': '0.0.0.0', 'port': 80}}

示例2：读取指定配置项

如果我们只需要读取配置文件中的一部分内容，可以使用下面的代码：

import yaml

with open('config.yaml', 'r') as f:
    config = yaml.load(f, Loader=yaml.FullLoader)

db_username = config['database']['username']
db_password = config['database']['password']

print('database username:', db_username)
print('database password:', db_password)