Fix inconsistency in post_load behavior (#76)

* Fix inconsistency in post_load behavior

* Remove unneeded import

* Update marshmallow_dataclass/__init__.py

Co-Authored-By: Steven Loria <[email protected]>

* Updated tests

Co-authored-by: Steven Loria <[email protected]>

Author: lovasoa

marshmallow_dataclass/__init__.py

@@ -34,10 +34,9 @@ class User:
       })
       Schema: ClassVar[Type[Schema]] = Schema # For the type checker
 """
-import dataclasses
 import inspect
-from functools import lru_cache
 from enum import EnumMeta
+from functools import lru_cache
 from typing import (
     overload,
     Dict,
@@ -54,6 +53,7 @@ class User:
     Set,
 )
 
+import dataclasses
 import marshmallow
 import typing_inspect
 
@@ -457,12 +457,17 @@ def _base_schema(
     Base schema factory that creates a schema for `clazz` derived either from `base_schema`
     or `BaseSchema`
     """
+
     # Remove `type: ignore` when mypy handles dynamic base classes
     # https://github.com/python/mypy/issues/2813
     class BaseSchema(base_schema or marshmallow.Schema):  # type: ignore
-        @marshmallow.post_load
-        def make_data_class(self, data, **_):
-            return clazz(**data)
+        def load(self, data: Mapping, *, many: bool = None, **kwargs):
+            all_loaded = super().load(data, many=many, **kwargs)
+            many = self.many if many is None else bool(many)
+            if many:
+                return [clazz(**loaded) for loaded in all_loaded]
+            else:
+                return clazz(**all_loaded)
 
     return BaseSchema
 

tests/test_post_load.py

@@ -0,0 +1,44 @@
+import unittest
+
+import marshmallow
+
+import marshmallow_dataclass
+
+
+# Regression test for https://github.com/lovasoa/marshmallow_dataclass/issues/75
+class TestPostLoad(unittest.TestCase):
+    @marshmallow_dataclass.dataclass
+    class Named:
+        first_name: str
+        last_name: str
+
+        @marshmallow.post_load
+        def a(self, data, **_kwargs):
+            data["first_name"] = data["first_name"].capitalize()
+            return data
+
+        @marshmallow.post_load
+        def z(self, data, **_kwargs):
+            data["last_name"] = data["last_name"].capitalize()
+            return data
+
+    def test_post_load_method_naming_does_not_affect_data(self):
+        actual = self.Named.Schema().load(
+            {"first_name": "matt", "last_name": "groening"}
+        )
+        expected = self.Named(first_name="Matt", last_name="Groening")
+        self.assertEqual(actual, expected)
+
+    def test_load_many(self):
+        actual = self.Named.Schema().load(
+            [
+                {"first_name": "matt", "last_name": "groening"},
+                {"first_name": "bart", "last_name": "simpson"},
+            ],
+            many=True,
+        )
+        expected = [
+            self.Named(first_name="Matt", last_name="Groening"),
+            self.Named(first_name="Bart", last_name="Simpson"),
+        ]
+        self.assertEqual(actual, expected)