[hvac] Add argument types for Kv, KvV1, KvV2 & Kubernetes (#16093)

fix(hvac): add argument types for kv & kubernetes

Author: pabepadu

stubs/hvac/hvac/api/auth_methods/kubernetes.pyi

@@ -5,30 +5,30 @@ DEFAULT_MOUNT_POINT: str
 class Kubernetes(VaultApiBase):
     def configure(
         self,
-        kubernetes_host,
-        kubernetes_ca_cert=None,
-        token_reviewer_jwt=None,
-        pem_keys=None,
-        issuer=None,
-        mount_point="kubernetes",
+        kubernetes_host: str,
+        kubernetes_ca_cert: str | None = None,
+        token_reviewer_jwt: str | None = None,
+        pem_keys: list[str] | None = None,
+        issuer: str | None = None,
+        mount_point: str = "kubernetes",
         disable_local_ca_jwt: bool = False,
     ): ...
-    def read_config(self, mount_point="kubernetes"): ...
+    def read_config(self, mount_point: str = "kubernetes"): ...
     def create_role(
         self,
-        name,
-        bound_service_account_names,
-        bound_service_account_namespaces,
-        ttl=None,
-        max_ttl=None,
-        period=None,
-        policies=None,
+        name: str,
+        bound_service_account_names: list[str] | str,
+        bound_service_account_namespaces: list[str] | str,
+        ttl: str | None = None,
+        max_ttl: str | None = None,
+        period: str | None = None,
+        policies: list[str] | str | None = None,
         token_type: str = "",
-        mount_point="kubernetes",
-        alias_name_source=None,
+        mount_point: str = "kubernetes",
+        alias_name_source: str | None = None,
         audience: str | None = None,
     ): ...
-    def read_role(self, name, mount_point="kubernetes"): ...
-    def list_roles(self, mount_point="kubernetes"): ...
-    def delete_role(self, name, mount_point="kubernetes"): ...
-    def login(self, role, jwt, use_token: bool = True, mount_point="kubernetes"): ...
+    def read_role(self, name: str, mount_point: str = "kubernetes"): ...
+    def list_roles(self, mount_point: str = "kubernetes"): ...
+    def delete_role(self, name: str, mount_point: str = "kubernetes"): ...
+    def login(self, role: str, jwt: str, use_token: bool = True, mount_point: str = "kubernetes"): ...

stubs/hvac/hvac/api/secrets_engines/kv.pyi

@@ -1,13 +1,15 @@
 import logging
+from typing import Any
 
+from hvac.adapters import Adapter
 from hvac.api.secrets_engines import KvV1, KvV2
 from hvac.api.vault_api_base import VaultApiBase
 
 logger: logging.Logger
 
 class Kv(VaultApiBase):
     allowed_kv_versions: list[str]
-    def __init__(self, adapter, default_kv_version: str = "2") -> None: ...
+    def __init__(self, adapter: Adapter[Any], default_kv_version: str = "2") -> None: ...
     @property
     def v1(self) -> KvV1: ...
     @property

stubs/hvac/hvac/api/secrets_engines/kv_v1.pyi

@@ -3,7 +3,9 @@ from hvac.api.vault_api_base import VaultApiBase
 DEFAULT_MOUNT_POINT: str
 
 class KvV1(VaultApiBase):
-    def read_secret(self, path, mount_point="secret"): ...
-    def list_secrets(self, path, mount_point="secret"): ...
-    def create_or_update_secret(self, path, secret, method=None, mount_point="secret"): ...
-    def delete_secret(self, path, mount_point="secret"): ...
+    def read_secret(self, path: str, mount_point: str = "secret"): ...
+    def list_secrets(self, path: str, mount_point: str = "secret"): ...
+    def create_or_update_secret(
+        self, path: str, secret: dict[str, str], method: str | None = None, mount_point: str = "secret"
+    ): ...
+    def delete_secret(self, path: str, mount_point: str = "secret"): ...

stubs/hvac/hvac/api/secrets_engines/kv_v2.pyi

@@ -3,25 +3,33 @@ from hvac.api.vault_api_base import VaultApiBase
 DEFAULT_MOUNT_POINT: str
 
 class KvV2(VaultApiBase):
-    def configure(self, max_versions: int = 10, cas_required=None, delete_version_after: str = "0s", mount_point="secret"): ...
-    def read_configuration(self, mount_point="secret"): ...
-    def read_secret(self, path, mount_point="secret", raise_on_deleted_version=None): ...
-    def read_secret_version(self, path, version=None, mount_point="secret", raise_on_deleted_version=None): ...
-    def create_or_update_secret(self, path, secret, cas=None, mount_point="secret"): ...
-    def patch(self, path, secret, mount_point="secret"): ...
-    def delete_latest_version_of_secret(self, path, mount_point="secret"): ...
-    def delete_secret_versions(self, path, versions, mount_point="secret"): ...
-    def undelete_secret_versions(self, path, versions, mount_point="secret"): ...
-    def destroy_secret_versions(self, path, versions, mount_point="secret"): ...
-    def list_secrets(self, path, mount_point="secret"): ...
-    def read_secret_metadata(self, path, mount_point="secret"): ...
+    def configure(
+        self,
+        max_versions: int = 10,
+        cas_required: bool | None = None,
+        delete_version_after: str = "0s",
+        mount_point: str = "secret",
+    ): ...
+    def read_configuration(self, mount_point: str = "secret"): ...
+    def read_secret(self, path: str, mount_point: str = "secret", raise_on_deleted_version: bool | None = None): ...
+    def read_secret_version(
+        self, path: str, version: int | None = None, mount_point: str = "secret", raise_on_deleted_version: bool | None = None
+    ): ...
+    def create_or_update_secret(self, path: str, secret: dict[str, str], cas: int | None = None, mount_point: str = "secret"): ...
+    def patch(self, path: str, secret: dict[str, str], mount_point: str = "secret"): ...
+    def delete_latest_version_of_secret(self, path: str, mount_point: str = "secret"): ...
+    def delete_secret_versions(self, path: str, versions: list[int], mount_point: str = "secret"): ...
+    def undelete_secret_versions(self, path: str, versions: list[int], mount_point: str = "secret"): ...
+    def destroy_secret_versions(self, path: str, versions: list[int], mount_point: str = "secret"): ...
+    def list_secrets(self, path: str, mount_point: str = "secret"): ...
+    def read_secret_metadata(self, path: str, mount_point: str = "secret"): ...
     def update_metadata(
         self,
-        path,
-        max_versions=None,
-        cas_required=None,
+        path: str,
+        max_versions: int | None = None,
+        cas_required: bool | None = None,
         delete_version_after: str = "0s",
-        mount_point="secret",
-        custom_metadata=None,
+        mount_point: str = "secret",
+        custom_metadata: dict[str, str] | None = None,
     ): ...
-    def delete_metadata_and_all_versions(self, path, mount_point="secret"): ...
+    def delete_metadata_and_all_versions(self, path: str, mount_point: str = "secret"): ...

stubs/hvac/hvac/api/vault_api_base.pyi

@@ -1,10 +1,10 @@
-from _typeshed import Incomplete
 from abc import ABCMeta
 from logging import Logger
+from typing import Any
 
 from hvac.adapters import Adapter
 
 logger: Logger
 
 class VaultApiBase(metaclass=ABCMeta):
-    def __init__(self, adapter: Adapter[Incomplete]) -> None: ...
+    def __init__(self, adapter: Adapter[Any]) -> None: ...

stubs/hvac/hvac/api/vault_api_category.pyi

@@ -10,7 +10,7 @@ logger: Logger
 class VaultApiCategory(VaultApiBase, metaclass=ABCMeta):
     implemented_class_names: list[str]
     def __init__(self, adapter: Adapter[Any]) -> None: ...
-    def __getattr__(self, item): ...
+    def __getattr__(self, item: str): ...
 
     @property
     def adapter(self) -> Adapter[Any]: ...
@@ -19,8 +19,8 @@ class VaultApiCategory(VaultApiBase, metaclass=ABCMeta):
 
     @property
     @abstractmethod
-    def implemented_classes(self): ...
+    def implemented_classes(self) -> list[type[VaultApiBase]]: ...
     @property
     def unimplemented_classes(self) -> list[str]: ...
     @staticmethod
-    def get_private_attr_name(class_name): ...
+    def get_private_attr_name(class_name: str) -> str: ...