linter

Author: zarirhamza

datadog_lambda/tracing.py

@@ -861,17 +861,23 @@ def determine_service_name(
     service_mapping, specific_key, generic_key, extracted_key, fallback=None
 ):
     # Check for mapped service (specific key first, then generic key)
-    mapped_service = service_mapping.get(specific_key) or service_mapping.get(generic_key)
+    mapped_service = service_mapping.get(specific_key) or service_mapping.get(
+        generic_key
+    )
     if mapped_service:
         return mapped_service
-    
+
     # Check if AWS service representation is disabled
-    aws_service_representation = os.environ.get("DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED", "").lower()
+    aws_service_representation = os.environ.get(
+        "DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED", ""
+    ).lower()
     if aws_service_representation in ("false", "0"):
         return fallback
-    
+
     # Use extracted_key if it exists and is not empty, otherwise use fallback
-    return extracted_key.strip() if extracted_key and extracted_key.strip() else fallback
+    return (
+        extracted_key.strip() if extracted_key and extracted_key.strip() else fallback
+    )
 
 
 # Initialization code
@@ -1452,12 +1458,14 @@ def create_function_execution_span(
     if config.service:
         service_name = config.service
     else:
-        aws_service_representation = os.environ.get("DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED", "").lower()
+        aws_service_representation = os.environ.get(
+            "DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED", ""
+        ).lower()
         if aws_service_representation in ("false", "0"):
             service_name = "aws.lambda"
         else:
             service_name = function_name if function_name else "aws.lambda"
-    
+
     span = tracer.trace(
         "aws.lambda",
         service=service_name,

tests/test_tracing.py

@@ -1283,7 +1283,7 @@ def test_determine_service_name(self):
             ),
             "default",
         )
-        
+
         # Test with DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED set to false
         os.environ["DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED"] = "false"
         self.assertEqual(
@@ -1292,7 +1292,7 @@ def test_determine_service_name(self):
             ),
             "fallback",
         )
-        
+
         # Test with DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED set to 0
         os.environ["DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED"] = "0"
         self.assertEqual(
@@ -1301,7 +1301,7 @@ def test_determine_service_name(self):
             ),
             "fallback",
         )
-        
+
         # Test with DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED not set (default behavior)
         if "DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED" in os.environ:
             del os.environ["DD_TRACE_AWS_SERVICE_REPRESENTATION_ENABLED"]
@@ -1311,7 +1311,7 @@ def test_determine_service_name(self):
             ),
             "extracted",
         )
-        
+
         # Test with empty extracted key
         self.assertEqual(
             determine_service_name(