Update API for SSPI authentication

(see MagicStack/asyncpg#1128)
bryanforbes · Mar 11, 2024 · 700e319 · 700e319
1 parent b8b5b8b
commit 700e319
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git 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
@@ -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]):