Run mypy in azure-keyvault-secrets CI

eng/tox/mypy_hard_failure_packages.py

@@ -10,6 +10,7 @@
   "azure-eventhub",
   "azure-identity",
   "azure-keyvault-administration",
+  "azure-keyvault-secrets",
   "azure-servicebus",
   "azure-ai-textanalytics",
   "azure-ai-formrecognizer",

sdk/keyvault/azure-keyvault-certificates/azure/keyvault/certificates/_shared/_polling.py

@@ -33,11 +33,11 @@ class KeyVaultOperationPoller(LROPoller):
     # pylint: disable=arguments-differ
     def __init__(self, polling_method):
         # type: (PollingMethod) -> None
-        super(KeyVaultOperationPoller, self).__init__(None, None, None, NoPolling())
+        super(KeyVaultOperationPoller, self).__init__(None, None, lambda *_: None, NoPolling())
         self._polling_method = polling_method
 
     # pylint: disable=arguments-differ
-    def result(self):
+    def result(self):  # type: ignore
         # type: () -> Any
         """Returns a representation of the final resource without waiting for the operation to complete.
 

sdk/keyvault/azure-keyvault-keys/azure/keyvault/keys/_shared/_polling.py

@@ -33,11 +33,11 @@ class KeyVaultOperationPoller(LROPoller):
     # pylint: disable=arguments-differ
     def __init__(self, polling_method):
         # type: (PollingMethod) -> None
-        super(KeyVaultOperationPoller, self).__init__(None, None, None, NoPolling())
+        super(KeyVaultOperationPoller, self).__init__(None, None, lambda *_: None, NoPolling())
         self._polling_method = polling_method
 
     # pylint: disable=arguments-differ
-    def result(self):
+    def result(self):  # type: ignore
         # type: () -> Any
         """Returns a representation of the final resource without waiting for the operation to complete.
 

sdk/keyvault/azure-keyvault-secrets/azure/keyvault/secrets/_client.py

@@ -4,6 +4,7 @@
 # ------------------------------------
 from functools import partial
 from azure.core.tracing.decorator import distributed_trace
+from azure.core.polling import LROPoller
 
 from ._models import KeyVaultSecret, DeletedSecret, SecretProperties
 from ._shared import KeyVaultClientBase
@@ -294,7 +295,7 @@ def restore_secret_backup(self, backup, **kwargs):
 
     @distributed_trace
     def begin_delete_secret(self, name, **kwargs):
-        # type: (str, **Any) -> DeletedSecret
+        # type: (str, **Any) -> LROPoller
         """Delete all versions of a secret. Requires secrets/delete permission.
 
         When this method returns Key Vault has begun deleting the secret. Deletion may take several seconds in a vault
@@ -416,7 +417,7 @@ def purge_deleted_secret(self, name, **kwargs):
 
     @distributed_trace
     def begin_recover_deleted_secret(self, name, **kwargs):
-        # type: (str, **Any) -> SecretProperties
+        # type: (str, **Any) -> LROPoller
         """Recover a deleted secret to its latest version. Possible only in a vault with soft-delete enabled.
 
         If the vault does not have soft-delete enabled, :func:`begin_delete_secret` is permanent, and this method will

sdk/keyvault/azure-keyvault-secrets/azure/keyvault/secrets/_models.py

@@ -11,19 +11,19 @@
 
 if TYPE_CHECKING:
     # pylint:disable=unused-import
-    from typing import Any, Dict, Optional
+    from typing import Any, Dict, Optional, Union
     from datetime import datetime
     from ._generated.v7_1 import models as _models
 
 
 class SecretProperties(object):
     """A secret's id and attributes."""
 
-    def __init__(self, attributes, vault_id, **kwargs):
-        # type: (_models.SecretAttributes, str, **Any) -> None
+    def __init__(self, attributes=None, vault_id=None, **kwargs):
+        # type: (Optional[_models.SecretAttributes], Optional[str], **Any) -> None
         self._attributes = attributes
         self._id = vault_id
-        self._vault_id = KeyVaultSecretIdentifier(vault_id)
+        self._vault_id = KeyVaultSecretIdentifier(vault_id) if vault_id else None
         self._content_type = kwargs.get("content_type", None)
         self._key_id = kwargs.get("key_id", None)
         self._managed = kwargs.get("managed", None)
@@ -69,10 +69,10 @@ def content_type(self):
 
     @property
     def id(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's id
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._id
 
@@ -87,55 +87,55 @@ def key_id(self):
 
     @property
     def enabled(self):
-        # type: () -> bool
+        # type: () -> Union[bool, None]
         """Whether the secret is enabled for use
 
-        :rtype: bool
+        :rtype: union[bool, None]
         """
-        return self._attributes.enabled
+        return self._attributes.enabled if self._attributes else None
 
     @property
     def not_before(self):
-        # type: () -> datetime
+        # type: () -> Union[datetime, None]
         """The time before which the secret can not be used, in UTC
 
-        :rtype: ~datetime.datetime
+        :rtype: union[~datetime.datetime, None]
         """
-        return self._attributes.not_before
+        return self._attributes.not_before if self._attributes else None
 
     @property
     def expires_on(self):
-        # type: () -> datetime
+        # type: () -> Union[datetime, None]
         """When the secret expires, in UTC
 
-        :rtype: ~datetime.datetime
+        :rtype: union[~datetime.datetime, None]
         """
-        return self._attributes.expires
+        return self._attributes.expires if self._attributes else None
 
     @property
     def created_on(self):
-        # type: () -> datetime
+        # type: () -> Union[datetime, None]
         """When the secret was created, in UTC
 
-        :rtype: ~datetime.datetime
+        :rtype: union[~datetime.datetime, None]
         """
-        return self._attributes.created
+        return self._attributes.created if self._attributes else None
 
     @property
     def updated_on(self):
-        # type: () -> datetime
+        # type: () -> Union[datetime, None]
         """When the secret was last updated, in UTC
 
-        :rtype: ~datetime.datetime
+        :rtype: union[~datetime.datetime, None]
         """
-        return self._attributes.updated
+        return self._attributes.updated if self._attributes else None
 
     @property
     def recoverable_days(self):
         # type: () -> Optional[int]
         """The number of days the key is retained before being deleted from a soft-delete enabled Key Vault.
 
-        :rtype: int
+        :rtype: optional[int]
         """
         # recoverable_days was added in 7.1-preview
         if self._attributes and hasattr(self._attributes, "recoverable_days"):
@@ -144,39 +144,39 @@ def recoverable_days(self):
 
     @property
     def recovery_level(self):
-        # type: () -> str
+        # type: () -> Union[str, None]
         """The vault's deletion recovery level for secrets
 
-        :rtype: str
+        :rtype: union[str, None]
         """
-        return self._attributes.recovery_level
+        return self._attributes.recovery_level if self._attributes else None
 
     @property
     def vault_url(self):
-        # type: () -> str
+        # type: () -> Union[str, None]
         """URL of the vault containing the secret
 
-        :rtype: str
+        :rtype: union[str, None]
         """
-        return self._vault_id.vault_url
+        return self._vault_id.vault_url if self._vault_id else None
 
     @property
     def name(self):
-        # type: () -> str
+        # type: () -> Union[str, None]
         """The secret's name
 
-        :rtype: str
+        :rtype: union[str, None]
         """
-        return self._vault_id.name
+        return self._vault_id.name if self._vault_id else None
 
     @property
     def version(self):
-        # type: () -> str
+        # type: () -> Union[str, None]
         """The secret's version
 
-        :rtype: str
+        :rtype: union[str, None]
         """
-        return self._vault_id.version
+        return self._vault_id.version if self._vault_id else None
 
     @property
     def tags(self):
@@ -191,7 +191,7 @@ class KeyVaultSecret(object):
     """All of a secret's properties, and its value."""
 
     def __init__(self, properties, value):
-        # type: (SecretProperties, str) -> None
+        # type: (SecretProperties, Optional[str]) -> None
         self._properties = properties
         self._value = value
 
@@ -210,19 +210,19 @@ def _from_secret_bundle(cls, secret_bundle):
 
     @property
     def name(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's name
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._properties.name
 
     @property
     def id(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's id
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._properties.id
 
@@ -237,10 +237,10 @@ def properties(self):
 
     @property
     def value(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's value
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._value
 
@@ -329,19 +329,19 @@ def _from_deleted_secret_item(cls, deleted_secret_item):
 
     @property
     def name(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's name
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._properties.name
 
     @property
     def id(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """The secret's id
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._properties.id
 
@@ -356,27 +356,27 @@ def properties(self):
 
     @property
     def deleted_date(self):
-        # type: () -> datetime
+        # type: () -> Optional[datetime]
         """When the secret was deleted, in UTC
 
-        :rtype: ~datetime.datetime
+        :rtype: optional[~datetime.datetime]
         """
         return self._deleted_date
 
     @property
     def recovery_id(self):
-        # type: () -> str
+        # type: () -> Optional[str]
         """An identifier used to recover the deleted secret. Returns ``None`` if soft-delete is disabled.
 
-        :rtype: str
+        :rtype: optional[str]
         """
         return self._recovery_id
 
     @property
     def scheduled_purge_date(self):
-        # type: () -> datetime
+        # type: () -> Optional[datetime]
         """When the secret is scheduled to be purged, in UTC. Returns ``None`` if soft-delete is disabled.
 
-        :rtype: ~datetime.datetime
+        :rtype: optional[~datetime.datetime]
         """
         return self._scheduled_purge_date

sdk/keyvault/azure-keyvault-secrets/azure/keyvault/secrets/_shared/_polling.py

@@ -33,11 +33,11 @@ class KeyVaultOperationPoller(LROPoller):
     # pylint: disable=arguments-differ
     def __init__(self, polling_method):
         # type: (PollingMethod) -> None
-        super(KeyVaultOperationPoller, self).__init__(None, None, None, NoPolling())
+        super(KeyVaultOperationPoller, self).__init__(None, None, lambda *_: None, NoPolling())
         self._polling_method = polling_method
 
     # pylint: disable=arguments-differ
-    def result(self):
+    def result(self):  # type: ignore
         # type: () -> Any
         """Returns a representation of the final resource without waiting for the operation to complete.
 

sdk/keyvault/azure-keyvault-secrets/mypy.ini

@@ -0,0 +1,7 @@
+[mypy]
+python_version = 3.6
+warn_unused_configs = True
+ignore_missing_imports = True
+
+[mypy-azure.keyvault.*._generated.*]
+ignore_errors = True