remove locktoken model

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_operations/_patch.py

@@ -156,7 +156,9 @@ def receive_cloud_events(
         for detail_item in received_result.get("value"):
             deserialized_cloud_event = CloudEvent.from_dict(detail_item.get("event"))
             detail_item["event"] = deserialized_cloud_event
-            detail_items.append(ReceiveDetails(event=detail_item.get("event"), broker_properties=detail_item.get("brokerProperties")))
+            detail_items.append(
+                ReceiveDetails(event=detail_item.get("event"), broker_properties=detail_item.get("brokerProperties"))
+            )
         receive_result_deserialized["value"] = detail_items
         return receive_result_deserialized
 

sdk/eventgrid/azure-eventgrid/azure/eventgrid/aio/_operations/_patch.py

@@ -129,7 +129,9 @@ async def receive_cloud_events(
         for detail_item in receive_result.value:
             deserialized_cloud_event = CloudEvent.from_dict(detail_item.event)
             detail_item.event = deserialized_cloud_event
-            detail_items.append(ReceiveDetails(event=detail_item.get("event"), broker_properties=detail_item.get("brokerProperties")))
+            detail_items.append(
+                ReceiveDetails(event=detail_item.get("event"), broker_properties=detail_item.get("brokerProperties"))
+            )
         receive_result_deserialized["value"] = detail_items
         return receive_result_deserialized
 

sdk/eventgrid/azure-eventgrid/azure/eventgrid/models/__init__.py

@@ -9,7 +9,6 @@
 from ._models import AcknowledgeOptions
 from ._models import AcknowledgeResult
 from ._models import FailedLockToken
-from ._models import LockToken
 from ._models import RejectOptions
 from ._models import RejectResult
 from ._models import ReleaseOptions
@@ -22,7 +21,6 @@
     "AcknowledgeOptions",
     "AcknowledgeResult",
     "FailedLockToken",
-    "LockToken",
     "RejectOptions",
     "RejectResult",
     "ReleaseOptions",

sdk/eventgrid/azure-eventgrid/azure/eventgrid/models/_models.py

@@ -95,12 +95,12 @@ class BrokerProperties(_model_base.Model):
     All required parameters must be populated in order to send to Azure.
 
     :ivar lock_token: The token used to lock the event. Required.
-    :vartype lock_token: ~azure.eventgrid.models.LockToken
+    :vartype lock_token: str
     :ivar delivery_attempt_count: The attempt count for deliverying the event. Required.
     :vartype delivery_attempt_count: int
     """
 
-    lock_token: "_models.LockToken" = rest_field(name="lockToken")
+    lock_token: str = rest_field(name="lockToken")
     """The token used to lock the event. Required."""
     delivery_attempt_count: int = rest_field(name="deliveryAttemptCount")
     """The attempt count for deliverying the event. Required."""
@@ -168,7 +168,7 @@ class FailedLockToken(_model_base.Model):
     All required parameters must be populated in order to send to Azure.
 
     :ivar lock_token: LockToken value. Required.
-    :vartype lock_token: ~azure.eventgrid.models.LockToken
+    :vartype lock_token: str
     :ivar error_code: Error code related to the token. Example of such error codes are BadToken:
      which indicates the Token is not formatted correctly, TokenLost: which indicates that token is
      not found, and InternalServerError: For any internal server errors. Required.
@@ -177,7 +177,7 @@ class FailedLockToken(_model_base.Model):
     :vartype error_description: str
     """
 
-    lock_token: "_models.LockToken" = rest_field(name="lockToken")
+    lock_token: str = rest_field(name="lockToken")
     """LockToken value. Required."""
     error_code: str = rest_field(name="errorCode")
     """Error code related to the token. Example of such error codes are BadToken: which indicates the
@@ -190,7 +190,7 @@ class FailedLockToken(_model_base.Model):
     def __init__(
         self,
         *,
-        lock_token: "_models.LockToken",
+        lock_token: str,
         error_code: str,
         error_description: str,
     ):
@@ -207,37 +207,6 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:  # pylint: disable=useles
         super().__init__(*args, **kwargs)
 
 
-class LockToken(_model_base.Model):
-    """LockToken information.
-
-    All required parameters must be populated in order to send to Azure.
-
-    :ivar lock_token: The token used to lock the event. Required.
-    :vartype lock_token: str
-    """
-
-    lock_token: str = rest_field(name="lockToken")
-    """The token used to lock the event. Required."""
-
-    @overload
-    def __init__(
-        self,
-        *,
-        lock_token: str,
-    ):
-        ...
-
-    @overload
-    def __init__(self, mapping: Mapping[str, Any]):
-        """
-        :param mapping: raw JSON to initialize the model.
-        :type mapping: Mapping[str, Any]
-        """
-
-    def __init__(self, *args: Any, **kwargs: Any) -> None:  # pylint: disable=useless-super-delegation
-        super().__init__(*args, **kwargs)
-
-
 class ReceiveDetails(_model_base.Model):
     """Receive operation details per Cloud Event.
 

sdk/eventgrid/azure-eventgrid/azure/eventgrid/models/_patch.py

@@ -8,9 +8,14 @@
 """
 from typing import List, overload, Mapping, Any
 from ._models import LockToken
-from ._models import ReceiveDetails as InternalReceiveDetails, ReceiveResult as InternalReceiveResult, BrokerProperties as InternalBrokerProperties
+from ._models import (
+    ReceiveDetails as InternalReceiveDetails,
+    ReceiveResult as InternalReceiveResult,
+    BrokerProperties as InternalBrokerProperties,
+)
 from azure.core.messaging import CloudEvent
 
+
 class ReceiveDetails(InternalReceiveDetails):
     """Receive operation details per Cloud Event.
 
@@ -41,6 +46,7 @@ def __init__(self, mapping: Mapping[str, Any]):
     def __init__(self, *args: Any, **kwargs: Any) -> None:  # pylint: disable=useless-super-delegation
         super().__init__(*args, **kwargs)
 
+
 class ReceiveResult(InternalReceiveResult):
     """Details of the Receive operation response.
 
@@ -68,6 +74,7 @@ def __init__(self, mapping: Mapping[str, Any]):
     def __init__(self, *args: Any, **kwargs: Any) -> None:  # pylint: disable=useless-super-delegation
         super().__init__(*args, **kwargs)
 
+
 class BrokerProperties(InternalBrokerProperties):
     """Properties of the Event Broker operation.
 
@@ -99,7 +106,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:  # pylint: disable=useles
         super().__init__(*args, **kwargs)
 
 
-__all__: List[str] = ["ReceiveDetails", "ReceiveResult", "BrokerProperties"]  # Add all objects you want publicly available to users at this package level
+__all__: List[str] = [
+    "ReceiveDetails",
+    "ReceiveResult",
+    "BrokerProperties",
+]  # Add all objects you want publicly available to users at this package level
 
 
 def patch_sdk():