added async methods

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

@@ -14,12 +14,13 @@
 from .._shared.response_handlers import return_response_headers, process_storage_error
 from .._generated.models import (
     StorageErrorException)
+from .._generated.operations import FileOperations
 from .._lease import ShareLeaseClient as LeaseClientBase
 
 if TYPE_CHECKING:
     from datetime import datetime
     ShareFileClient = TypeVar("ShareFileClient")
-
+    ShareClient = TypeVar("ShareClient")
 
 class ShareLeaseClient(LeaseClientBase):
     """Creates a new ShareLeaseClient.
@@ -57,17 +58,23 @@ async def __aexit__(self, *args):
         await self.release()
 
     @distributed_trace_async
-    async def acquire(self, **kwargs):
-        # type: (int, Any) -> None
+    async def acquire(self, lease_duration=-1, **kwargs):
+        # type: (int, **Any) -> None
         """Requests a new lease. This operation establishes and manages a lock on a
-        file for write and delete operations. If the file does not have an active lease,
-        the File service creates a lease on the file. If the file has an active lease,
+        file or share for write and delete operations. If the file or share does not have an active lease,
+        the File or Share service creates a lease on the file or share. If the file has an active lease,
         you can only request a new lease using the active lease ID.
 
 
-        If the file does not have an active lease, the File service creates a
+        If the file or share does not have an active lease, the File or Share service creates a
         lease on the file and returns a new lease ID.
 
+        :param int lease_duration:
+            Specifies the duration of the lease, in seconds, or negative one
+            (-1) for a lease that never expires. File leases never expire. A non-infinite share lease can be
+            between 15 and 60 seconds. A share lease duration cannot be changed
+            using renew or change. Default is -1 (infinite share lease).
+
         :keyword int timeout:
             The timeout parameter is expressed in seconds.
         :rtype: None
@@ -78,24 +85,68 @@ async def acquire(self, **kwargs):
                 duration=-1,
                 proposed_lease_id=self.id,
                 cls=return_response_headers,
+                **kwargs) if isinstance(self._client, FileOperations) \
+                else self._client.acquire_lease(
+                timeout=kwargs.pop('timeout', None),
+                duration=lease_duration,
+                sharesnapshot=self.snapshot,
+                proposed_lease_id=self.id,
+                cls=return_response_headers,
                 **kwargs)
+
         except StorageErrorException as error:
             process_storage_error(error)
         self.id = response.get('lease_id')  # type: str
         self.last_modified = response.get('last_modified')   # type: datetime
         self.etag = response.get('etag')  # type: str
 
     @distributed_trace_async
-    async def release(self, **kwargs):
+    async def renew(self, **kwargs):
         # type: (Any) -> None
-        """Releases the lease. The lease may be released if the lease ID specified on the request matches
-        that associated with the file. Releasing the lease allows another client to immediately acquire the lease
-        for the file as soon as the release is complete.
+        """Renews the share lease.
+
+        The share lease can be renewed if the lease ID specified in the
+        lease client matches that associated with the share. Note that
+        the lease may be renewed even if it has expired as long as the share
+        has not been leased again since the expiration of that lease. When you
+        renew a lease, the lease duration clock resets.
+
+        :keyword str etag:
+            An ETag value, or the wildcard character (*). Used to check if the resource has changed,
+            and act according to the condition specified by the `match_condition` parameter.
+        :keyword ~azure.core.MatchConditions match_condition:
+            The match condition to use upon the etag.
+        :keyword str if_tags_match_condition
+            Specify a SQL where clause on blob tags to operate only on blob with a matching value.
+            eg. "\"tagname\"='my tag'"
 
+            .. versionadded:: 12.4.0
+
+        :keyword int timeout:
+            The timeout parameter is expressed in seconds.
+        :return: None
+        """
+        if isinstance(self._client, FileOperations):
+            raise TypeError("Lease renewal operations are only valid for ShareClient.")
+        try:
+            response = await self._client.renew_lease(
+                lease_id=self.id,
+                timeout=kwargs.pop('timeout', None),
+                sharesnapshot=self.snapshot,
+                cls=return_response_headers,
+                **kwargs)
+        except StorageErrorException as error:
+            process_storage_error(error)
+        self.etag = response.get('etag')  # type: str
+        self.id = response.get('lease_id')  # type: str
+        self.last_modified = response.get('last_modified')   # type: datetime
 
-        The lease may be released if the client lease id specified matches
-        that associated with the file. Releasing the lease allows another client
-        to immediately acquire the lease for the file as soon as the release is complete.
+    @distributed_trace_async
+    async def release(self, **kwargs):
+        # type: (Any) -> None
+        """Releases the lease. The lease may be released if the lease ID specified on the request matches
+        that associated with the share or file. Releasing the lease allows another client to immediately acquire the lease
+        for the share or file as soon as the release is complete.
 
         :keyword int timeout:
             The timeout parameter is expressed in seconds.
@@ -106,7 +157,14 @@ async def release(self, **kwargs):
                 lease_id=self.id,
                 timeout=kwargs.pop('timeout', None),
                 cls=return_response_headers,
+                **kwargs) if isinstance(self._client, FileOperations) \
+                else self._client.renew_lease(
+                lease_id=self.id,
+                timeout=kwargs.pop('timeout', None),
+                sharesnapshot=self.snapshot,
+                cls=return_response_headers,
                 **kwargs)
+
         except StorageErrorException as error:
             process_storage_error(error)
         self.etag = response.get('etag')  # type: str
@@ -121,7 +179,7 @@ async def change(self, proposed_lease_id, **kwargs):
 
 
         :param str proposed_lease_id:
-            Proposed lease ID, in a GUID string format. The File service returns 400
+            Proposed lease ID, in a GUID string format. The File or Share service returns 400
             (Invalid request) if the proposed lease ID is not in the correct format.
         :keyword int timeout:
             The timeout parameter is expressed in seconds.
@@ -133,7 +191,15 @@ async def change(self, proposed_lease_id, **kwargs):
                 proposed_lease_id=proposed_lease_id,
                 timeout=kwargs.pop('timeout', None),
                 cls=return_response_headers,
+                **kwargs) if isinstance(self._client, FileOperations) \
+                else self._client.change_lease(
+                lease_id=self.id,
+                proposed_lease_id=proposed_lease_id,
+                timeout=kwargs.pop('timeout', None),
+                sharesnapshot=self.snapshot,
+                cls=return_response_headers,
                 **kwargs)
+
         except StorageErrorException as error:
             process_storage_error(error)
         self.etag = response.get('etag')  # type: str
@@ -143,7 +209,7 @@ async def change(self, proposed_lease_id, **kwargs):
     @distributed_trace_async
     async def break_lease(self, **kwargs):
         # type: (Optional[int], Any) -> int
-        """Force breaks the lease if the file has an active lease. Any authorized request can break the lease;
+        """Force breaks the lease if the file or share has an active lease. Any authorized request can break the lease;
         the request is not required to specify a matching lease ID. An infinite lease breaks immediately.
 
         Once a lease is broken, it cannot be changed. Any authorized request can break the lease;
@@ -160,7 +226,13 @@ async def break_lease(self, **kwargs):
             response = await self._client.break_lease(
                 timeout=kwargs.pop('timeout', None),
                 cls=return_response_headers,
+                **kwargs) if isinstance(self._client, FileOperations) \
+                else self._client.break_lease(
+                timeout=kwargs.pop('timeout', None),
+                sharesnapshot=self.snapshot,
+                cls=return_response_headers,
                 **kwargs)
+
         except StorageErrorException as error:
             process_storage_error(error)
         return response.get('lease_time') # type: ignore