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 79bec9f
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 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,19 @@
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")
return (now - updated_dt).days < MAX_REQUEST_AGE_DAYS

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 79bec9f

Please sign in to comment.