diff --git a/src/eligibility_signposting_api/repos/campaign_repo.py b/src/eligibility_signposting_api/repos/campaign_repo.py index 6dc06de6e..42fc2efd0 100644 --- a/src/eligibility_signposting_api/repos/campaign_repo.py +++ b/src/eligibility_signposting_api/repos/campaign_repo.py @@ -26,12 +26,12 @@ def __init__( self.s3_client = s3_client self.bucket_name = bucket_name - @xray_recorder.capture("CampaignRepo.get_campaign_configs") # pyright: ignore[reportCallIssue] def get_campaign_configs(self) -> Generator[CampaignConfig]: - with xray_recorder.in_subsegment("CampaignRepo.get_campaign_configs:list_objects"): - campaign_objects = self.s3_client.list_objects(Bucket=self.bucket_name) - with xray_recorder.in_subsegment("CampaignRepo.get_campaign_configs:get_objects"): - for campaign_object in campaign_objects["Contents"]: - response = self.s3_client.get_object(Bucket=self.bucket_name, Key=f"{campaign_object['Key']}") - body = response["Body"].read() - yield Rules.model_validate(json.loads(body)).campaign_config + with xray_recorder.in_subsegment("CampaignRepo.get_campaign_configs"): + with xray_recorder.in_subsegment("list_objects"): + campaign_objects = self.s3_client.list_objects(Bucket=self.bucket_name) + with xray_recorder.in_subsegment("get_objects"): + for campaign_object in campaign_objects["Contents"]: + response = self.s3_client.get_object(Bucket=self.bucket_name, Key=f"{campaign_object['Key']}") + body = response["Body"].read() + yield Rules.model_validate(json.loads(body)).campaign_config