[MegaLinter] Apply linters fixes

Author: lrafeei

newrelic/api/opentelemetry.py

@@ -16,24 +16,19 @@
 from contextlib import contextmanager
 
 from opentelemetry import trace as otel_api_trace
+
 from newrelic.api.application import application_instance, register_application
-from newrelic.api.web_transaction import WebTransaction
 from newrelic.api.background_task import BackgroundTask
-from newrelic.api.message_transaction import MessageTransaction
-from newrelic.api.function_trace import FunctionTrace
 from newrelic.api.datastore_trace import DatastoreTrace
-from newrelic.api.message_trace import MessageTrace
 from newrelic.api.external_trace import ExternalTrace
+from newrelic.api.function_trace import FunctionTrace
+from newrelic.api.message_trace import MessageTrace
+from newrelic.api.message_transaction import MessageTransaction
 from newrelic.api.time_trace import current_trace, notice_error
-from newrelic.api.transaction import (
-    current_transaction,
-    record_custom_metric,
-    record_dimensional_metric,
-    Sentinel,
-)
+from newrelic.api.transaction import Sentinel, current_transaction
+from newrelic.api.web_transaction import WebTransaction
 from newrelic.core.otlp_utils import create_resource
 
-
 # Attributes that help distinguish span types are
 # sometimes added after span creation and sometimes
 # they are added during span creation.  During the case
@@ -95,11 +90,13 @@ def __init__(
         self.otel_parent = parent
         self.attributes = attributes or {}
         self.kind = kind
-        self.nr_transaction = nr_transaction or current_transaction()   # This attribute is purely to prevent garbage collection
+        self.nr_transaction = (
+            nr_transaction or current_transaction()
+        )  # This attribute is purely to prevent garbage collection
         self.nr_trace = None
         self.instrumenting_module = instrumenting_module
 
-        # We should never reach this point.  Leave this in 
+        # We should never reach this point.  Leave this in
         # for debug purposes but eventually take this out.
         if not self.nr_transaction:
             raise ValueError("HOW DID WE GET HERE??")
@@ -108,10 +105,7 @@ def __init__(
         current_nr_trace = current_trace()
         if (
             not self.otel_parent
-            or (
-                self.otel_parent
-                and self.otel_parent.span_id == int(current_nr_trace.guid, 16)
-            )
+            or (self.otel_parent and self.otel_parent.span_id == int(current_nr_trace.guid, 16))
             or (self.otel_parent and isinstance(current_nr_trace, Sentinel))
         ):
             # Expected to come here if one of three scenarios have occured:
@@ -129,11 +123,7 @@ def __init__(
             raise ValueError("Unexpected span parent scenario encountered")
 
         if nr_trace_type == FunctionTrace:
-            trace_kwargs = {
-                "name": self.name,
-                "params": self.attributes,
-                "parent": self.nr_parent,
-            }
+            trace_kwargs = {"name": self.name, "params": self.attributes, "parent": self.nr_parent}
             self.nr_trace = nr_trace_type(**trace_kwargs)
         elif nr_trace_type == DatastoreTrace:
             trace_kwargs = {
@@ -165,41 +155,30 @@ def __init__(
         else:
             # TODO: Still need to implement GraphQLOperationTrace
             # and GraphQLResolverTrace
-            trace_kwargs = {
-                "name": self.name,
-                "params": self.attributes,
-                "parent": self.nr_parent,
-            }
+            trace_kwargs = {"name": self.name, "params": self.attributes, "parent": self.nr_parent}
             self.nr_trace = nr_trace_type(**trace_kwargs)
 
         self.nr_trace.__enter__()
 
     def _is_sampled(self):
         # Uses NR to determine if the trace is sampled
-        
+
         # transaction.sampled can be None, True, False.
         # If None, this has not been computed by NR which
         # can also mean the following:
         # 1. There was not a context passed in that explicitly has sampling disabled.
         #   This flag would be found in the traceparent or traceparent and tracespan headers.
         # 2. Transaction was not created where DT headers are accepted during __init__
         # Therefore, we will treat a value of `None` as `True` for now.
-        
+
         if self.otel_parent:
             return bool(self.otel_parent.trace_flags)
         else:
-            return bool(
-                self.nr_transaction
-                and (
-                    self.nr_transaction.sampled
-                    or
-                    (self.nr_transaction._sampled is None)
-                    )
-                )
+            return bool(self.nr_transaction and (self.nr_transaction.sampled or (self.nr_transaction._sampled is None)))
 
     def _is_remote(self):
         # Remote span denotes if propagated from a remote parent
-        if (self.otel_parent and self.otel_parent.is_remote):
+        if self.otel_parent and self.otel_parent.is_remote:
             return True
         return False
 
@@ -248,28 +227,21 @@ def update_name(self, name):
 
     def is_recording(self):
         return self._is_sampled() and not (
-            hasattr(self, "nr_trace")
-            and hasattr(self.nr_trace, "end_time")
-            and self.nr_trace.end_time
+            hasattr(self, "nr_trace") and hasattr(self.nr_trace, "end_time") and self.nr_trace.end_time
         )
 
     def set_status(self, status, description=None):
         # TODO: not implemented yet
         pass
 
-    def record_exception(
-        self, exception, attributes=None, timestamp=None, escaped=False
-    ):
+    def record_exception(self, exception, attributes=None, timestamp=None, escaped=False):
         if not hasattr(self, "nr_trace"):
             if exception:
                 notice_error((type(exception), exception, exception.__traceback__))
             else:
                 notice_error(sys.exc_info(), attributes=attributes)
         else:
-            self.nr_trace.notice_error(
-                (type(exception), exception, exception.__traceback__),
-                attributes=attributes,
-            )
+            self.nr_trace.notice_error((type(exception), exception, exception.__traceback__), attributes=attributes)
 
     def end(self, end_time=None, *args, **kwargs):
         # We will ignore the end_time parameter and use NR's end_time
@@ -291,7 +263,7 @@ def end(self, end_time=None, *args, **kwargs):
         # Add attributes as Trace parameters
         self._set_attributes_in_nr(self.attributes)
 
-        # For each kind of NR Trace, we will need to add 
+        # For each kind of NR Trace, we will need to add
         # specific attributes since they were likely not
         # available at the time of the trace's creation.
         if self.instrumenting_module in ("Redis", "Mongodb"):
@@ -315,18 +287,10 @@ class Tracer(otel_api_trace.Tracer):
     def __init__(self, resource=None, instrumentation_library=None, *args, **kwargs):
         self.resource = resource
         self.instrumentation_library = instrumentation_library.split(".")[-1].capitalize()
-        self.nr_application = application_instance(
-            activate=False
-        ) or register_application("OtelTracer")
-        self.global_settings = (
-            self.nr_application and self.nr_application.global_settings
-        )
+        self.nr_application = application_instance(activate=False) or register_application("OtelTracer")
+        self.global_settings = self.nr_application and self.nr_application.global_settings
 
-        if (
-            self.nr_application
-            and self.global_settings.enabled
-            and self.nr_application.enabled
-        ):
+        if self.nr_application and self.global_settings.enabled and self.nr_application.enabled:
             self._settings = self.nr_application.settings
             if not self._settings:
                 self.nr_application.activate()
@@ -378,17 +342,13 @@ def start_span(
                     request_path=request_path,
                     headers=headers,
                 )
-            elif kind in (
-                otel_api_trace.SpanKind.PRODUCER,
-                otel_api_trace.SpanKind.INTERNAL,
-            ):
+            elif kind in (otel_api_trace.SpanKind.PRODUCER, otel_api_trace.SpanKind.INTERNAL):
                 transaction = BackgroundTask(self.nr_application, name=name)
             elif kind == otel_api_trace.SpanKind.CONSUMER:
                 # NOTE: NR uses MessageTransaction for Pika, RabbitMQ, Kafka
                 if (
                     self.instrumentation_library in INSTRUMENTING_MODULE_TYPE
-                    and INSTRUMENTING_MODULE_TYPE[self.instrumentation_library]
-                    == "message"
+                    and INSTRUMENTING_MODULE_TYPE[self.instrumentation_library] == "message"
                 ):
                     transaction = MessageTransaction(
                         library=self.instrumentation_library,
@@ -399,12 +359,8 @@ def start_span(
                         headers=headers,
                     )
                 else:
-                    transaction = BackgroundTask(
-                        self.nr_application,
-                        name=name,
-                        group="Celery",
-                    )
-            
+                    transaction = BackgroundTask(self.nr_application, name=name, group="Celery")
+
             transaction.__enter__()
 
         # If not parent_span_context or not parent_span_context.is_remote
@@ -434,7 +390,7 @@ def start_span(
                         request_method=request_method,
                         request_path=request_path,
                         headers=headers,
-                            )
+                    )
                 transaction.__enter__()
             elif kind == otel_api_trace.SpanKind.INTERNAL:
                 if transaction:
@@ -445,12 +401,8 @@ def start_span(
                 if transaction:
                     if (
                         (self.instrumentation_library in INSTRUMENTING_MODULE_TYPE)
-                        and (
-                            INSTRUMENTING_MODULE_TYPE[self.instrumentation_library]
-                            == "db"
-                        )
-                        or (attributes and ("db.system" in attributes))
-                    ):
+                        and (INSTRUMENTING_MODULE_TYPE[self.instrumentation_library] == "db")
+                    ) or (attributes and ("db.system" in attributes)):
                         nr_trace_type = DatastoreTrace
                     else:
                         nr_trace_type = ExternalTrace
@@ -463,8 +415,7 @@ def start_span(
                     # NOTE: NR uses MessageTransaction for Pika, RabbitMQ, Kafka
                     if (
                         self.instrumentation_library in INSTRUMENTING_MODULE_TYPE
-                        and INSTRUMENTING_MODULE_TYPE[self.instrumentation_library]
-                        == "message"
+                        and INSTRUMENTING_MODULE_TYPE[self.instrumentation_library] == "message"
                     ):
                         transaction = MessageTransaction(
                             library=self.instrumentation_library,
@@ -475,11 +426,7 @@ def start_span(
                             headers=headers,
                         )
                     else:
-                        transaction = BackgroundTask(
-                            self.nr_application,
-                            name=name,
-                            group="Celery",
-                        )
+                        transaction = BackgroundTask(self.nr_application, name=name, group="Celery")
                     transaction.__enter__()
             elif kind == otel_api_trace.SpanKind.PRODUCER:
                 if transaction:
@@ -503,7 +450,6 @@ def start_span(
 
         return span
 
-
     @contextmanager
     def start_as_current_span(
         self,
@@ -525,9 +471,7 @@ def start_as_current_span(
             set_status_on_exception=set_status_on_exception,
         )
 
-        with otel_api_trace.use_span(
-            span, end_on_exit=end_on_exit, record_exception=record_exception
-        ) as current_span:
+        with otel_api_trace.use_span(span, end_on_exit=end_on_exit, record_exception=record_exception) as current_span:
             yield current_span
 
 
@@ -544,8 +488,4 @@ def get_tracer(
         *args,
         **kwargs,
     ):
-        return Tracer(
-            resource=self._resource, instrumentation_library=instrumenting_module_name
-                    )
-                        
-                    
+        return Tracer(resource=self._resource, instrumentation_library=instrumenting_module_name)

newrelic/config.py

@@ -4362,15 +4362,11 @@ def _process_module_builtin_defaults():
 
     # Hybrid Agent Hooks
     _process_module_definition(
-        "opentelemetry.trace",
-        "newrelic.hooks.hybridagent_opentelemetry",
-        "instrument_trace_api",
+        "opentelemetry.trace", "newrelic.hooks.hybridagent_opentelemetry", "instrument_trace_api"
     )
-     
+
     _process_module_definition(
-        "opentelemetry.instrumentation.utils",
-        "newrelic.hooks.hybridagent_opentelemetry",
-        "instrument_utils",
+        "opentelemetry.instrumentation.utils", "newrelic.hooks.hybridagent_opentelemetry", "instrument_utils"
     )
 
 

newrelic/core/config.py

@@ -519,6 +519,7 @@ class EventHarvestConfigHarvestLimitSettings(Settings):
 class OtelBridgeSettings(Settings):
     pass
 
+
 _settings = TopLevelSettings()
 _settings.agent_limits = AgentLimitsSettings()
 _settings.application_logging = ApplicationLoggingSettings()
@@ -1227,9 +1228,7 @@ def default_otlp_host(host):
 _settings.azure_operator.enabled = _environ_as_bool("NEW_RELIC_AZURE_OPERATOR_ENABLED", default=False)
 _settings.package_reporting.enabled = _environ_as_bool("NEW_RELIC_PACKAGE_REPORTING_ENABLED", default=True)
 _settings.ml_insights_events.enabled = _environ_as_bool("NEW_RELIC_ML_INSIGHTS_EVENTS_ENABLED", default=False)
-_settings.otel_bridge.enabled = _environ_as_bool(
-    "NEW_RELIC_OTEL_BRIDGE_ENABLED", default=False
-)
+_settings.otel_bridge.enabled = _environ_as_bool("NEW_RELIC_OTEL_BRIDGE_ENABLED", default=False)
 
 
 def global_settings():