创建基于psycopg3的SQLAlchemy引擎

Question

我需要将以下代码升级为基于

psycopg

版本 3 的等效代码：

import psycopg2
from sqlalchemy import create_engine

engine = create_engine('postgresql+psycopg2://', creator=connector)

这个 psycopg2 URL 很有魅力，但是：

import psycopg # v3.1
from sqlalchemy import create_engine

engine = create_engine('postgresql+psycopg://', creator=connector)

（我也尝试过“psycopg3”这个词，但没有成功）

返回：

Traceback (most recent call last):

  File "/tmp/ipykernel_1032556/253047102.py", line 1, in <cell line: 1>
    engine = create_engine('postgresql+psycopg://', creator=connector)

  File "<string>", line 2, in create_engine

  File "/usr/local/lib/python3.10/dist-packages/sqlalchemy/util/deprecations.py", line 309, in warned
    return fn(*args, **kwargs)

  File "/usr/local/lib/python3.10/dist-packages/sqlalchemy/engine/create.py", line 534, in create_engine
    entrypoint = u._get_entrypoint()

  File "/usr/local/lib/python3.10/dist-packages/sqlalchemy/engine/url.py", line 661, in _get_entrypoint
    cls = registry.load(name)

  File "/usr/local/lib/python3.10/dist-packages/sqlalchemy/util/langhelpers.py", line 343, in load
    raise exc.NoSuchModuleError(

NoSuchModuleError: Can't load plugin: sqlalchemy.dialects:postgresql.psycopg

那么，如何正确创建基于

psycopg

（v3.x）的SQLAlchemy引擎？

我的

sqlalchemy

版本是：“1.4.35”（尝试过版本

1.4.40

但面临

AttributeError: module 'sqlalchemy' has no attribute 'dialects'

错误）。

psycopg3 文档：https://www.psycopg.org/psycopg3/docs/api/
sqlalchemy 文档：https://docs.sqlalchemy.org/en/14/core/engines.html

Answer 1

只是对当前答案的更新：Sqlalchemy 2.0 已经发布，并且它确实支持 psycopg3。您需要升级到2.0才能使用。

请注意，连接字符串必须从

postgresql

更改为

postgresql+psycopg

，否则 SqlAlchemy 将（在撰写本文时）尝试使用

psycopg2

。

有关更多信息，请参阅此处的文档：https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#module-sqlalchemy.dialects.postgresql.psycopg

Answer 2

SQLAlchemy 不支持 1.4.x 系列中的 psycopg3。 + 符号后面的部分是 sqlalchemy.dialects.postgresql.{engine} 内的包，其中引擎是 psycopg2、asyncpg 或其他。

其他方言的工作方式相同。

但是1.4.x系列中没有psycopg和psycopg3驱动程序。