feat: enabled patch_dict to add new attributes and elements

biosimulators · Oct 23, 2021 · fd4e3f4 · fd4e3f4
1 parent ff948d1
commit fd4e3f4
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 2 deletions.
diff --git a/biosimulators_utils/_version.py b/biosimulators_utils/_version.py
@@ -1 +1 @@
-__version__ = '0.1.139'
+__version__ = '0.1.140'
diff --git a/biosimulators_utils/utils/core.py b/biosimulators_utils/utils/core.py
@@ -271,7 +271,15 @@ def patch_dict(dictionary, patch):
     while patch_queue:
         props, props_patch = patch_queue.pop()
         for key, new_val in props_patch.items():
-            value = props[key]
+            if isinstance(props, dict):
+                value = props.get(key, None)
+            else:
+                if key >= len(props):
+                    props.extend([None] * (key + 1 - len(props)))
+                    value = None
+                else:
+                    value = props[key]
+
             if isinstance(value, dict):
                 patch_queue.append((props[key], new_val))
 

diff --git a/tests/utils/test_core_utils.py b/tests/utils/test_core_utils.py
@@ -289,12 +289,18 @@ def test_patch_dict(self):
             'version': 'x2',
             'image': {
                 'url': 'y2',
+                'digest': 'z2',
             },
             'algorithms': {
                 1: {
                     'kisaoId': {
                         'id': 'KISAO_XXXXXXX'
                     }
+                },
+                4: {
+                    'kisaoId': {
+                        'id': 'KISAO_YYYYYYY'
+                    }
                 }
             },
             'authors': ['A', 'B', 'C', 'D'],
@@ -303,11 +309,19 @@ def test_patch_dict(self):
         expected_dictionary = copy.deepcopy(dictionary)
         expected_dictionary['version'] = 'x2'
         expected_dictionary['image']['url'] = 'y2'
+        expected_dictionary['image']['digest'] = 'z2'
         expected_dictionary['algorithms'][1]['kisaoId']['id'] = 'KISAO_XXXXXXX'
+        expected_dictionary['algorithms'].append(None)
+        expected_dictionary['algorithms'].append({
+            'kisaoId': {
+                'id': 'KISAO_YYYYYYY'
+            }
+        })
         expected_dictionary['authors'] = ['A', 'B', 'C', 'D']
 
         utils.patch_dict(dictionary, patch)
 
+        print(dictionary['algorithms'])
         self.assertEqual(dictionary, expected_dictionary)
 
     def test_flatten_nested_list_of_strings(self):