diff --git a/asyncpg-stubs/connect_utils.pyi b/asyncpg-stubs/connect_utils.pyi index 075b213..7b4d492 100644 --- a/asyncpg-stubs/connect_utils.pyi +++ b/asyncpg-stubs/connect_utils.pyi @@ -32,6 +32,7 @@ class _ConnectionParameters(NamedTuple): server_settings: dict[str, str] | None target_session_attrs: SessionAttribute krbsrvname: str | None + gsslib: connection._GSSLibType | None class _ClientConfiguration(NamedTuple): command_timeout: float | None diff --git a/asyncpg-stubs/connection.pyi b/asyncpg-stubs/connection.pyi index 8945c1b..91a2230 100644 --- a/asyncpg-stubs/connection.pyi +++ b/asyncpg-stubs/connection.pyi @@ -52,6 +52,7 @@ _OutputType: TypeAlias = PathLike[Any] | BinaryIO | _Writer _SourceType: TypeAlias = PathLike[Any] | BinaryIO | AsyncIterable[bytes] _CopyFormat: TypeAlias = Literal['text', 'csv', 'binary'] +_GSSLibType: TypeAlias = Literal['gssapi', 'sspi'] class _Listener(Protocol): def __call__( @@ -376,6 +377,7 @@ async def connect( server_settings: dict[str, str] | None = ..., target_session_attrs: connect_utils.SessionAttribute | None = ..., krbsrvname: str | None = ..., + gsslib: _GSSLibType | None = ..., ) -> Connection[_Record]: ... @overload async def connect( @@ -400,6 +402,7 @@ async def connect( server_settings: dict[str, str] | None = ..., target_session_attrs: connect_utils.SessionAttribute | None = ..., krbsrvname: str | None = ..., + gsslib: _GSSLibType | None = ..., ) -> _Connection: ... @overload async def connect( @@ -422,6 +425,7 @@ async def connect( server_settings: dict[str, str] | None = ..., target_session_attrs: connect_utils.SessionAttribute | None = ..., krbsrvname: str | None = ..., + gsslib: _GSSLibType | None = ..., ) -> Connection[protocol.Record]: ... class _ConnectionProxy(Generic[_Record]):