EventGrid 100% TypeScore (#36132)

Author: lmazuel

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

@@ -214,7 +214,7 @@ def __init__(
         self.data = data
         self.event_type = event_type
         self.event_time = event_time
-        self.metadata_version = None
+        self.metadata_version: Optional[str] = None
         self.data_version = data_version
 
 

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_legacy/_messaging_shared.py

@@ -10,11 +10,11 @@
 # Both the files should always be identical.
 # ==========================================================================
 
-
+from typing import Any
 import json
 
 
-def _get_json_content(obj):  # pylint: disable=inconsistent-return-statements
+def _get_json_content(obj: Any) -> Any:  # pylint: disable=inconsistent-return-statements
     """Event mixin to have methods that are common to different Event types
     like CloudEvent, EventGridEvent etc.
 

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

@@ -81,8 +81,7 @@ class EventGridEvent(InternalEventGridEvent):
         "data_version": {"key": "dataVersion", "type": "str"},
     }
 
-    def __init__(self, subject, event_type, data, data_version, **kwargs):
-        # type: (str, str, object, str, Any) -> None
+    def __init__(self, subject: str, event_type: str, data: Any, data_version: str, **kwargs: Any) -> None:
         kwargs.setdefault("id", uuid.uuid4())
         kwargs.setdefault("subject", subject)
         kwargs.setdefault("event_type", event_type)
@@ -92,14 +91,13 @@ def __init__(self, subject, event_type, data, data_version, **kwargs):
 
         super(EventGridEvent, self).__init__(**kwargs)
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return "EventGridEvent(subject={}, event_type={}, id={}, event_time={})".format(
             self.subject, self.event_type, self.id, self.event_time
         )[:1024]
 
     @classmethod
-    def from_json(cls, event):
-        # type: (Any) -> EventGridEvent
+    def from_json(cls, event: Any) -> "EventGridEvent":
         """
         Returns the deserialized EventGridEvent object when a json payload is provided.
         :param event: The json string that should be converted into a EventGridEvent. This can also be

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

@@ -6,15 +6,16 @@
 
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
 """
-from typing import List, overload, Mapping, Any
+from typing import List, overload, Mapping, Any, Generic, TypeVar
 from azure.core.messaging import CloudEvent
 from ._models import (
     ReceiveDetails as InternalReceiveDetails,
     BrokerProperties as InternalBrokerProperties,
 )
 
+DataType = TypeVar("DataType")
 
-class ReceiveDetails(InternalReceiveDetails):
+class ReceiveDetails(InternalReceiveDetails, Generic[DataType]):
     """Receive operation details per Cloud Event.
 
     All required parameters must be populated in order to send to Azure.
@@ -30,7 +31,7 @@ def __init__(
         self,
         *,
         broker_properties: "BrokerProperties",
-        event: "CloudEvent",
+        event: CloudEvent[DataType],
     ): ...
 
     @overload