diff --git a/server/src/functions/return_all_data/lambda_function.py b/server/src/functions/return_all_data/lambda_function.py index e61f1d1..0f5239b 100644 --- a/server/src/functions/return_all_data/lambda_function.py +++ b/server/src/functions/return_all_data/lambda_function.py @@ -11,6 +11,10 @@ def lambda_handler(event, context): items = response.get('Items', []) + if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']: + objectType = event['queryStringParameters']['objectType'] + items = [item for item in items if item['objectType'] == objectType] + return { 'statusCode': 200, 'body': json.dumps(items)