Merge pull request #1944 from kili-technology/feature/lab-3842-aa-sdk-user-when-i-have-a-fail-in-lock-i-can-see-the-reason

feat: aa sdk user when i have a fail in lock i can see the reason

Author: baptiste-olivier

src/kili/adapters/kili_api_gateway/issue/__init__.py

@@ -41,7 +41,6 @@ def create_issues(
                 payload = {
                     "issues": [
                         {
-                            "issueNumber": 0,
                             "labelID": issue.label_id,
                             "objectMid": issue.object_mid,
                             "type": type_,

src/kili/core/graphql/exceptions.py

@@ -0,0 +1,12 @@
+"""Helper functions to extract context from GraphQL error messages."""
+import ast
+from typing import Dict, Optional
+
+
+def extract_error_context(message: str) -> Optional[Dict[str, str]]:
+    """Parse the string error message to extract the first context information."""
+    try:
+        parsed_errors = ast.literal_eval(message)
+        return parsed_errors[0].get("extensions").get("context")
+    except (SyntaxError, ValueError):
+        return None

src/kili/core/graphql/graphql_client.py

@@ -33,6 +33,7 @@
 from kili.adapters.http_client import HttpClient
 from kili.core.constants import MAX_CALLS_PER_MINUTE
 from kili.core.graphql.clientnames import GraphQLClientName
+from kili.core.graphql.exceptions import extract_error_context
 from kili.utils.logcontext import LogContext
 
 gql_requests_logger.setLevel(logging.WARNING)
@@ -281,8 +282,8 @@ def execute(
                 raise kili.exceptions.GraphQLError(error=err.errors) from err
 
         except exceptions.TransportQueryError as err:  # remove validation error
-            # the server refused the query after some retries, we crash
-            raise kili.exceptions.GraphQLError(error=err.errors) from err
+            context = extract_error_context(str(err.errors))
+            raise kili.exceptions.GraphQLError(error=err.errors, context=context) from err
 
     @retry(
         reraise=True,  # re-raise the last exception

src/kili/entrypoints/mutations/issue/__init__.py

@@ -73,7 +73,6 @@ def append_to_issues(
         variables = {
             "issues": [
                 {
-                    "issueNumber": 0,
                     "labelID": label_id,
                     "objectMid": object_mid,
                     "type": type_,

src/kili/exceptions.py

@@ -6,8 +6,9 @@
 class GraphQLError(Exception):
     """Raised when the GraphQL call returns an error."""
 
-    def __init__(self, error, batch_number=None) -> None:
+    def __init__(self, error, batch_number=None, context=None) -> None:
         self.error = error
+        self.context = context
 
         if isinstance(error, List):
             error = error[0]