部署FastAPI到Heroku时，如何更新Pydantic的DATABASE_URL变量？

我想用Heroku和Heroku Postgres部署FastAPI应用，但Heroku提供的数据库URL以postgres://开头，而SQLAlchemy仅支持postgresql://格式。我尝试了多种修改DATABASE_URL的方案，但均触发相同错误：

sqlalchemy.exc.NoSuchModuleError: Can't load plugin: sqlalchemy.dialects:postgres

尝试过的代码如下：

def get_database_url(url):
    if url.startswith("postgres://"):
        url = url.replace("postgres://", "postgresql://", 1)
    return URL  # 此处存在拼写错误！应返回小写的url而非大写URL

# 尝试1
class Settings(BaseSettings):
    load_dotenv()

    DATABASE_URL: str = get_database_url(os.getenv("DATABASE_URL"))

# 尝试2
class Settings(BaseSettings):
    ...

    class Config:
        @validator("DATABASE_URL", pre=True)
        def parse_env_var(cls, value: str):
            return get_database_url(value)

# 尝试3
class Settings(BaseSettings):
    ...

    class Config:
        @classmethod
        def parse_env_var(cls, field_name: str, raw_val: str) -> Any:
            if field_name == 'DATABASE_URL':
                return get_database_url(raw_val)
            return cls.json_loads(raw_val)

解决方案 #

首先修正代码中的拼写错误：get_database_url函数最后返回的URL是未定义变量，应改为返回小写的url——这是导致你所有方案失败的核心原因之一。

方案1：初始化Settings前直接处理URL #

from pydantic import BaseSettings
import os
from dotenv import load_dotenv

# 提前加载环境变量（本地开发用，Heroku部署时可省略）
load_dotenv()

def get_database_url(url):
    if url and url.startswith("postgres://"):
        return url.replace("postgres://", "postgresql://", 1)
    return url

# 先修正URL再传入Settings
raw_db_url = os.getenv("DATABASE_URL")
corrected_db_url = get_database_url(raw_db_url)

class Settings(BaseSettings):
    DATABASE_URL: str = corrected_db_url

方案2：正确使用Pydantic字段验证器 #

确保验证器定义规范，字段已明确声明：

from pydantic import BaseSettings, validator
from dotenv import load_dotenv

load_dotenv()

def get_database_url(url):
    if url and url.startswith("postgres://"):
        return url.replace("postgres://", "postgresql://", 1)
    return url

class Settings(BaseSettings):
    DATABASE_URL: str

    @validator("DATABASE_URL", pre=True)
    def fix_postgres_url(cls, v):
        return get_database_url(v)

方案3：创建SQLAlchemy引擎时修正URL #

如果前两种方案仍有问题，可绕开Settings，在引擎创建的最后一步处理：

from sqlalchemy import create_engine
from pydantic import BaseSettings
import os

class Settings(BaseSettings):
    DATABASE_URL: str = os.getenv("DATABASE_URL")

settings = Settings()
# 修正URL后创建引擎
engine_url = settings.DATABASE_URL.replace("postgres://", "postgresql://", 1) if settings.DATABASE_URL.startswith("postgres://") else settings.DATABASE_URL
engine = create_engine(engine_url)

关键注意事项 #

• 本地开发时确保load_dotenv()正确加载了.env文件，Heroku部署时平台会自动注入DATABASE_URL环境变量，无需本地配置
• 验证器的pre=True表示在Pydantic默认解析逻辑前执行，保证修正后的URL被正确识别
• 若使用Pydantic v2，验证器写法需调整为@field_validator("DATABASE_URL", mode="before")

内容的提问来源于stack exchange，提问作者Ha Tran