diff options
Diffstat (limited to 'lib/sqlalchemy/engine/create.py')
-rw-r--r-- | lib/sqlalchemy/engine/create.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/lib/sqlalchemy/engine/create.py b/lib/sqlalchemy/engine/create.py index a252b7cfe..ac3d6a2d8 100644 --- a/lib/sqlalchemy/engine/create.py +++ b/lib/sqlalchemy/engine/create.py @@ -12,11 +12,13 @@ from typing import Union from . import base from . import url as _url +from .interfaces import DBAPIConnection from .mock import create_mock_engine from .. import event from .. import exc -from .. import pool as poollib from .. import util +from ..pool import _AdhocProxiedConnection +from ..pool import ConnectionPoolEntry from ..sql import compiler @@ -603,10 +605,13 @@ def create_engine(url: Union[str, "_url.URL"], **kwargs: Any) -> "base.Engine": if builtin_on_connect: event.listen(pool, "connect", builtin_on_connect) - def first_connect(dbapi_connection, connection_record): + def first_connect( + dbapi_connection: DBAPIConnection, + connection_record: ConnectionPoolEntry, + ): c = base.Connection( engine, - connection=poollib._AdhocProxiedConnection( + connection=_AdhocProxiedConnection( dbapi_connection, connection_record ), _has_events=False, |