QuotaGuard代理环境下Heroku Django连接Azure SQL DB失败，未使用代理触发IP白名单限制

看起来你遇到的核心问题是QuotaGuard代理没有真正接管Azure SQL的连接请求，导致Heroku实例的原生IP直接访问数据库触发了白名单限制——而且QuotaGuard仪表盘没看到请求，也证实了这一点。我帮你拆解下代码里的关键问题，以及对应的修复方案：

为什么你的代理设置没生效？ #

你用的pyodbc驱动底层是C实现的ODBC组件，它直接调用系统的socket接口，完全不会走Python的socket模块。所以你用socket.socket = socks.socksocket这种“猴子补丁”的方式，对pyodbc的数据库连接是完全无效的——这就是代理没起作用的根本原因。

另外你的代码还有个小隐患：代理配置的变量（比如proxy_host）是在if QUOTAGUARD_URL块里定义的，但代理设置的代码却在块外面，如果环境变量没配置的话会直接报错，这个也需要修正。

两种可行的修复方案 #

方案一：改用纯Python的MSSQL驱动（代码修改少）

纯Python实现的MSSQL驱动（比如tds-python）会使用Python的socket模块，这样你的代理补丁就能生效。步骤如下：

1. 安装依赖：

pip install tds-python

2. 修改你的代码：

import pandas as pd
from PIL import Image
from sqlalchemy import create_engine
from sqlalchemy.pool import NullPool
from dotenv import load_dotenv
import socks
import socket
import os
import io

load_dotenv()

# 加载环境变量
CLIENT_ID = os.getenv("CLIENT_ID")
TENANT_ID = os.getenv("TENANT_ID")
CLIENT_SECRET = os.getenv("CLIENT_SECRET")
DB_SERVER = os.getenv("DB_SERVER")
DB_NAME = os.getenv("DB_NAME")
DB_TIMEOUT = os.getenv("DB_TIMEOUT")
QUOTAGUARD_URL = os.getenv("QUOTAGUARD_URL")

# 仅当代理URL存在时配置代理，避免未定义变量错误
if QUOTAGUARD_URL:
    proxy_protocol, proxy_auth_host = QUOTAGUARD_URL.split("://")
    proxy_auth, proxy_host_port = proxy_auth_host.split("@")
    proxy_user, proxy_pass = proxy_auth.split(":")
    proxy_host, proxy_port = proxy_host_port.split(":")
    proxy_port = int(proxy_port)

    # 应用SOCKS代理补丁（现在对纯Python驱动有效）
    socks.set_default_proxy(socks.SOCKS5, proxy_host, proxy_port, username=proxy_user, password=proxy_pass)
    socket.socket = socks.socksocket

# 改用纯Python的TDS驱动，适配Azure AD服务主体认证
connection_string = (
    f"mssql+tds://{CLIENT_ID}:{CLIENT_SECRET}@{DB_SERVER}:1433/{DB_NAME}?"
    f"encrypt=yes&trust_server_certificate=yes&connection_timeout={DB_TIMEOUT}"
    f"&authentication=ActiveDirectoryServicePrincipal&authority_id={TENANT_ID}"
)

engine = create_engine(connection_string, poolclass=NullPool)

try:
    query = "SELECT * FROM list_photos"
    with engine.connect() as conn:
        df = pd.read_sql(query, conn)
        print("DataFrame loaded successfully!")
        print(df.head())
except Exception as e:
    print(f"Error: {e}")

for index, row in df.iterrows():
    holdon = input("Press Enter to view the next image...")
    try:
        image = Image.open(io.BytesIO(row['photo']))
        image.show()
    except Exception as e:
        print(f"Error displaying image at index {index}: {e}")

方案二：用QuotaGuard SSH隧道（兼容性最强）

这种方式是在系统层面建立端口转发，不管你用什么数据库驱动，所有流量都会通过QuotaGuard的IP发出，完全绕过Python socket的限制，适合任何场景：

1. 添加QuotaGuard SSH的Buildpack到Heroku：

heroku buildpacks:add https://github.com/quotaguard/quotaguard-ssh-buildpack.git

2. 设置隧道转发的环境变量（把your-azure-sql-server换成你的实际Azure SQL服务器地址）：

heroku config:set QUOTAGUARD_SSH_TUNNEL="1433:your-azure-sql-server.database.windows.net:1433"

3. 修改你的数据库连接字符串，把DB_SERVER设置为localhost（因为隧道会把本地1433端口转发到Azure SQL的真实地址）。

测试验证 #

修改完成后，重新部署到Heroku，然后：

1. 检查QuotaGuard仪表盘，应该能看到对应的数据库连接请求记录；
2. 数据库的IP白名单限制会被自动绕过，因为流量是从QuotaGuard的白名单IP发出的。

备注：内容来源于stack exchange，提问作者Max Chergik