Issue #5 add included_keys arg to MultiDictGetter.simple_merge

src/openeo_aggregator/utils.py

@@ -110,14 +110,17 @@ def merge_arrays(self, key: str,skip_duplicates=False) -> list:
                 result.append(item)
         return result
 
-    def simple_merge(self) -> dict:
+    def simple_merge(self, included_keys=None) -> dict:
         """
         All dictionaries are merged following simple rules:
         For list or sets: all elements are merged into a single list, without duplicates.
         For dictionaries: all keys are added to a single dict, duplicate keys are merged recursively.
         For all other types: the first value is returned.
 
         It assumes that all duplicate keys in a dictionary have items of the same type.
+
+        Args:
+            included_keys: If given, only these top level keys are merged into the final dict.
         """
         if len(self.dictionaries) == 0:
             return {}
@@ -127,6 +130,8 @@ def simple_merge(self) -> dict:
         result = {}
         for dictionary in self.dictionaries:
             for key, item in dictionary.items():
+                if included_keys is not None and key not in included_keys:
+                    continue
                 if key in result:
                     if isinstance(item, list) or isinstance(item, set):
                         result[key] = self.merge_arrays(key, skip_duplicates=True)