🔧 Update handler.py logic (#21)

* Update handler.py logic * linting
ministryofjustice · Apr 9, 2024 · 9490387 · 9490387
1 parent 95b3397
commit 9490387
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/var/task/handler.py b/src/var/task/handler.py
@@ -99,10 +99,11 @@ def handler(event, context):  # pylint: disable=unused-argument
         elif mode == "transferred":
             # This mode expects CSV style notifications from
             # the transfer Lambda
-            # e.g, "transferred,${supplier}/${file_name},${timestamp}"
-            state, object_key, timestamp = event[  # pylint: disable=unused-variable
-                "message"
-            ].split(",")
+            # e.g, "transferred,{supplier}/{file_name},{timestamp}"
+            message = event["Records"][0]["Sns"]["Message"]
+            state, object_key, timestamp = message.split(  # pylint: disable=unused-variable
+                ","
+            )
             supplier, file_name = object_key.split("/")[:2]
             supplier_config = supplier_configuration(supplier=supplier)