make helpers private

Author: michael-zhao459

datadog_lambda/tracing.py

@@ -71,14 +71,14 @@
 def _dsm_set_context_sqs_or_sns_event(event, event_type):
     for record in event.get("Records", [])[1:]:
         if arn := get_source_arn_sqs_or_sns(record, event_type):
-            context = extract_context_from_sqs_or_sns_record(record)
+            context = _extract_context_from_sqs_or_sns_record(record)
             _dsm_set_checkpoint(context, event_type, arn)
 
 
 def _dsm_set_context_kinesis_event(event):
     for record in event.get("Records", [])[1:]:
         if (arn := record.get("eventSourceARN")) and (kinesis := record.get("kinesis")):
-            context = extract_context_from_kinesis_record(kinesis)
+            context = _extract_context_from_kinesis_record(kinesis)
             _dsm_set_checkpoint(context, "kinesis", arn)
 
 
@@ -272,7 +272,7 @@ def extract_context_from_sqs_or_sns_event_or_context(
     try:
         first_record = event.get("Records")[0]
         source_arn = get_source_arn_sqs_or_sns(first_record, event_type)
-        dd_data = extract_context_from_sqs_or_sns_record(first_record)
+        dd_data = _extract_context_from_sqs_or_sns_record(first_record)
         if dd_data:
             if is_step_function_event(dd_data):
                 try:
@@ -322,7 +322,7 @@ def extract_context_from_sqs_or_sns_event_or_context(
         return extract_context_from_lambda_context(lambda_context)
 
 
-def extract_context_from_sqs_or_sns_record(record):
+def _extract_context_from_sqs_or_sns_record(record):
     # logic to deal with SNS => SQS event
     if "body" in record:
         body_str = record.get("body")
@@ -428,7 +428,7 @@ def extract_context_from_kinesis_event(event, lambda_context):
         kinesis = record.get("kinesis")
         if not kinesis:
             return extract_context_from_lambda_context(lambda_context)
-        dd_ctx = extract_context_from_kinesis_record(kinesis)
+        dd_ctx = _extract_context_from_kinesis_record(kinesis)
         if dd_ctx:
             context = propagator.extract(dd_ctx)
             _dsm_set_checkpoint(dd_ctx, "kinesis", source_arn)
@@ -444,7 +444,7 @@ def extract_context_from_kinesis_event(event, lambda_context):
     return extract_context_from_lambda_context(lambda_context)
 
 
-def extract_context_from_kinesis_record(record_kinesis_data):
+def _extract_context_from_kinesis_record(record_kinesis_data):
     data = record_kinesis_data.get("data")
     if data:
         import base64