cleaner approach

Author: michael-zhao459

datadog_lambda/tracing.py

@@ -70,11 +70,9 @@
 
 def _dsm_set_context_sqs_or_sns_event(event, event_type):
     for record in event.get("Records", [])[1:]:
-        arn = _dsm_get_source_arn(record, event_type)
-        if not arn:
-            continue
-        context = extract_context_from_sqs_or_sns_record(record)
-        _dsm_set_checkpoint(context, event_type, arn)
+        if arn := _dsm_get_source_arn(record, event_type):
+            context = extract_context_from_sqs_or_sns_record(record)
+            _dsm_set_checkpoint(context, event_type, arn)
 
 
 def _dsm_set_context_kinesis_event(event):