Python中如何使用环境变量设置数据库连接信息

在Python中，可以使用os.environ模块来操作环境变量，并使用它来设置数据库连接信息。下面是一个使用例子，展示了如何使用环境变量设置数据库连接信息。

首先，需要在操作系统中设置数据库连接的环境变量。以Windows系统为例，在命令提示符下执行以下命令：

set DB_HOST=localhost
set DB_PORT=3306
set DB_USERNAME=username
set DB_PASSWORD=password
set DB_DATABASE=database_name

然后，在Python代码中，可以使用os.environ模块来获取环境变量的值。例如，可以通过os.environ.get()方法获取环境变量的值，如下所示：

import os

db_host = os.environ.get('DB_HOST')
db_port = os.environ.get('DB_PORT')
db_username = os.environ.get('DB_USERNAME')
db_password = os.environ.get('DB_PASSWORD')
db_database = os.environ.get('DB_DATABASE')

# 使用获取到的环境变量值连接数据库
# 代码省略...

在上面的代码中，通过os.environ.get()方法获取了环境变量中的数据库连接信息。可以将这些变量直接用于连接数据库的代码。

此外，为了方便管理数据库连接信息，还可以将这些环境变量值保存在一个.env文件中，并使用第三方库python-dotenv来加载这些环境变量。首先，需要安装python-dotenv库，可以使用以下命令来安装：

pip install python-dotenv

然后，可以在代码中使用以下方式加载.env文件中的环境变量：

from dotenv import load_dotenv
import os

# 加载.env文件中的环境变量
load_dotenv()

db_host = os.environ.get('DB_HOST')
db_port = os.environ.get('DB_PORT')
db_username = os.environ.get('DB_USERNAME')
db_password = os.environ.get('DB_PASSWORD')
db_database = os.environ.get('DB_DATABASE')

# 使用获取到的环境变量值连接数据库
# 代码省略...

在上面的代码中，使用load_dotenv()函数加载了.env文件中的环境变量。然后就可以通过os.environ.get()方法获取这些环境变量的值了。

总结来说，通过使用os.environ模块或者python-dotenv库，可以方便地在Python中使用环境变量设置数据库连接信息。这样，就可以将敏感信息（如数据库用户名和密码）从代码中分离出来，提高代码的安全性和可维护性。