add logging

This commit is contained in:
Daniel Roth 2026-02-09 17:00:59 +00:00
parent 3fa51e3c3a
commit 9b6f99929e

View file

@ -17,20 +17,29 @@ def handler(event: Mapping[str, Any], context: Any) -> None:
for record in event.get("Records", []): for record in event.get("Records", []):
try: try:
body_dict = json.loads(record["body"]) body_dict = json.loads(record["body"])
logger.info("Validating request body")
payload = ConditionTriggerRequest.model_validate(body_dict) payload = ConditionTriggerRequest.model_validate(body_dict)
logger.info("Successfully validated request body")
if payload.uprn_lookup_file_bucket and payload.uprn_lookup_file_key: if payload.uprn_lookup_file_bucket and payload.uprn_lookup_file_key:
logger.info("Getting UPRN lookup file from s3")
uprn_lookup = UprnLookupS3( uprn_lookup = UprnLookupS3(
bucket=payload.uprn_lookup_file_bucket, bucket=payload.uprn_lookup_file_bucket,
key=payload.uprn_lookup_file_key, key=payload.uprn_lookup_file_key,
) # TODO: replace with postgres implementation ) # TODO: replace with postgres implementation
logger.info("Successfully got UPRN lookup file from s3")
else: else:
uprn_lookup = None uprn_lookup = None
logger.info("Getting conditions data from s3")
file_bytes: BytesIO = read_io_from_s3( file_bytes: BytesIO = read_io_from_s3(
bucket_name=payload.trigger_file_bucket, bucket_name=payload.trigger_file_bucket,
file_key=payload.trigger_file_key, file_key=payload.trigger_file_key,
) )
logger.info(
"Successfully got conditions data from s3. Moving on to process file..."
)
process_file( process_file(
file_stream=file_bytes, file_stream=file_bytes,