From 0045d8d03205a0abdd3c543713bb6a8e5f5602a9 2021-01-09 15:09:08
From: Brett Smith <brettcsmith@brettcsmith.org>
Date: 2021-01-09 15:09:08
Subject: [PATCH] fields: Add FormField.add_kid() method.

---

diff --git a/conservancy_beancount/pdfforms/fields.py b/conservancy_beancount/pdfforms/fields.py
index 23ebd6eff7fc997433e6c944ea4152a29801107f..e8786e461d64abca6e4fb04254a4f7a97b486689 100644
--- a/conservancy_beancount/pdfforms/fields.py
+++ b/conservancy_beancount/pdfforms/fields.py
@@ -131,6 +131,12 @@ class FormField:
         except KeyError:
             return None
 
+    def add_kid(self, kid: 'FormField') -> None:
+        if kid.parent() is not None:
+            raise ValueError("given kid field already has a parent")
+        kid._source['Parent'] = self._source
+        self._source.setdefault('Kids', []).append(kid._source)
+
     def is_terminal(self) -> bool:
         return not self._get_value('Kids', None)
 
diff --git a/tests/test_pdfforms_fields.py b/tests/test_pdfforms_fields.py
index 8f54120d94b07eaa17621eb04227bc375fd6cf37..5ff3475be90e42fc33f62b11a12affaaae62ece7 100644
--- a/tests/test_pdfforms_fields.py
+++ b/tests/test_pdfforms_fields.py
@@ -368,3 +368,11 @@ def test_recursive_as_mapping():
         _, _, expected_name = expected_key.rpartition('.')
         assert field.name() == expected_name
     assert next(actual, None) is None
+
+def test_add_kid():
+    parent = fieldsmod.FormField(field_source('parent'))
+    kid = fieldsmod.FormField(field_source('kid'))
+    parent.add_kid(kid)
+    actual, = parent.kids()
+    assert actual.name() == 'kid'
+    assert actual.parent().name() == 'parent'