Python中使用.py配置文件的方法详解

在Python开发中，我们通常需要读取配置文件，将一些地址、路径、参数等内容从代码中独立出来，方便管理和维护。Python支持常见的多种配置文件格式，如INI格式、JSON格式、XML格式等，其中.py格式配置文件则相对比较特殊，其特殊之处在于.py格式本身就是Python模块，可以直接在代码中引用，具有更高的灵活性和扩展性。

基本用法

在Python中，我们可以直接创建一个.py格式的模块文件，比如config.py，在该模块文件中定义需要配置的变量、常量、函数等，然后在本地或远程环境中运行Python脚本时，直接在代码中引用该模块文件，就能够读取其中的配置内容了。比如：

# config.py
DATABASE = {
    'driver': 'mysql',
    'host': 'localhost',
    'port': 8889,
    'user': 'root',
    'password': '123456',
    'database': 'test'
}

# main.py
import config

def connect():
    db = create_db_conn(config.DATABASE)
    return db

if __name__ == '__main__':
    connect()

上述代码中，我们通过import config语句将config.py模块文件引入到当前脚本中，并读取其中的DATABASE配置项，用于创建数据库连接。由于.py格式模块本身就是一个可执行的Python脚本，因此我们可以在config.py中定义任何Python代码，包括函数、类、变量等。

灵活处理路径地址

有时候，我们在配置文件中需要定义一些路径地址，如日志文件路径、图片文件路径等，这些路径地址可能涉及到跨平台、文件夹日期格式化、文件夹创建等细节问题，需要比较灵活的处理方式。此时，我们可以使用os.path模块来进行路径处理，如果需要对路径进行格式化或创建文件夹，还可以使用datetime模块和os.mkdir函数来实现。示例代码如下：

# config.py
import os
from datetime import datetime

LOG_DIR = os.path.join(os.path.dirname(__file__), 'log')

LOG_FILE = os.path.join(LOG_DIR, datetime.now().strftime('%Y-%m-%d') + '.log')

if not os.path.exists(LOG_DIR):
    os.mkdir(LOG_DIR)

def get_log_file():
    return LOG_FILE

上述代码中，我们通过os.path.join函数来将当前模块文件所在的目录和log文件夹名称连接起来，得到日志文件PATH；然后使用datetime.now()函数来获取当前的时间戳，并调用其strftime函数将时间戳格式化为YEAR-MONTH-DAY的形式，拼接到日志文件名中得到LOG_FILE。最后我们使用os.mkdir函数来检查日志文件夹LOG_DIR是否存在，如果不存在则创建它。我们还定义了一个get_log_file函数，用于其他模块中获取日志文件PATH。

处理复杂配置内容

在实际Python开发中，配置文件可能包含比较复杂的数据结构，如多级字典、嵌套列表、元组等。对于这样的数据结构，我们可以使用Python内置的json、pickle等模块来将其序列化为字符串形式，然后在代码中读取并反序列化。

# config.py
import json

DB_CONFIG = {
    'driver': 'mysql',
    'host': 'localhost',
    'port': 8889,
    'user': 'root',
    'password': '123456',
    'database': 'test'
}

def get_db_config():
    return json.dumps(DB_CONFIG)

# main.py
import config
import json

def connect():
    db_config_str = config.get_db_config()
    db_config = json.loads(db_config_str)
    db = create_db_conn(db_config)
    return db

if __name__ == '__main__':
    connect()

在上述代码中，我们在config.py模块文件中统一定义了数据库配置的字典DB_CONFIG，然后在get_db_config函数中使用json.dumps函数将其转换为字符串；在main.py中读取字符串后，使用json.loads函数将其反序列化为Python中的字典类型，然后传递给创建数据库连接的函数create_db_conn。

小结

本文介绍了Python中使用.py配置文件的方法，包括基本用法、灵活处理路径地址、处理复杂配置内容等方面。总的来说，使用.py格式配置文件能够提供更灵活、更规范的配置管理方式，对于小型项目或中小团队的Python项目开发十分适用。