使用现有的psycopg2连接p创建sqlalchemy引擎

Question

我正在使用sqlalchemy的声明性映射系统添加一个新的ORM类。我的代码库有一个现有的psycopg2连接池，我想重用它——我不希望使用我的orm类的代码有自己的池。有很多现有代码直接调用psycopg2池上的get_conn，所以我也不想直接替换它。在

我在构造要连接的引擎时遇到问题。在

pool_config = {...}
POOL = psycopg2.pool.ThreadedConnectionPool(0, 32, **pool_config)

[...]

engine = sqlalchemy.create_engine('postgresql://', pool=POOL)
Session = sqlalchemy.orm.sessionmaker(bind=engine)
...

问题是我调用create_engine

^{pr2}$

是否可以用这种方式使用我现有的池，或者我需要创建一个单独的连接池供这些类使用？在

Answer 1

您可以使用custom connection function：

create_engine('postgresql+psycopg2://', creator=POOL.getconn)