Python的ini配置文件你了解吗

当我们在开发Python程序时，尤其是需要读取配置文件时，INI配置文件被广泛使用。下面是从头到尾完整的INI配置文件攻略，包含如何使用Python读取、写入、修改INI配置文件。

什么是INI文件

INI文件是一种纯文本文件格式，通常用作Windows操作系统中应用程序的配置文件。它的基本语法是以节（section）和键值对（key-value）的形式组织，一个节包括几个键值对，所以可以将INI文件视为多个节的集合。

INI文件通常具有以下结构：

[Section Name 1]
Key1 = Value1
Key2 = Value2

[Section Name 2]
Key3 = Value3
Key4 = Value4

在这里，Section Name是节的名称，通常被放置在方括号内。在任何给定的节中，键值对可以没有顺序，并且键和值可以由等号（=）、冒号、或空格符分隔。

使用Python的configparser模块操作INI文件

Python标准库中的configparser模块提供了一个方便的方式来读取，写入和修改INI配置文件。下面我们将介绍如何创建、读取、写入和修改INI文件。

读取INI文件

使用configparser中的ConfigParser类可以实现读取INI文件。

import configparser

config = configparser.ConfigParser()

config.read('config.ini')

通过read()方法将配置文件文件名传递给ConfigParser类实例，然后可以使用实例的方法来读取配置文件的键值对。

# 获取所有节的名称
sections = config.sections()
print(sections)

# 获取某个节中的所有键值对
options = config.options('database')
print(options)

# 获取键值对的值
DBHOST = config.get('database', 'dbhost')
DBPORT = config.get('database', 'dbport')
DBUSER = config.get('database', 'dbuser')
DBPWD = config.get('database', 'dbpwd')

print(DBHOST, DBPORT, DBUSER, DBPWD)

写入INI文件

使用configparser中的ConfigParser类可以实现将配置写入INI文件中。

import configparser

config = configparser.ConfigParser()

config['database'] = {'dbhost': 'localhost',
                 'dbport': '3306',
                 'dbuser': 'root',
                 'dbpwd': 'password'}

with open('config.ini', 'w') as configfile:
    config.write(configfile)

修改INI文件

使用configparser中的ConfigParser类可以实现修改INI文件中的配置。

import configparser

config = configparser.ConfigParser()

config.read('config.ini')

config.set('database', 'dbhost', 'newhost')

with open('config.ini', 'w') as configfile:
    config.write(configfile)

通过set()方法可以修改指定键的值，然后使用write()方法将更改写回文件。

示例1：使用INI文件存储爬虫信息

[Spider Settings]
URL: https://www.example.com
Max Depth: 3
Timeout: 10

import configparser

config = configparser.ConfigParser()
config.read('spider.ini')

url = config.get('Spider Settings', 'URL')
max_depth = config.getint('Spider Settings', 'Max Depth')
timeout = config.getint('Spider Settings', 'Timeout')

这个例子演示了如何通过get()和getint()方法从INI配置文件中读取信息。

示例2：使用INI文件存储机器学习模型和超参数

[Model Settings]
Model Name: RandomForestRegressor
Model Parameters: n_estimators=100
                  max_depth=10
                  min_samples_split=5
                  min_samples_leaf=2

[Training Settings]
Training Epoch: 50
Learning Rate: 0.001
Batch Size: 64

import configparser

config = configparser.ConfigParser()
config.read('model.ini')

model_name = config.get('Model Settings', 'Model Name')
model_params = dict(config.items('Model Settings')['Model Parameters'])
training_epoch = config.getint('Training Settings', 'Training Epoch')
learning_rate = config.getfloat('Training Settings', 'Learning Rate')
batch_size = config.getint('Training Settings', 'Batch Size')

这个例子演示了如何使用items()方法从INI配置文件中获取所有参数（多行字符串），然后将其转换为字典。