Pydantic和.env文件管理環(huán)境配置

更新時間：2025年04月09日 10:39:06 作者：三叔木卯

Pydantic 是一個 Python 第三方包,本文主要介紹了Pydantic和.env文件管理環(huán)境配置,具有一定的參考價值,感興趣的可以了解一下

1、安裝依賴庫

Pydantic 第一個版本時，安裝 Pydantic 庫就可以了，而第二個版本之后，settings 相關(guān)的功能被拆分到獨立的包了。獨立出來的包叫 pydantic-settings ，目前最新版本為 2.8.1。

現(xiàn)在只需要安裝 pydantic-settings 包，安裝命令如下：

pip install pydantic-settings==2.8.1

安裝 pydantic-settings 包時，會自動安裝其他 pydantic、python-dotenv 等依賴包。

2、創(chuàng)建 .env 文件

創(chuàng)建一個 .env 文件，根據(jù)具體項目需要填入相應的內(nèi)容。例如：

DATABASE_URL=your_database_url
DATABASE_USERNAME=your_username
DATABASE_PASSWORD=your_password

3、創(chuàng)建 config.py 文件

創(chuàng)建一個 config.py 文件，在里面創(chuàng)建一個 Settings 類，該類繼承自 pydantic_settings 的 BaseSettings 類。Settings 類添加和 .env 文件里的環(huán)境變量相同的類屬性，并創(chuàng)建一個 model_config 類屬性，用 SettingsConfigDict 初始化該類屬性。

代碼如下：

from pydantic_settings import BaseSettings, SettingsConfigDict  
  
  
class Settings(BaseSettings):  
    DATABASE_URL: str  
    DATABASE_USERNAME: str  
    DATABASE_PASSWORD: str  
  
    model_config = SettingsConfigDict(  
        env_file=".env"  
    )

為了方便獲取配置，還可以在 config.py 文件里添加一個 get_settings() 函數(shù)，如下所示：

def get_settings():  
    return Settings()

測試代碼如下：

if __name__ == '__main__':  
    settings = get_settings()  
    assert settings.DATABASE_URL == "your_database_url"  
    assert settings.DATABASE_USERNAME == "your_username"  
    assert settings.DATABASE_PASSWORD == "your_password"

4、為環(huán)境變量添加前綴以區(qū)別不同的環(huán)境

使用 .env 文件來配置環(huán)境變量，一個原因是為了避免敏感數(shù)據(jù)直接寫到代碼里（安全性）；一個原因是避免將變量直接硬編碼到代碼里，在需要的時候便于修改；還有一個原因是為了在不同的執(zhí)行環(huán)境下獲取不同的環(huán)境變量。

在上面的代碼里，每個環(huán)境變量只設(shè)置了一個值。那么如何給每個變量設(shè)置不同的值呢？

可以給環(huán)境變量添加前綴來區(qū)分不同的執(zhí)行環(huán)境。比如生產(chǎn)環(huán)境使用 PROD_ 前綴，開發(fā)環(huán)境使用 DEV_ 前綴。

初始化 model_config 類屬性時，SettingsConfigDict 類添加 env_prefix 參數(shù)，值為 os.getenv("ENVIRONMENT_PREFIX", "DEV_") 。表示會先從環(huán)境變量 ENVIRONMENT_PREFIX 讀取前綴，如果讀取不到，則默認使用 DEV_ 前綴。

同時 SettingsConfigDict 類添加 extra 參數(shù)，其值為 “allow”。因為 Pydantic 默認會忽略未在模型中定義的字段（即 extra="ignore"），這些字段不會被校驗或存儲。當設(shè)置為 extra="ignore" 后，模型會 ?接受并保留未定義的字段，將它們作為動態(tài)屬性存儲，但不會進行類型校驗。

注意：添加了 env_prefix 屬性之后，要記得根據(jù)需要設(shè)置一下 ENVIRONMENT_PREFIX 這個環(huán)境變量。

代碼如下：

	# 其他代碼...
	model_config = SettingsConfigDict(  
	    env_prefix=os.getenv("ENVIRONMENT_PREFIX", "DEV_"),  
	    env_file=".env",  
	    extra="allow"  
	)

5、最后的完整代碼

.env 文件的完整代碼示例：

DEV_DATABASE_URL=your_database_url  
DEV_DATABASE_USERNAME=your_username  
DEV_DATABASE_PASSWORD=your_password  
  
PROD_DATABASE_URL=prod_your_database_url  
PROD_DATABASE_USERNAME=prod_your_username  
PROD_DATABASE_PASSWORD=prod_your_password

config.py 的完整代碼示例：

import os  
  
from pydantic_settings import BaseSettings, SettingsConfigDict  
  
  
class Settings(BaseSettings):  
    DATABASE_URL: str  
    DATABASE_USERNAME: str  
    DATABASE_PASSWORD: str  
  
    model_config = SettingsConfigDict(  
        env_prefix=os.getenv("ENVIRONMENT_PREFIX", "DEV_"),  
        env_file=".env",  
        extra="allow"  
    )  
  
  
def get_settings():  
    return Settings()  
  
  
if __name__ == '__main__':  
    settings = get_settings()  
    prefix = os.getenv("ENVIRONMENT_PREFIX", "DEV_")  
    if prefix == "DEV_":  
        assert settings.DATABASE_URL == "your_database_url"  
        assert settings.DATABASE_USERNAME == "your_username"  
        assert settings.DATABASE_PASSWORD == "your_password"  
    elif prefix == "PROD_":  
        assert settings.DATABASE_URL == "prod_your_database_url"  
        assert settings.DATABASE_USERNAME == "prod_your_username"  
        assert settings.DATABASE_PASSWORD == "prod_your_password"

到此這篇關(guān)于Pydantic和.env文件管理環(huán)境配置的文章就介紹到這了,更多相關(guān)Pydantic .env環(huán)境配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: