[Storage] Fix some mypy typing errors and typing imports (Azure#23098)

Author: jalauzon-msft

sdk/storage/azure-storage-blob/azure/storage/blob/_blob_client.py

@@ -64,6 +64,7 @@
     from ._generated.models import BlockList
     from ._models import (  # pylint: disable=unused-import
         ContentSettings,
+        ImmutabilityPolicy,
         PremiumPageBlobTier,
         StandardBlobTier,
         SequenceNumberAction
@@ -1407,7 +1408,7 @@ def set_blob_metadata(self, metadata=None, **kwargs):
 
     @distributed_trace
     def set_immutability_policy(self, immutability_policy, **kwargs):
-        # type: (azure.storage.blob.ImmutabilityPolicy, **Any) -> Dict[str, str]
+        # type: (ImmutabilityPolicy, **Any) -> Dict[str, str]
         """The Set Immutability Policy operation sets the immutability policy on the blob.
 
         .. versionadded:: 12.10.0

sdk/storage/azure-storage-blob/azure/storage/blob/_serialize.py

@@ -4,6 +4,10 @@
 # license information.
 # --------------------------------------------------------------------------
 # pylint: disable=no-self-use
+from typing import (  # pylint: disable=unused-import
+    Any, Dict, Optional, Tuple, Union,
+    TYPE_CHECKING)
+
 try:
     from urllib.parse import quote
 except ImportError:
@@ -29,6 +33,9 @@
     BlobTags, LeaseAccessConditions
 )
 
+if TYPE_CHECKING:
+    from ._lease import BlobLeaseClient
+
 
 _SUPPORTED_API_VERSIONS = [
     '2019-02-02',

sdk/storage/azure-storage-blob/azure/storage/blob/aio/_blob_client_async.py

@@ -37,6 +37,7 @@
     from datetime import datetime
     from .._models import (  # pylint: disable=unused-import
         ContentSettings,
+        ImmutabilityPolicy,
         PremiumPageBlobTier,
         StandardBlobTier,
         SequenceNumberAction
@@ -806,7 +807,7 @@ async def set_blob_metadata(self, metadata=None, **kwargs):
 
     @distributed_trace_async
     async def set_immutability_policy(self, immutability_policy, **kwargs):
-        # type: (**Any) -> Dict[str, str]
+        # type: (ImmutabilityPolicy, **Any) -> Dict[str, str]
         """The Set Immutability Policy operation sets the immutability policy on the blob.
 
         .. versionadded:: 12.10.0