[Fileshare] Added support for set share properties including access tier

sdk/storage/azure-storage-file-share/azure/storage/fileshare/__init__.py

@@ -28,13 +28,15 @@
     ShareSmbSettings,
     SmbMultichannel,
     ShareProtocolSettings,
+    ShareSetPropertiesOptions,
     AccessPolicy,
     FileSasPermissions,
     ShareSasPermissions,
     ContentSettings,
     NTFSAttributes)
 from ._generated.models import (
-    HandleItem
+    HandleItem,
+    ShareAccessTier
 )
 
 __version__ = VERSION
@@ -56,8 +58,10 @@
     'RetentionPolicy',
     'CorsRule',
     'ShareSmbSettings',
+    'ShareAccessTier',
     'SmbMultichannel',
     'ShareProtocolSettings',
+    'ShareSetPropertiesOptions',
     'AccessPolicy',
     'FileSasPermissions',
     'ShareSasPermissions',

sdk/storage/azure-storage-file-share/azure/storage/fileshare/_models.py

@@ -171,6 +171,22 @@ def _from_generated(cls, generated):
             smb=generated.smb)
 
 
+class ShareSetPropertiesOptions(object):
+    """Sets the properties for the share.
+
+    :param access_tier:
+        Specifies the access tier of the share.
+        Possible values: 'TransactionOptimized', 'Hot', and 'Cool'
+    :type access_tier: str or ~azure.storage.fileshare.models.ShareAccessTier
+    :param int quota:
+        Specifies the maximum size of the share, in gigabytes.
+        Must be greater than 0, and less than or equal to 5TB.
+    """
+    def __init__(self, quota=None, access_tier=None):
+        self.quota = quota
+        self.access_tier = access_tier
+
+
 class AccessPolicy(GenAccessPolicy):
     """Access Policy class used by the set and get acl methods in each service.
 
@@ -308,6 +324,8 @@ class ShareProperties(DictMixin):
         conditionally.
     :ivar int quota:
         The allocated quota.
+    :ivar str access_tier:
+        The share's access tier.
     :ivar dict metadata: A dict with name_value pairs to associate with the
         share as metadata.
     :ivar str snapshot:
@@ -331,6 +349,7 @@ def __init__(self, **kwargs):
         self.last_modified = kwargs.get('Last-Modified')
         self.etag = kwargs.get('ETag')
         self.quota = kwargs.get('x-ms-share-quota')
+        self.access_tier = kwargs.get('x-ms-access-tier')
         self.next_allowed_quota_downgrade_time = kwargs.get('x-ms-share-next-allowed-quota-downgrade-time')
         self.metadata = kwargs.get('metadata')
         self.snapshot = None
@@ -350,6 +369,7 @@ def _from_generated(cls, generated):
         props.last_modified = generated.properties.last_modified
         props.etag = generated.properties.etag
         props.quota = generated.properties.quota
+        props.access_tier = generated.properties.access_tier
         props.next_allowed_quota_downgrade_time = generated.properties.next_allowed_quota_downgrade_time
         props.metadata = generated.metadata
         props.snapshot = generated.snapshot

sdk/storage/azure-storage-file-share/azure/storage/fileshare/_share_client.py

@@ -36,7 +36,7 @@
 from ._lease import ShareLeaseClient
 
 if TYPE_CHECKING:
-    from ._models import ShareProperties, AccessPolicy
+    from ._models import ShareProperties, AccessPolicy, ShareSetPropertiesOptions
 
 
 class ShareClient(StorageAccountHostsMixin):
@@ -305,6 +305,13 @@ def create_share(self, **kwargs):
             Name-value pairs associated with the share as metadata.
         :keyword int quota:
             The quota to be allotted.
+        :keyword access_tier:
+            Specifies the access tier of the share.
+            Possible values: 'TransactionOptimized', 'Hot', 'Cool'
+        :paramtype access_tier: str or ~azure.storage.fileshare.models.ShareAccessTier
+
+            .. versionadded:: 12.6.0
+
         :keyword int timeout:
             The timeout parameter is expressed in seconds.
         :returns: Share-updated property dict (Etag and last modified).
@@ -321,6 +328,7 @@ def create_share(self, **kwargs):
         """
         metadata = kwargs.pop('metadata', None)
         quota = kwargs.pop('quota', None)
+        access_tier = kwargs.pop('access_tier', None)
         timeout = kwargs.pop('timeout', None)
         headers = kwargs.pop('headers', {})
         headers.update(add_metadata_headers(metadata)) # type: ignore
@@ -330,6 +338,7 @@ def create_share(self, **kwargs):
                 timeout=timeout,
                 metadata=metadata,
                 quota=quota,
+                access_tier=access_tier,
                 cls=return_response_headers,
                 headers=headers,
                 **kwargs)
@@ -504,6 +513,47 @@ def set_share_quota(self, quota, **kwargs):
             return self._client.share.set_properties( # type: ignore
                 timeout=timeout,
                 quota=quota,
+                access_tier=None,
+                lease_access_conditions=access_conditions,
+                cls=return_response_headers,
+                **kwargs)
+        except StorageErrorException as error:
+            process_storage_error(error)
+
+    @distributed_trace
+    def set_share_properties(self, options, **kwargs):
+        # type: (ShareSetPropertiesOptions, Any) ->  Dict[str, Any]
+        """Sets the share properties.
+
+        .. versionadded:: 12.6.0
+
+        :param options:
+            Specifies the properties to set on the share.
+        :type options: ~azure.storage.fileshare.models.ShareSetPropertiesOptions
+        :keyword int timeout:
+            The timeout parameter is expressed in seconds.
+        :keyword lease:
+            Required if the share has an active lease. Value can be a ShareLeaseClient object
+            or the lease ID as a string.
+        :returns: Share-updated property dict (Etag and last modified).
+        :rtype: dict(str, Any)
+
+        .. admonition:: Example:
+
+            .. literalinclude:: ../samples/file_samples_share.py
+                :start-after: [START set_share_properties]
+                :end-before: [END set_share_properties]
+                :language: python
+                :dedent: 12
+                :caption: Sets the share properties.
+        """
+        access_conditions = get_access_conditions(kwargs.pop('lease', None))
+        timeout = kwargs.pop('timeout', None)
+        try:
+            return self._client.share.set_properties( # type: ignore
+                timeout=timeout,
+                quota=options.quota,
+                access_tier=options.access_tier,
                 lease_access_conditions=access_conditions,
                 cls=return_response_headers,
                 **kwargs)

sdk/storage/azure-storage-file-share/azure/storage/fileshare/aio/_share_client_async.py

@@ -33,7 +33,7 @@
 
 
 if TYPE_CHECKING:
-    from .._models import ShareProperties, AccessPolicy
+    from .._models import ShareProperties, AccessPolicy, ShareSetPropertiesOptions
 
 
 class ShareClient(AsyncStorageAccountHostsMixin, ShareClientBase):
@@ -176,6 +176,13 @@ async def create_share(self, **kwargs):
             Name-value pairs associated with the share as metadata.
         :keyword int quota:
             The quota to be allotted.
+        :keyword access_tier:
+            Specifies the access tier of the share.
+            Possible values: 'TransactionOptimized', 'Hot', 'Cool'
+        :paramtype access_tier: str or ~azure.storage.fileshare.models.ShareAccessTier
+
+            .. versionadded:: 12.6.0
+
         :keyword int timeout:
             The timeout parameter is expressed in seconds.
         :returns: Share-updated property dict (Etag and last modified).
@@ -192,6 +199,7 @@ async def create_share(self, **kwargs):
         """
         metadata = kwargs.pop('metadata', None)
         quota = kwargs.pop('quota', None)
+        access_tier = kwargs.pop('access_tier', None)
         timeout = kwargs.pop('timeout', None)
         headers = kwargs.pop('headers', {})
         headers.update(add_metadata_headers(metadata)) # type: ignore
@@ -201,6 +209,7 @@ async def create_share(self, **kwargs):
                 timeout=timeout,
                 metadata=metadata,
                 quota=quota,
+                access_tier=access_tier,
                 cls=return_response_headers,
                 headers=headers,
                 **kwargs)
@@ -374,12 +383,52 @@ async def set_share_quota(self, quota, **kwargs):
             return await self._client.share.set_properties( # type: ignore
                 timeout=timeout,
                 quota=quota,
+                access_tier=None,
                 cls=return_response_headers,
                 lease_access_conditions=access_conditions,
                 **kwargs)
         except StorageErrorException as error:
             process_storage_error(error)
 
+    async def set_share_properties(self, options, **kwargs):
+        # type: (ShareSetPropertiesOptions, Any) ->  Dict[str, Any]
+        """Sets the share properties.
+
+        .. versionadded:: 12.6.0
+
+        :param options:
+            Specifies the properties to set on the share.
+        :type options: ~azure.storage.fileshare.models.ShareSetPropertiesOptions
+        :keyword int timeout:
+            The timeout parameter is expressed in seconds.
+        :keyword lease:
+            Required if the share has an active lease. Value can be a ShareLeaseClient object
+            or the lease ID as a string.
+        :returns: Share-updated property dict (Etag and last modified).
+        :rtype: dict(str, Any)
+
+        .. admonition:: Example:
+
+            .. literalinclude:: ../samples/file_samples_share_async.py
+                :start-after: [START set_share_properties]
+                :end-before: [END set_share_properties]
+                :language: python
+                :dedent: 16
+                :caption: Sets the share properties.
+        """
+        access_conditions = get_access_conditions(kwargs.pop('lease', None))
+        timeout = kwargs.pop('timeout', None)
+        try:
+            return await self._client.share.set_properties( # type: ignore
+                timeout=timeout,
+                quota=options.quota,
+                access_tier=options.access_tier,
+                lease_access_conditions=access_conditions,
+                cls=return_response_headers,
+                **kwargs)
+        except StorageErrorException as error:
+            process_storage_error(error)
+
     @distributed_trace_async
     async def set_share_metadata(self, metadata, **kwargs):
         # type: (Dict[str, Any], Any) ->  Dict[str, Any]

sdk/storage/azure-storage-file-share/samples/file_samples_share.py

@@ -22,6 +22,7 @@
 """
 
 import os
+from azure.storage.fileshare import ShareSetPropertiesOptions, ShareAccessTier
 
 SOURCE_FILE = './SampleSource.txt'
 DEST_FILE = './SampleDestination.txt'
@@ -37,7 +38,8 @@ def create_share_snapshot(self):
         share = ShareClient.from_connection_string(self.connection_string, "sharesamples1")
 
         # [START create_share]
-        share.create_share()
+        # Create share with Access Tier set to Hot
+        share.create_share(access_tier=ShareAccessTier("Hot"))
         # [END create_share]
         try:
             # [START create_share_snapshot]
@@ -75,10 +77,40 @@ def set_share_quota_and_metadata(self):
             # Delete the share
             share.delete_share()
 
+    def set_share_properties(self):
+        from azure.storage.fileshare import ShareClient
+        share1 = ShareClient.from_connection_string(self.connection_string, "sharesamples3a")
+        share2 = ShareClient.from_connection_string(self.connection_string, "sharesamples3b")
+
+        # Create the share
+        share1.create_share()
+        share2.create_share()
+
+        try:
+            # [START set_share_properties]
+            # Set the tier for the first share to Hot
+            share1.set_share_properties(ShareSetPropertiesOptions(access_tier="Hot"))
+            # Set the quota for the first share to 3
+            share1.set_share_properties(ShareSetPropertiesOptions(quota=3))
+            # Set the tier for the second share to Cool and quota to 2
+            share2.set_share_properties(ShareSetPropertiesOptions(access_tier=ShareAccessTier("Cool"), quota=2))
+
+            # Get the shares' properties
+            print(share1.get_share_properties().access_tier)
+            print(share1.get_share_properties().quota)
+            print(share2.get_share_properties().access_tier)
+            print(share2.get_share_properties().quota)
+            # [END set_share_properties]
+
+        finally:
+            # Delete the shares
+            share1.delete_share()
+            share2.delete_share()
+
     def list_directories_and_files(self):
         # Instantiate the ShareClient from a connection string
         from azure.storage.fileshare import ShareClient
-        share = ShareClient.from_connection_string(self.connection_string, "sharesamples3")
+        share = ShareClient.from_connection_string(self.connection_string, "sharesamples4")
 
         # Create the share
         share.create_share()
@@ -103,7 +135,7 @@ def list_directories_and_files(self):
     def get_directory_or_file_client(self):
         # Instantiate the ShareClient from a connection string
         from azure.storage.fileshare import ShareClient
-        share = ShareClient.from_connection_string(self.connection_string, "sharesamples4")
+        share = ShareClient.from_connection_string(self.connection_string, "sharesamples5")
 
         # Get the directory client to interact with a specific directory
         my_dir = share.get_directory_client("dir1")
@@ -131,6 +163,7 @@ def acquire_share_lease(self):
     sample = ShareSamples()
     sample.create_share_snapshot()
     sample.set_share_quota_and_metadata()
+    sample.set_share_properties()
     sample.list_directories_and_files()
     sample.get_directory_or_file_client()
     sample.acquire_share_lease()