[MegaLinter] Apply linters fixes

Author: hmstepanek

newrelic/api/transaction.py

@@ -1020,18 +1020,36 @@ def sampling_algo_compute_sampled_and_priority(self, priority, sampled):
                 priority += 1
         return priority, sampled
 
-    def _compute_sampled_and_priority(self, priority, sampled, remote_parent_sampled_path, remote_parent_sampled_setting, remote_parent_not_sampled_path, remote_parent_not_sampled_setting):
+    def _compute_sampled_and_priority(
+        self,
+        priority,
+        sampled,
+        remote_parent_sampled_path,
+        remote_parent_sampled_setting,
+        remote_parent_not_sampled_path,
+        remote_parent_not_sampled_setting,
+    ):
         if self._remote_parent_sampled is None:
             config = "default"  # Use sampling algo.
             _logger.debug("Sampling decision made based on no remote parent sampling decision present.")
         elif self._remote_parent_sampled:
             setting_path = remote_parent_sampled_path
             config = remote_parent_sampled_setting
-            _logger.debug("Sampling decision made based on remote_parent_sampled=%s and %s=%s.", self._remote_parent_sampled, setting_path, config)
+            _logger.debug(
+                "Sampling decision made based on remote_parent_sampled=%s and %s=%s.",
+                self._remote_parent_sampled,
+                setting_path,
+                config,
+            )
         else:  # self._remote_parent_sampled is False.
             setting_path = remote_parent_not_sampled_path
             config = remote_parent_not_sampled_setting
-            _logger.debug("Sampling decision made based on remote_parent_sampled=%s and %s=%s.", self._remote_parent_sampled, setting_path, config)
+            _logger.debug(
+                "Sampling decision made based on remote_parent_sampled=%s and %s=%s.",
+                self._remote_parent_sampled,
+                setting_path,
+                config,
+            )
         if config == "always_on":
             sampled = True
             priority = 2.0
@@ -1042,7 +1060,9 @@ def _compute_sampled_and_priority(self, priority, sampled, remote_parent_sampled
             if config not in ("default", "adaptive"):
                 _logger.warning("%s=%s is not a recognized value. Using 'default' instead.", setting_path, config)
 
-            _logger.debug("Let adaptive sampler algorithm decide based on sampled=%s and priority=%s.", sampled, priority)
+            _logger.debug(
+                "Let adaptive sampler algorithm decide based on sampled=%s and priority=%s.", sampled, priority
+            )
             priority, sampled = self.sampling_algo_compute_sampled_and_priority(priority, sampled)
         return priority, sampled
 
@@ -1055,14 +1075,18 @@ def _make_sampling_decision(self):
         sampled = self._sampled
         # Compute sampling decision for full granularity.
         if self.settings.distributed_tracing.sampler.full_granularity.enabled:
-            _logger.debug("Full granularity tracing is enabled. Asking if full granularity wants to sample. priority=%s, sampled=%s", priority, sampled)
+            _logger.debug(
+                "Full granularity tracing is enabled. Asking if full granularity wants to sample. priority=%s, sampled=%s",
+                priority,
+                sampled,
+            )
             computed_priority, computed_sampled = self._compute_sampled_and_priority(
                 priority,
                 sampled,
-                remote_parent_sampled_path = "distributed_tracing.sampler.full_granularity.remote_parent_sampled",
-                remote_parent_sampled_setting = self.settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled,
-                remote_parent_not_sampled_path = "distributed_tracing.sampler.full_granularity.remote_parent_not_sampled",
-                remote_parent_not_sampled_setting = self.settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled,
+                remote_parent_sampled_path="distributed_tracing.sampler.full_granularity.remote_parent_sampled",
+                remote_parent_sampled_setting=self.settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled,
+                remote_parent_not_sampled_path="distributed_tracing.sampler.full_granularity.remote_parent_not_sampled",
+                remote_parent_not_sampled_setting=self.settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled,
             )
             _logger.debug("Full granularity sampling decision was %s with priority=%s.", sampled, priority)
             if computed_sampled:
@@ -1077,12 +1101,14 @@ def _make_sampling_decision(self):
             self._priority, self._sampled = self._compute_sampled_and_priority(
                 priority,
                 sampled,
-                remote_parent_sampled_path = "distributed_tracing.sampler.partial_granularity.remote_parent_sampled",
-                remote_parent_sampled_setting = self.settings.distributed_tracing.sampler.partial_granularity.remote_parent_sampled,
-                remote_parent_not_sampled_path = "distributed_tracing.sampler.partial_granularity.remote_parent_not_sampled",
-                remote_parent_not_sampled_setting = self.settings.distributed_tracing.sampler.partial_granularity.remote_parent_not_sampled,
+                remote_parent_sampled_path="distributed_tracing.sampler.partial_granularity.remote_parent_sampled",
+                remote_parent_sampled_setting=self.settings.distributed_tracing.sampler.partial_granularity.remote_parent_sampled,
+                remote_parent_not_sampled_path="distributed_tracing.sampler.partial_granularity.remote_parent_not_sampled",
+                remote_parent_not_sampled_setting=self.settings.distributed_tracing.sampler.partial_granularity.remote_parent_not_sampled,
+            )
+            _logger.debug(
+                "Partial granularity sampling decision was %s with priority=%s.", self._sampled, self._priority
             )
-            _logger.debug("Partial granularity sampling decision was %s with priority=%s.", self._sampled, self._priority)
             self._sampling_decision_made = True
             if self._sampled:
                 self.partial_granularity_sampled = True

newrelic/core/agent_protocol.py

@@ -297,7 +297,9 @@ def _connect_payload(app_name, linked_applications, environment, settings):
         connect_settings["browser_monitoring.loader"] = settings["browser_monitoring.loader"]
         connect_settings["browser_monitoring.debug"] = settings["browser_monitoring.debug"]
         connect_settings["ai_monitoring.enabled"] = settings["ai_monitoring.enabled"]
-        connect_settings["distributed_tracing.sampler.adaptive_sampling_target"] = settings["distributed_tracing.sampler.adaptive_sampling_target"]
+        connect_settings["distributed_tracing.sampler.adaptive_sampling_target"] = settings[
+            "distributed_tracing.sampler.adaptive_sampling_target"
+        ]
 
         security_settings = {}
         security_settings["capture_params"] = settings["capture_params"]

newrelic/core/config.py

@@ -855,14 +855,18 @@ def default_otlp_host(host):
 _settings.distributed_tracing.sampler.remote_parent_not_sampled = os.environ.get(
     "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_REMOTE_PARENT_NOT_SAMPLED", "default"
 )
-_settings.distributed_tracing.sampler.full_granularity.enabled = _environ_as_bool("NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_FULL_GRANULARITY_ENABLED", default=True)
+_settings.distributed_tracing.sampler.full_granularity.enabled = _environ_as_bool(
+    "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_FULL_GRANULARITY_ENABLED", default=True
+)
 _settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled = os.environ.get(
     "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_FULL_GRANULARITY_REMOTE_PARENT_SAMPLED", None
 )
 _settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled = os.environ.get(
     "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_FULL_GRANULARITY_REMOTE_PARENT_NOT_SAMPLED", None
 )
-_settings.distributed_tracing.sampler.partial_granularity.enabled = _environ_as_bool("NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_PARTIAL_GRANULARITY_ENABLED", default=False)
+_settings.distributed_tracing.sampler.partial_granularity.enabled = _environ_as_bool(
+    "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_PARTIAL_GRANULARITY_ENABLED", default=False
+)
 _settings.distributed_tracing.sampler.partial_granularity.type = os.environ.get(
     "NEW_RELIC_DISTRIBUTED_TRACING_SAMPLER_PARTIAL_GRANULARITY_TYPE", "essential"
 )
@@ -1411,9 +1415,13 @@ def simplify_distributed_tracing_sampler_granularity_settings(settings):
     # To simplify logic in the code that uses these settings, store the values that
     # should be used at the `distributed_tracing.sampler.full_granularity` path.
     if not settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled:
-        settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled = settings.distributed_tracing.sampler.remote_parent_sampled
+        settings.distributed_tracing.sampler.full_granularity.remote_parent_sampled = (
+            settings.distributed_tracing.sampler.remote_parent_sampled
+        )
     if not settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled:
-        settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled = settings.distributed_tracing.sampler.remote_parent_not_sampled
+        settings.distributed_tracing.sampler.full_granularity.remote_parent_not_sampled = (
+            settings.distributed_tracing.sampler.remote_parent_not_sampled
+        )
 
 
 def _remove_ignored_configs(server_settings):

newrelic/core/database_node.py

@@ -279,7 +279,15 @@ def trace_node(self, stats, root, connections):
             start_time=start_time, end_time=end_time, name=name, params=params, children=children, label=None
         )
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         sql = self.formatted
 
         if sql:
@@ -288,4 +296,11 @@ def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dic
 
         self.agent_attributes["db.statement"] = sql
 
-        return super().span_event(settings, base_attrs=base_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=base_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )

newrelic/core/external_node.py

@@ -169,7 +169,15 @@ def trace_node(self, stats, root, connections):
             start_time=start_time, end_time=end_time, name=name, params=params, children=children, label=None
         )
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         self.agent_attributes["http.url"] = self.http_url
 
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
@@ -180,4 +188,11 @@ def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dic
         if self.method:
             i_attrs["http.method"] = self.method
 
-        return super().span_event(settings, base_attrs=i_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=i_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )

newrelic/core/function_node.py

@@ -114,8 +114,23 @@ def trace_node(self, stats, root, connections):
             start_time=start_time, end_time=end_time, name=name, params=params, children=children, label=self.label
         )
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
         i_attrs["name"] = f"{self.group}/{self.name}"
 
-        return super().span_event(settings, base_attrs=i_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=i_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )

newrelic/core/loop_node.py

@@ -79,8 +79,23 @@ def trace_node(self, stats, root, connections):
             start_time=start_time, end_time=end_time, name=name, params=params, children=children, label=None
         )
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
         i_attrs["name"] = f"EventLoop/Wait/{self.name}"
 
-        return super().span_event(settings, base_attrs=i_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=i_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )

newrelic/core/node_mixin.py

@@ -49,7 +49,15 @@ def get_trace_segment_params(self, settings, params=None):
         _params["exclusive_duration_millis"] = 1000.0 * self.exclusive
         return _params
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
         i_attrs["type"] = "Span"
         i_attrs["name"] = i_attrs.get("name") or self.name
@@ -114,17 +122,39 @@ def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dic
                 ct_exit_spans[span_attrs][0]["nr.durations"] += self.duration
                 return None
 
-    def span_events(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
-        span = self.span_event(settings, base_attrs=base_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+    def span_events(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
+        span = self.span_event(
+            settings,
+            base_attrs=base_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )
         parent_id = parent_guid
         if span:  # span will be None if the span is an inprocess span or repeated exit span.
             yield span
             # Compressed spans are always reparented onto the entry span.
-            if not settings.distributed_tracing.sampler.partial_granularity.type == "compact" or span[0].get("nr.entryPoint"):
+            if not settings.distributed_tracing.sampler.partial_granularity.type == "compact" or span[0].get(
+                "nr.entryPoint"
+            ):
                 parent_id = self.guid
         for child in self.children:
-            for event in child.span_events(  # noqa: UP028
-                settings, base_attrs=base_attrs, parent_guid=parent_id, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans
+            for event in child.span_events(
+                settings,
+                base_attrs=base_attrs,
+                parent_guid=parent_id,
+                attr_class=attr_class,
+                partial_granularity_sampled=partial_granularity_sampled,
+                ct_exit_spans=ct_exit_spans,
             ):
                 if event:  # event will be None if the span is an inprocess span or repeated exit span.
                     yield event
@@ -156,7 +186,15 @@ def db_instance(self):
         self._db_instance = db_instance_attr
         return db_instance_attr
 
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         a_attrs = self.agent_attributes
         a_attrs["db.instance"] = self.db_instance
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
@@ -188,4 +226,11 @@ def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dic
         except Exception:
             pass
 
-        return super().span_event(settings, base_attrs=i_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=i_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )

newrelic/core/root_node.py

@@ -37,7 +37,15 @@
 
 
 class RootNode(_RootNode, GenericNodeMixin):
-    def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dict, partial_granularity_sampled=False, ct_exit_spans=None):
+    def span_event(
+        self,
+        settings,
+        base_attrs=None,
+        parent_guid=None,
+        attr_class=dict,
+        partial_granularity_sampled=False,
+        ct_exit_spans=None,
+    ):
         i_attrs = (base_attrs and base_attrs.copy()) or attr_class()
         i_attrs["transaction.name"] = self.path
         i_attrs["nr.entryPoint"] = True
@@ -46,7 +54,14 @@ def span_event(self, settings, base_attrs=None, parent_guid=None, attr_class=dic
         if self.tracing_vendors:
             i_attrs["tracingVendors"] = self.tracing_vendors
 
-        return super().span_event(settings, base_attrs=i_attrs, parent_guid=parent_guid, attr_class=attr_class, partial_granularity_sampled=partial_granularity_sampled, ct_exit_spans=ct_exit_spans)
+        return super().span_event(
+            settings,
+            base_attrs=i_attrs,
+            parent_guid=parent_guid,
+            attr_class=attr_class,
+            partial_granularity_sampled=partial_granularity_sampled,
+            ct_exit_spans=ct_exit_spans,
+        )
 
     def trace_node(self, stats, root, connections):
         name = self.path