Python配置文件数据库链接

本文将详细阐述如何使用Python配置文件来实现数据库链接。

一、配置文件介绍

配置文件是存储应用程序配置信息的文件，可以包含数据库连接字符串、服务器地址、访问密钥等敏感信息。在Python中，常用的配置文件格式有ini、json和yaml。

以下是一个简单的ini配置文件的例子：

[Database]
host = localhost
port = 3306
username = user
password = pwd
database = mydb

二、读取配置文件

在Python中，可以使用ConfigParser模块来读取ini格式的配置文件。

import configparser

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

# 获取数据库配置信息
host = config.get('Database', 'host')
port = config.getint('Database', 'port')
username = config.get('Database', 'username')
password = config.get('Database', 'password')
database = config.get('Database', 'database')

print(f"数据库配置信息：{host}:{port}, {username}:{password}@{database}")

三、建立数据库连接

Python中有多种方式来建立数据库连接，比如使用第三方库如MySQLdb、psycopg2等，这些库需要使用配置文件中的参数来建立连接。

以下是一个使用MySQLdb库建立MySQL数据库连接的示例：

import MySQLdb

# 建立数据库连接
conn = MySQLdb.connect(
    host=host,  # 从配置文件中获取
    port=port,  # 从配置文件中获取
    user=username,  # 从配置文件中获取
    passwd=password,  # 从配置文件中获取
    db=database  # 从配置文件中获取
)

# 执行数据库操作
cursor = conn.cursor()
cursor.execute("SELECT * FROM table")
result = cursor.fetchall()

# 关闭数据库连接
cursor.close()
conn.close()

四、优化配置文件读取

为了提高配置文件的读取效率，可以将配置文件的内容缓存到内存中，这样可以减少文件IO的次数。

import configparser

class Config:
    def __init__(self):
        self.config = configparser.ConfigParser()
        self.config.read('config.ini')

    def get_database_config(self):
        host = self.config.get('Database', 'host')
        port = self.config.getint('Database', 'port')
        username = self.config.get('Database', 'username')
        password = self.config.get('Database', 'password')
        database = self.config.get('Database', 'database')

        return host, port, username, password, database

config = Config()
host, port, username, password, database = config.get_database_config()

conn = MySQLdb.connect(
    host=host,
    port=port,
    user=username,
    passwd=password,
    db=database
)

五、结语

通过使用Python配置文件来进行数据库链接，可以更加方便地管理和维护应用程序的配置信息，同时保护敏感信息的安全性。希望本文对您在Python开发中实现数据库链接有所帮助。