Graphene-Django introspection fix

newrelic/api/graphql_trace.py

@@ -134,7 +134,7 @@ def wrap_graphql_operation_trace(module, object_path):
 
 
 class GraphQLResolverTrace(TimeTrace):
-    def __init__(self, field_name=None, **kwargs):
+    def __init__(self, field_name=None, field_parent_type=None, field_return_type=None, field_path=None, **kwargs):
         parent = kwargs.pop("parent", None)
         source = kwargs.pop("source", None)
         if kwargs:
@@ -143,6 +143,9 @@ def __init__(self, field_name=None, **kwargs):
         super(GraphQLResolverTrace, self).__init__(parent=parent, source=source)
 
         self.field_name = field_name
+        self.field_parent_type = field_parent_type
+        self.field_return_type = field_return_type
+        self.field_path = field_path
         self._product = None
 
     def __repr__(self):
@@ -170,6 +173,9 @@ def product(self):
 
     def finalize_data(self, *args, **kwargs):
         self._add_agent_attribute("graphql.field.name", self.field_name)
+        self._add_agent_attribute("graphql.field.parentType", self.field_parent_type)
+        self._add_agent_attribute("graphql.field.returnType", self.field_return_type)
+        self._add_agent_attribute("graphql.field.path", self.field_path)
 
         return super(GraphQLResolverTrace, self).finalize_data(*args, **kwargs)
 

newrelic/config.py

@@ -2210,6 +2210,10 @@ def _process_module_builtin_defaults():
         "instrument_graphene_types_schema",
     )
 
+    _process_module_definition(
+        "graphene_django.views", "newrelic.hooks.component_graphenedjango", "instrument_graphene_django_views"
+    )
+
     _process_module_definition(
         "graphql.graphql",
         "newrelic.hooks.framework_graphql",

newrelic/hooks/component_graphenedjango.py

@@ -0,0 +1,65 @@
+# Copyright 2010 New Relic, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from newrelic.api.error_trace import ErrorTrace
+from newrelic.api.graphql_trace import GraphQLOperationTrace
+from newrelic.api.transaction import current_transaction
+from newrelic.common.object_names import callable_name
+from newrelic.common.object_wrapper import wrap_function_wrapper
+from newrelic.core.graphql_utils import graphql_statement
+from newrelic.hooks.framework_graphene import (
+    framework_details as graphene_framework_details,
+)
+from newrelic.hooks.framework_graphql import (
+    framework_version as graphql_framework_version,
+)
+from newrelic.hooks.framework_graphql import ignore_graphql_duplicate_exception
+
+
+def bind_execute_graphql_request(request, data, query, variables, operation_name, show_graphiql=False):
+    return query
+
+
+def wrap_execute_graphql_request(wrapped, instance, args, kwargs):
+    transaction = current_transaction()
+
+    if not transaction:
+        return wrapped(*args, **kwargs)
+
+    try:
+        query = bind_execute_graphql_request(*args, **kwargs)
+    except TypeError:
+        query = None
+
+    framework = graphene_framework_details()
+    transaction.add_framework_info(name=framework[0], version=framework[1])
+    transaction.add_framework_info(name="GraphQL", version=graphql_framework_version())
+
+    if hasattr(query, "body"):
+        query = query.body
+
+    transaction.set_transaction_name(callable_name(wrapped), "GraphQL", priority=10)
+
+    with GraphQLOperationTrace(source=wrapped) as trace:
+        trace.product = "Graphene"
+        if query:
+            trace.statement = graphql_statement(query)
+
+        with ErrorTrace(ignore=ignore_graphql_duplicate_exception):
+            return wrapped(*args, **kwargs)
+
+
+def instrument_graphene_django_views(module):
+    if hasattr(module, "GraphQLView"):
+        wrap_function_wrapper(module, "GraphQLView.execute_graphql_request", wrap_execute_graphql_request)

newrelic/hooks/framework_ariadne.py

@@ -29,9 +29,17 @@
 
 
 def framework_details():
-    import ariadne
-
-    return ("Ariadne", getattr(ariadne, "__version__", None))
+    try:
+        import ariadne
+        version = ariadne.__version__
+    except Exception:
+        try:
+            import pkg_resources
+            version = pkg_resources.get_distribution("ariadne").version
+        except Exception:
+            version = None
+
+    return ("Ariadne", version)
 
 
 def bind_graphql(schema, data, *args, **kwargs):