在開發(fā)過程中，經(jīng)常需要使用到各種配置信息，如數(shù)據(jù)庫連接字符串、API 密鑰等。為了保證這些信息的安全性，以及提高代碼的可維護(hù)性和靈活性，我們通常會(huì)將這些敏感信息存儲(chǔ)在環(huán)境變量中，而不是直接硬編碼在源代碼里。本文將介紹如何使用 python-dotenv 包來管理這些環(huán)境變量，確保你的 Python 項(xiàng)目更加安全和靈活。

為什么要使用 .env 文件儲(chǔ)存系統(tǒng)變量

安全性

將敏感信息如 API 密鑰、數(shù)據(jù)庫密碼等存放在 .env 文件中可以有效地避免這些信息被提交到版本控制系統(tǒng)（如 Git）中，減少敏感信息泄露的風(fēng)險(xiǎn)。
有助于遵循最佳實(shí)踐，即不在源代碼中硬編碼敏感信息。

便于管理

通過使用 .env 文件，所有的配置信息都集中在一個(gè)地方，方便管理和更新。特別是在項(xiàng)目需要在多種環(huán)境（開發(fā)、測(cè)試、生產(chǎn)等）運(yùn)行時(shí)，可以通過更換不同的 .env 文件來快速切換環(huán)境配置。

靈活性

環(huán)境變量的使用提高了項(xiàng)目的靈活性，允許同一份代碼在不同的環(huán)境下運(yùn)行而無需修改代碼本身。這在多人協(xié)作和應(yīng)用部署時(shí)尤為重要。

使用 python-dotenv 管理環(huán)境變量

python-dotenv 是一個(gè) Python 庫，可以用來讀取 .env 文件中的環(huán)境變量，并將它們加載到 Python 的環(huán)境變量中。這使得在代碼中使用這些變量變得非常簡單和直接。

安裝 python-dotenv

首先，你需要安裝 python-dotenv?？梢酝ㄟ^ pip 命令輕松安裝：

pip install python-dotenv

創(chuàng)建和配置 .env 文件

在項(xiàng)目的根目錄下創(chuàng)建一個(gè)名為 .env 的文件。在這個(gè)文件中，你可以定義需要的環(huán)境變量，例如：

# 這是 .env 文件中的注釋
DATABASE_URL="mysql://username:password@localhost/dbname"
API_KEY="your-secret-api-key"

請(qǐng)注意，.env 文件支持 # 開頭的注釋。

在 Python 中加載和使用環(huán)境變量

接下來，你可以在 Python 代碼中使用 python-dotenv 來加載這些環(huán)境變量。一種常見的做法是在項(xiàng)目的入口文件（如 app.py 或 main.py）中加載這些變量：

from dotenv import load_dotenv
import os

# 加載 .env 文件
load_dotenv()

# 使用環(huán)境變量
database_url = os.getenv('DATABASE_URL')
api_key = os.getenv('API_KEY')

print("Database URL:", database_url)
print("API Key:", api_key)

這段代碼首先從 .env 文件加載環(huán)境變量，然后通過 os.getenv 方法訪問這些變量。

dotenv() 加載環(huán)境變量后，這些變量將會(huì)覆蓋同名的已存在的環(huán)境變量。確保你了解這一行為，尤其是當(dāng)環(huán)境變量在不同配置文件中被重復(fù)定義時(shí)。

處理不同環(huán)境的配置

在代碼中，使用 load_dotenv() 默認(rèn)加載 .env 文件。
如果你有多個(gè)環(huán)境（比如開發(fā)環(huán)境和生產(chǎn)環(huán)境），你可以為每個(gè)環(huán)境創(chuàng)建不同的.env文件，例如.env.dev和.env.prod。然后，根據(jù)需要加載相應(yīng)的配置文件：

from dotenv import load_dotenv
import os

# 根據(jù)需要加載不同的.env文件
if some_condition_for_production:
    load_dotenv('.env.prod')
else:
    load_dotenv('.env.dev')

注意事項(xiàng)

1. 確保不要將.env文件推送到版本控制系統(tǒng)中，如Git。通常，你應(yīng)該在.gitignore文件中添加.env。
2. 在生產(chǎn)環(huán)境中，不建議使用python-dotenv直接加載環(huán)境變量，因?yàn)楦扑]使用操作系統(tǒng)級(jí)別的環(huán)境變量設(shè)置。

結(jié)語

通過使用 python-dotenv 和 .env 文件來管理環(huán)境變量，你的 Python 項(xiàng)目將更加安全、易于管理和靈活。這種方法尤其適合處理敏感信息和支持多環(huán)境配置，是現(xiàn)代應(yīng)用開發(fā)的一種最佳實(shí)踐。

這句話的意思是在生產(chǎn)環(huán)境（即實(shí)際運(yùn)行的服務(wù)器或者是用于正式業(yè)務(wù)的環(huán)境）中，不建議使用 python-dotenv 這個(gè)庫來管理環(huán)境變量。python-dotenv 是一個(gè)Python庫，它可以從一個(gè)文件（通常是 .env 文件）中讀取環(huán)境變量，并將它們導(dǎo)入到Python應(yīng)用的環(huán)境變量中。這種方式在開發(fā)和測(cè)試環(huán)境中非常方便，因?yàn)樗梢詭椭_發(fā)者快速設(shè)置和修改環(huán)境變量，而不需要修改操作系統(tǒng)的配置。

然而，在生產(chǎn)環(huán)境中，直接在操作系統(tǒng)級(jí)別設(shè)置環(huán)境變量通常被認(rèn)為是更安全和更穩(wěn)定的方法。操作系統(tǒng)級(jí)別的環(huán)境變量可以通過操作系統(tǒng)的配置工具或命令行來設(shè)置，不需要通過外部文件。這樣做的好處包括：

① 安全性：避免將敏感信息（如數(shù)據(jù)庫密碼）存放在可能被不當(dāng)訪問或泄露的文件中。
② 穩(wěn)定性：操作系統(tǒng)級(jí)的環(huán)境變量設(shè)置通常不易于因應(yīng)用程序錯(cuò)誤而被修改，且對(duì)所有運(yùn)行的程序都是一致的。
③ 管理方便：在多個(gè)應(yīng)用間共享環(huán)境變量時(shí)，統(tǒng)一在操作系統(tǒng)層面管理可以簡化配置過程，特別是在大型系統(tǒng)或多服務(wù)環(huán)境中。

因此，在生產(chǎn)環(huán)境中，直接通過操作系統(tǒng)管理環(huán)境變量，而非依賴于應(yīng)用層面的庫（如 python-dotenv），是推薦的做法。

在 CentOS 系統(tǒng)中設(shè)置環(huán)境變量：

①臨時(shí)設(shè)置：
可以在終端直接使用 export 命令設(shè)置環(huán)境變量，這種方式設(shè)置的環(huán)境變量只在當(dāng)前會(huì)話有效：export MY_VARIABLE=value
②永久設(shè)置：
對(duì)于所有用戶，可以將 export 命令添加到 /etc/profile 或 /etc/environment 文件中。修改這些文件需要管理員權(quán)限。

對(duì)于單個(gè)用戶，可以將 export 命令添加到用戶的 ~/.bashrc 或 ~/.bash_profile 文件中。
修改后，重新登錄或者使用 source ~/.bashrc 命令來立即應(yīng)用更改。

注意事項(xiàng)：
①確保在編輯這些文件時(shí)，保持語法正確，避免使用錯(cuò)誤的命令格式。
②對(duì)于涉及敏感信息的環(huán)境變量（如數(shù)據(jù)庫密碼），務(wù)必確保文件權(quán)限設(shè)置得當(dāng)，防止非授權(quán)訪問。
③修改系統(tǒng)級(jí)配置文件（如 /etc/profile 或 /etc/environment）時(shí)應(yīng)特別小心，因?yàn)殄e(cuò)誤的設(shè)置可能影響到系統(tǒng)的正常使用。

通過這些方法，你可以根據(jù)你的需要和安全要求，在操作系統(tǒng)層面安全地管理環(huán)境變量。

在Python中調(diào)用系統(tǒng)環(huán)境變量相對(duì)簡單。你可以使用Python標(biāo)準(zhǔn)庫中的 os 模塊來獲取環(huán)境變量。以下是一個(gè)基本的示例，展示如何在Python代碼中讀取和使用環(huán)境變量：
①導(dǎo)入 os 模塊：
首先，需要導(dǎo)入 os 模塊，這個(gè)模塊提供了許多與操作系統(tǒng)交互的功能，包括環(huán)境變量的管理。
②使用 os.getenv() 或 os.environ：
os.getenv()：這個(gè)函數(shù)允許你讀取一個(gè)環(huán)境變量的值，如果該環(huán)境變量不存在，你可以指定一個(gè)默認(rèn)值。
os.environ：這是一個(gè)環(huán)境變量字典，它包含了所有當(dāng)前環(huán)境的變量。如果變量不存在，訪問 os.environ 時(shí)會(huì)拋出一個(gè)異常。

示例代碼：

下面的Python代碼演示了如何獲取名為 MY_VARIABLE 的環(huán)境變量的值，并處理可能的異常：

import os
# 使用 os.getenv() 獲取環(huán)境變量

variable_value = os.getenv(‘MY_VARIABLE', ‘Default Value')
print(“The value of MY_VARIABLE is:”, variable_value)
# 使用 os.environ 獲取環(huán)境變量，如果變量不存在，則處理異常

try:
  variable_value = os.environ[‘MY_VARIABLE']
  print(“The value of MY_VARIABLE is:”, variable_value)
except KeyError:
  print(“MY_VARIABLE is not set in the environment.”)

使用這些方法的場(chǎng)景：

當(dāng)你確定環(huán)境變量一定存在時(shí)，可以直接使用 os.environ，這種方式可以讓你的程序在環(huán)境變量缺失時(shí)立即報(bào)錯(cuò)，便于調(diào)試。

如果你不確定環(huán)境變量是否設(shè)置，或者你想為未設(shè)置的環(huán)境變量提供一個(gè)默認(rèn)值，os.getenv() 是更安全的選擇。

這樣，你就可以根據(jù)系統(tǒng)中設(shè)置的環(huán)境變量來調(diào)整Python程序的行為了，比如配置數(shù)據(jù)庫連接信息、API密鑰等敏感信息。 

以上就是Python使用python-dotenv管理環(huán)境變量的方法的詳細(xì)內(nèi)容，更多關(guān)于Python python-dotenv環(huán)境變量的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！