[server]: Add pagination handling to return_all_data

This commit is contained in:
2025-02-28 23:28:39 +00:00
parent 47e6cb5dc5
commit 4357c4b518

View File

@ -7,9 +7,15 @@ table = dynamodb.Table(os.environ['TABLE_NAME'])
def lambda_handler(event, context):
try:
items = []
response = table.scan()
items = response.get('Items', [])
items.extend(response.get('Items', []))
# continue to scan while there are more pages
while 'LastEvaluatedKey' in response:
response = table.scan(ExclusiveStartKey=response['LastEvaluatedKey'])
items.extend(response.get('Items', []))
if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']:
objectType = event['queryStringParameters']['objectType']