极客笔记Flask Flask应用程序中的环境变量的设置和获取
在本文中，我们将介绍如何在Flask应用程序中设置和获取环境变量。环境变量是应用程序中可被访问的全局变量，可以存储敏感信息、配置参数、数据库链接等。通过使用环境变量，我们可以安全地将敏感信息分离出应用程序代码，并轻松地在不同的环境中切换和配置应用程序。
阅读更多：Flask 教程
使用os模块设置和获取环境变量
在Flask应用程序中，我们可以使用Python内置的os模块来设置和获取环境变量。os模块提供了一些函数来操作操作系统相关功能，包括获取和设置环境变量。
设置环境变量
要设置环境变量，我们可以使用os.environ字典对象来添加新的键值对。下面是一个设置环境变量的示例：
import os
from flask import Flask
app = Flask(__name__)
@app.route('/')
def hello():
return 'Hello World!'
if __name__ == '__main__':
# 设置环境变量
os.environ['DB_URL'] = 'mongodb://localhost:27017/mydatabase'
app.run()
在上面的示例中，我们使用os.environ字典对象将名为DB_URL的环境变量设置为mongodb://localhost:27017/mydatabase。
获取环境变量
要获取已设置的环境变量，我们可以使用os.environ.get()函数。下面是一个示例：
import os
from flask import Flask
app = Flask(__name__)
@app.route('/')
def hello():
# 获取环境变量
db_url = os.environ.get('DB_URL')
return f'The DB URL is: {db_url}'
if __name__ == '__main__':
app.run()
在上面的例子中，我们使用os.environ.get()函数获取名为DB_URL的环境变量，并将其存储在db_url变量中。然后，在hello()函数中，我们将环境变量的值返回给客户端。
使用Python-dotenv库设置和获取环境变量
除了使用os模块之外，我们还可以使用第三方库python-dotenv来设置和获取环境变量。python-dotenv是一个用于从.env文件加载环境变量的库，它可以使我们在不同的环境中轻松地切换和配置应用程序。
安装和配置python-dotenv
要使用python-dotenv库，我们首先需要安装它。可以使用以下命令来安装：
pip install python-dotenv
安装完成后，我们可以在项目的根目录下创建一个名为.env的文件，并在其中添加我们的环境变量。例如：
DB_URL=mongodb://localhost:27017/mydatabase
在Flask应用程序中，我们可以使用如下代码读取.env文件：
from dotenv import load_dotenv
load_dotenv()
设置和获取环境变量
在使用python-dotenv后，我们可以使用os.environ字典对象来设置和获取环境变量，它将从.env文件中加载环境变量。
下面的示例演示了如何使用python-dotenv设置和获取环境变量：
import os
from flask import Flask
from dotenv import load_dotenv
load_dotenv()
app = Flask(__name__)
@app.route('/')
def hello():
# 获取环境变量
db_url = os.environ.get('DB_URL')
return f'The DB URL is: {db_url}'
if __name__ == '__main__':
app.run()
在上面的示例中，我们先使用load_dotenv()函数从.env文件中加载环境变量。然后，我们使用os.environ.get()函数获取名为DB_URL的环境变量，并将其存储在db_url变量中。
总结
在本文中，我们介绍了如何在Flask应用程序中设置和获取环境变量。我们可以使用os模块或第三方库python-dotenv来实现这个功能。通过设置环境变量，我们可以安全地存储敏感信息和配置参数，并轻松地在不同的环境中切换和配置应用程序。使用环境变量是一个良好的实践，可以使我们的应用程序更灵活和可配置。希望本文对你有所帮助！