From 627576ebd011a8acadf5e3992bd16338112e0a5e Mon Sep 17 00:00:00 2001 From: Simon Li Date: Wed, 20 Mar 2024 11:53:46 +0000 Subject: [PATCH] Don't return eggress requests older than 90 days in list_requests --- egress_backend/lambda/egress_api/list_requests.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/egress_backend/lambda/egress_api/list_requests.py b/egress_backend/lambda/egress_api/list_requests.py index c4b3250..4db55bd 100644 --- a/egress_backend/lambda/egress_api/list_requests.py +++ b/egress_backend/lambda/egress_api/list_requests.py @@ -4,10 +4,13 @@ # agreement between Customer and Amazon Web Services, Inc. import os +from datetime import datetime import boto3 from aws_lambda_powertools import Logger, Tracer +MAX_REQUEST_AGE_DAYS = 90 + tracer = Tracer(service="ListRequestsAPI") logger = Logger(service="ListRequestsAPI") @@ -18,13 +21,20 @@ def list_requests(): logger.debug("List Requests API invoked") + now = datetime.now() + + def is_recent(item): + updated_dt = datetime.strptime(item["updated_dt"], "%Y-%m-%dT%H:%M:%S.%fZ") + if (now - updated_dt).days < MAX_REQUEST_AGE_DAYS: + return True + ddb_table = ddb.Table(table) response = ddb_table.scan() - data = response["Items"] + data = [item for item in response["Items"] if is_recent(item)] while response.get("LastEvaluatedKey"): response = ddb_table.scan(ExclusiveStartKey=response["LastEvaluatedKey"]) - data.extend(response["Items"]) + data.extend([item for item in response["Items"] if is_recent(item)]) logger.debug("Successful database scan of all egress requests") return data