Python中MySQL Connector重复要求输入密码的解决方法咨询

以下几种方案可以在保留认证机制的前提下避免重复输入密码：

方案1：缓存密码与连接实例 #

第一次调用时输入密码，之后复用缓存的密码（甚至直接复用连接对象），避免重复输入和重复创建连接。

修改后的代码示例：

import mysql.connector

# 缓存变量，存储已输入的密码和连接实例
_cached_password = None
_cached_connection = None

def libdb_connect_auth():
    global _cached_password, _cached_connection
    try:
        # 如果已有缓存连接且未断开，直接返回
        if _cached_connection and _cached_connection.is_connected():
            return _cached_connection
        
        # 如果没有缓存密码，提示输入
        if not _cached_password:
            _cached_password = input("Enter the password of the database:")
        
        # 创建新连接并缓存
        _cached_connection = mysql.connector.connect(
            host='localhost', 
            user='root', 
            passwd=_cached_password,
            database='lib', 
            charset='utf8'
        )
        return _cached_connection
    except Exception as e:
        print("Connection error...", e)
        # 连接失败时清空缓存，下次重新输入
        _cached_password = None
        _cached_connection = None
        return None

优点：完全保留交互式输入的认证方式，无需额外配置；复用连接提升性能。
缺点：如果连接意外断开，会自动重新尝试使用缓存密码连接，若密码变更则需要重启程序。

方案2：通过环境变量传递密码 #

让用户在启动程序前设置环境变量，代码从环境变量中读取密码，避免交互式输入。

代码示例：

import mysql.connector
import os

def libdb_connect_auth():
    try:
        # 从环境变量读取密码
        db_password = os.environ.get('DB_PASSWORD')
        if not db_password:
            # 环境变量未设置时， fallback 到交互式输入
            db_password = input("Enter the password of the database:")
        
        libdb = mysql.connector.connect(
            host='localhost', 
            user='root', 
            passwd=db_password,
            database='lib', 
            charset='utf8'
        )
        return libdb
    except Exception as e:
        print("Connection error...", e)
        return None

使用方式：启动程序前执行（Linux/macOS）：

export DB_PASSWORD="你的数据库密码"
python 你的主程序.py

Windows命令行：

set DB_PASSWORD=你的数据库密码
python 你的主程序.py

优点：无需修改代码逻辑，密码不会硬编码；支持非交互式运行场景。
缺点：需要用户提前设置环境变量，临时会话关闭后需要重新设置。

方案3：使用本地配置文件 #

创建一个权限受限的配置文件存储密码，代码读取该文件获取密码，只需配置一次即可长期使用。

1. 创建配置文件 db_config.ini（注意设置文件权限为仅自己可读，比如Linux下chmod 600 db_config.ini）：

[database]
host = localhost
user = root
password = 你的数据库密码
database = lib
charset = utf8

2. 修改连接函数读取配置文件：

import mysql.connector
import configparser

def libdb_connect_auth():
    try:
        config = configparser.ConfigParser()
        config.read('db_config.ini')
        
        # 读取配置项
        db_host = config.get('database', 'host')
        db_user = config.get('database', 'user')
        db_password = config.get('database', 'password')
        db_name = config.get('database', 'database')
        db_charset = config.get('database', 'charset')
        
        libdb = mysql.connector.connect(
            host=db_host, 
            user=db_user, 
            passwd=db_password,
            database=db_name, 
            charset=db_charset
        )
        return libdb
    except Exception as e:
        print("Connection error...", e)
        return None

优点：一次配置长期生效，无需每次输入；配置集中管理，方便修改。
缺点：需要妥善保管配置文件，避免密码泄露；不适合多用户共享的运行环境。

内容的提问来源于stack exchange，提问作者Pipal Deka