Merge branch 'dataclass'

Author: gertjanvanzwieten

ags/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3"
+__version__ = "0.3.1"
 
 
 def _get_backend_for(path):

ags/_mapping.py

@@ -357,7 +357,7 @@ def unlower(self, obj, surject):
 @dataclasses.dataclass
 class DataClass:
     cls: type
-    fields: dict[str, typing.Any]
+    fields: dict[str, Mapping]
 
     def lower(self, obj, inject):
         if not dataclasses.is_dataclass(obj) or isinstance(obj, type):
@@ -371,9 +371,12 @@ def lower(self, obj, inject):
     def unlower(self, obj, surject):
         dobj = surject(obj, dict)
         d = {}
-        for name, mapping in self.fields.items():
+        for name, value in dobj.items():
+            mapping = self.fields.get(name)
+            if mapping is None:
+                raise ValueError(f"invalid field: {name!r}")
             with context(f".{name}"):
-                d[name] = mapping.unlower(dobj[name], surject)
+                d[name] = mapping.unlower(value, surject)
         return self.cls(**d)