Fixed redirects with new dictionary

lambda_redirect/lambda_redirect.py

@@ -47,9 +47,11 @@ def lambda_handler(event: dict, context):
     try:
         redirects = json.load(resp['Body'])
 
-        params =  RequestParams(**event.get('queryStringParameters', {}))                               
-        customer = redirects.get(params.id, {})
-        tag = customer.get(params.tag_id, {})
+        params =  RequestParams(**event.get('queryStringParameters', {}))  
+        customers = redirects.get('customers', {})                             
+        customer = customers.get(params.id, {})
+        tags = customer.get('tags', {})
+        tag = tags.get(params.tag_id, {})
         content = tag.get('content', None)
         
         # In case of multi face tag select the correct face
@@ -63,7 +65,8 @@ def lambda_handler(event: dict, context):
         content = Content(**content)
         match content.type:
             case "s3":
-                    key = f'{params.id}/{params.tag_id}/{content.key}'
+                    file_name = content.key.replace('+',' ') # remove slugify of filename
+                    key = f'{params.id}/{params.tag_id}{'/'+params.face_id if params.face_id else ''}/{file_name}'
                     final_redirect = s3_client.generate_presigned_url('get_object',
                                                                 Params={'Bucket': bucket_data,
                                                                         'Key': key},
@@ -71,7 +74,6 @@ def lambda_handler(event: dict, context):
             case "url":
                 final_redirect = content.url
 
-
     except json.decoder.JSONDecodeError as je:
         print(je)
         return {
@@ -104,8 +106,8 @@ if __name__ == "__main__":
     "httpMethod": "GET",
     "queryStringParameters": {
         "id": "customer1",
-        "tag_id": "tag3",
-        "face_id": "face1"
+        "tag_id": "tag2",
+        "face_id": "face2"
         },
     }