feat: made exception have operation inside

Author: jhnnsrs

rath/links/aiohttp.py

@@ -39,7 +39,9 @@ class AIOHttpLink(AsyncTerminatingLink):
 
     endpoint_url: str
     """endpoint_url is the URL to send operations to."""
-    ssl_context: SSLContext = Field(default_factory=lambda: ssl.create_default_context(cafile=certifi.where()))
+    ssl_context: SSLContext = Field(
+        default_factory=lambda: ssl.create_default_context(cafile=certifi.where())
+    )
     """ssl_context is the SSLContext to use for the aiohttp session. By default, this
     is a context that uses the certifi CA bundle."""
 
@@ -105,7 +107,9 @@ async def aexecute(self, operation: Operation) -> AsyncIterator[GraphQLResult]:
         payload: Payload = {"query": operation.document}
 
         if operation.node.operation == OperationType.SUBSCRIPTION:
-            raise NotImplementedError("Aiohttp Transport does not support subscriptions")
+            raise NotImplementedError(
+                "Aiohttp Transport does not support subscriptions"
+            )
 
         if len(operation.context.files.items()) > 0:
             payload["variables"] = operation.variables
@@ -120,7 +124,9 @@ async def aexecute(self, operation: Operation) -> AsyncIterator[GraphQLResult]:
             file_streams = {str(i): files[path] for i, path in enumerate(files)}
             operations_str = json.dumps(payload, cls=self.json_encoder)
 
-            data.add_field("operations", operations_str, content_type="application/json")
+            data.add_field(
+                "operations", operations_str, content_type="application/json"
+            )
             file_map_str = json.dumps(file_map)
             data.add_field("map", file_map_str, content_type="application/json")
 
@@ -141,17 +147,25 @@ async def aexecute(self, operation: Operation) -> AsyncIterator[GraphQLResult]:
             connector=aiohttp.TCPConnector(ssl=self.ssl_context),
             json_serialize=lambda x: json.dumps(x, cls=self.json_encoder),
         ) as session:
-            async with session.post(self.endpoint_url, headers=operation.context.headers, **post_kwargs) as response:
+            async with session.post(
+                self.endpoint_url, headers=operation.context.headers, **post_kwargs
+            ) as response:
                 if response.status == HTTPStatus.OK:
                     await response.json()
 
                 if response.status in self.auth_errors:
-                    raise AuthenticationError(f"Token Expired Error {operation.context.headers}")
+                    raise AuthenticationError(
+                        f"Token Expired Error {operation.context.headers}"
+                    )
 
                 json_response = await response.json()
 
                 if "errors" in json_response:
-                    raise GraphQLException("\n".join([e["message"] for e in json_response["errors"]]))
+                    raise GraphQLException(
+                        "\n".join([e["message"] for e in json_response["errors"]]),
+                        operation=operation,
+                        errors=json_response["errors"],
+                    )
 
                 if "data" not in json_response:
                     raise Exception(f"Response does not contain data {json_response}")

rath/links/auth.py

@@ -86,16 +86,15 @@ async def aexecute(
         token = await self.aload_token(operation)
         operation.context.headers["Authorization"] = f"Bearer {token}"
         operation.context.initial_payload["token"] = token
-        
-        
 
         try:
             async for result in self.next.aexecute(operation):
                 yield result
 
         except AuthenticationError as e:
             retry = retry + 1
-            await self.arefresh_token(operation)
+            token = await self.arefresh_token(operation)
+
             if retry > self.maximum_refresh_attempts:
                 raise AuthenticationError("Maximum refresh attempts reached") from e
 

rath/operation.py

@@ -54,7 +54,17 @@ class GraphQLResult(BaseModel):
 class GraphQLException(Exception):
     """GraphQLException is the base exception for all GraphQL errors."""
 
-    pass
+    def __init__(
+        self,
+        message: str,
+        operation: Optional[Operation] = None,
+        errors: Optional[Dict[str, Any]] = None,
+    ):
+        """Initialize the GraphQLException with a message and optional errors."""
+        super().__init__(message)
+        self.message = message
+        self.errors = errors or {}
+        self.operation = operation
 
 
 class SubscriptionDisconnect(GraphQLException):

rath/scalars.py

@@ -44,7 +44,7 @@ def validate(cls: Type["ID"], v: "IDCoercible") -> "ID":
 class WithId(Protocol):
     "A protocol for objects that have an id attribute."
 
-    id: ID
+    id: "IDCoercible"
 
 
 IDCoercible = str | ID | WithId | int