Merge pull request #205 from jacebrowning/add-dict-hooks

Add hooks to dictionary values

datafiles/hooks.py

@@ -64,6 +64,12 @@ def apply(instance, mapper):
                 item.datafile = create_mapper(item, root=mapper)
                 apply(item, mapper)
 
+    elif isinstance(instance, dict):
+        for value in instance.values():
+            with suppress(AttributeError):
+                value.datafile = create_mapper(value, root=mapper)
+                apply(value, mapper)
+
 
 def load_before(cls, method):
     """Decorate methods that should load before call."""

pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 
 name = "datafiles"
-version = "0.13b3"
+version = "0.13b4"
 description = "File-based ORM for dataclasses."
 
 license = "MIT"

pytest.ini

@@ -4,6 +4,7 @@ addopts =
     --pdbcls=IPython.terminal.debugger:Pdb
 
     -r sxX
+    --show-capture=log
 
     --cov=datafiles
     --cov-report=html

tests/test_file_inference.py

@@ -73,3 +73,40 @@ def test_float_inference(expect):
           - 4.0
         """
     )
+
+
+def test_nested_mutables(expect):
+    write(
+        'tmp/sample.yml',
+        """
+        name: Test
+        roles:
+          category1:
+            - value1
+            - value2
+          category2:
+            - something
+            - else
+        """,
+    )
+
+    logbreak("Inferring object")
+    sample = auto('tmp/sample.yml')
+
+    logbreak("Updating attributes")
+    sample.roles['category1'].append('value3')
+
+    logbreak("Reading file")
+    expect(read('tmp/sample.yml')) == dedent(
+        """
+        name: Test
+        roles:
+          category1:
+            - value1
+            - value2
+            - value3
+          category2:
+            - something
+            - else
+        """
+    )