Apache Superset Docker 部署中解决 psycopg2 模块缺失问题

在使用 Docker 部署 Apache Superset 5.0.0rc1 版本时，许多用户遇到了一个常见问题：系统提示"ModuleNotFoundError: No module named 'psycopg2'"错误。这个问题主要出现在尝试连接 PostgreSQL 数据库时，因为新版本的 Superset 默认不再包含 PostgreSQL 驱动程序。

问题背景

Apache Superset 是一个现代化的企业级商业智能和数据分析平台。从 5.0.0 版本开始，项目团队对 Python 包管理方式进行了重大调整，用 uv 替代了传统的 pip 作为默认包管理器。这一变化虽然带来了性能提升，但也导致了一些兼容性问题，特别是对于依赖特定数据库驱动程序的用户。

问题表现

当用户尝试使用以下 Docker Compose 配置部署 Superset 5.0.0rc1 时：

services:
  superset:
    image: apache/superset:5.0.0rc1
    environment:
      - EXTRA_PIP_PACKAGES=psycopg2-binary
    command: >
      sh -c "
      pip install psycopg2-binary &&
      superset db upgrade &&
      superset init &&
      gunicorn -w 2 --timeout 60 -b 0.0.0.0:8088 'superset.app:create_app()'
      "

系统会抛出 psycopg2 模块缺失的错误。这是因为在 Superset 5.0 中，虽然用户尝试通过 pip 安装 psycopg2-binary，但实际使用的是 uv 包管理器，导致安装命令不兼容。

解决方案

方法一：使用 uv 包管理器安装

正确的解决方法是使用 uv 替代 pip 来安装 psycopg2-binary：

command: >
  sh -c "
  uv pip install psycopg2-binary &&
  superset db upgrade &&
  superset init &&
  gunicorn -w 2 --timeout 60 -b 0.0.0.0:8088 'superset.app:create_app()'
  "

方法二：修改基础镜像

另一种方法是使用包含 PostgreSQL 驱动程序的开发版本镜像：

image: apache/superset:latest-dev

方法三：修改本地依赖文件

对于需要长期维护的部署环境，可以考虑修改 Superset 的本地依赖文件，在 superset/docker/requirements-local.txt 中添加：

psycopg2-binary>=2.9.0

技术原理

uv 是 Rust 编写的新一代 Python 包管理器，相比传统 pip 具有更快的安装速度和更好的依赖解析能力。Superset 5.0 采用 uv 是为了提高构建效率和减少依赖冲突。然而，这种改变需要用户在部署时注意包管理方式的差异。

对于 PostgreSQL 连接，psycopg2 是 Python 中最流行的 PostgreSQL 适配器，它提供了与 PostgreSQL 数据库交互的接口。psycopg2-binary 是预编译的二进制版本，避免了编译依赖的问题。

最佳实践

1. 对于生产环境，建议明确指定 Superset 版本号，而不是使用 latest 标签
2. 数据库驱动程序最好在构建阶段安装，而不是在运行时安装
3. 考虑使用数据库连接池配置来优化性能
4. 定期检查依赖版本，确保与数据库服务器版本兼容

总结

Apache Superset 5.0 的包管理方式变更虽然带来了性能提升，但也需要用户在部署时做出相应调整。理解 uv 和 pip 的区别，以及正确安装数据库驱动程序，是确保 Superset 顺利运行的关键。通过本文提供的解决方案，用户可以轻松解决 psycopg2 模块缺失的问题，顺利部署最新版本的 Superset。