From 6814d5ef8d81181e85196ae9015a6ab63d50192d Mon Sep 17 00:00:00 2001
From: Simon Li <spli@dundee.ac.uk>
Date: Wed, 20 Mar 2024 11:50:53 +0000
Subject: [PATCH] Don't return eggress requests older than 90 days in
 list_requests

---
 .../lambda/egress_api/list_requests.py           | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/egress_backend/lambda/egress_api/list_requests.py b/egress_backend/lambda/egress_api/list_requests.py
index c4b3250..02bcbd2 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"]
+    response = ddb_table.scan(Limit=10)
+    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