feat(idempotency): support dataclasses & pydantic models payloads

aws_lambda_powertools/utilities/idempotency/persistence/base.py

@@ -1,7 +1,6 @@
 """
 Persistence layers supporting idempotency
 """
-
 import datetime
 import hashlib
 import json
@@ -226,7 +225,13 @@ def _generate_hash(self, data: Any) -> str:
             Hashed representation of the provided data
 
         """
-        data = getattr(data, "raw_event", data)  # could be a data class depending on decorator order
+        if hasattr(data, "__dataclass_fields__"):
+            import dataclasses
+
+            data = dataclasses.asdict(data)
+        else:
+            data = getattr(data, "raw_event", data)
+
         hashed_data = self.hash_function(json.dumps(data, cls=Encoder, sort_keys=True).encode())
         return hashed_data.hexdigest()
 

tests/functional/idempotency/test_idempotency.py

@@ -1057,3 +1057,33 @@ def two(data):
     assert one(data=mock_event) == "one"
     assert two(data=mock_event) == "two"
     assert len(persistence_store.table.method_calls) == 4
+
+
+def test_idempotent_function_dataclasses():
+    try:
+        # Scenario idempotent_function should allow for python dataclasses
+        from dataclasses import asdict, dataclass
+
+        @dataclass
+        class Foo:
+            name: str
+
+        mock_event = Foo(name="Bar")
+        persistence_layer = MockPersistenceLayer(
+            "test-func.record_handler#" + hashlib.md5(serialize(asdict(mock_event)).encode()).hexdigest()
+        )
+        expected_result = {"message": "Foo"}
+
+        @idempotent_function(persistence_store=persistence_layer, data_keyword_argument="record")
+        def record_handler(record):
+            assert isinstance(record, Foo)
+            return expected_result
+
+        # WHEN calling the function
+        result = record_handler(record=mock_event)
+        # THEN we expect the function to execute successfully
+        assert result == expected_result
+
+    except ModuleNotFoundError:
+        # Python 3.6
+        pass