Skip to content

Commit

Permalink
Don't return eggress requests older than 90 days in list_requests
Browse files Browse the repository at this point in the history
  • Loading branch information
manics committed Mar 20, 2024
1 parent b5c43ca commit 627576e
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions egress_backend/lambda/egress_api/list_requests.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand All @@ -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

0 comments on commit 627576e

Please sign in to comment.