Update API for GSSAPI authentication

(see MagicStack/asyncpg#1122)
bryanforbes · Mar 4, 2024 · ffd43bb · ffd43bb
1 parent 9368378
commit ffd43bb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/asyncpg-stubs/connect_utils.pyi b/asyncpg-stubs/connect_utils.pyi
@@ -31,6 +31,7 @@ class _ConnectionParameters(NamedTuple):
     direct_tls: bool
     server_settings: dict[str, str] | None
     target_session_attrs: SessionAttribute
+    krbsrvname: str | None
 
 class _ClientConfiguration(NamedTuple):
     command_timeout: float | None

diff --git a/asyncpg-stubs/connection.pyi b/asyncpg-stubs/connection.pyi
@@ -375,6 +375,7 @@ async def connect(
     record_class: type[_Record],
     server_settings: dict[str, str] | None = ...,
     target_session_attrs: connect_utils.SessionAttribute | None = ...,
+    krbsrvname: str | None = ...,
 ) -> Connection[_Record]: ...
 @overload
 async def connect(
@@ -398,6 +399,7 @@ async def connect(
     record_class: type[_Record] = ...,
     server_settings: dict[str, str] | None = ...,
     target_session_attrs: connect_utils.SessionAttribute | None = ...,
+    krbsrvname: str | None = ...,
 ) -> _Connection: ...
 @overload
 async def connect(
@@ -419,6 +421,7 @@ async def connect(
     direct_tls: bool = ...,
     server_settings: dict[str, str] | None = ...,
     target_session_attrs: connect_utils.SessionAttribute | None = ...,
+    krbsrvname: str | None = ...,
 ) -> Connection[protocol.Record]: ...
 
 class _ConnectionProxy(Generic[_Record]):

diff --git a/asyncpg-stubs/protocol/protocol.pyi b/asyncpg-stubs/protocol/protocol.pyi
@@ -107,7 +107,7 @@ class CoreProtocol:
     backend_pid: Any
     backend_secret: Any
     __pyx_vtable__: Any
-    def __init__(self, con_params: _ConnectionParameters) -> None: ...
+    def __init__(self, addr: object, con_params: _ConnectionParameters) -> None: ...
     def is_in_transaction(self) -> bool: ...
     def __reduce__(self) -> Any: ...