#!/usr/bin/env python3
+"""Helper functions for dealing with dictionaries."""
+
from itertools import islice
from typing import Any, Callable, Dict, Iterator, List, Tuple
"""
items = d.items()
for x in range(0, len(d), size):
- yield {key: value for (key, value) in islice(items, x, x + size)}
+ yield dict(islice(items, x, x + size))
-def coalesce_by_creating_list(key, new_value, old_value):
+def coalesce_by_creating_list(_, new_value, old_value):
from list_utils import flatten
return flatten([new_value, old_value])
return set(coalesce_by_creating_list(key, new_value, old_value))
-def coalesce_last_write_wins(key, new_value, old_value):
+def coalesce_last_write_wins(_, new_value, discarded_old_value):
return new_value
-def coalesce_first_write_wins(key, new_value, old_value):
+def coalesce_first_write_wins(_, discarded_new_value, old_value):
return old_value
def coalesce(
inputs: Iterator[Dict[Any, Any]],
*,
- aggregation_function: Callable[[Any, Any], Any] = coalesce_by_creating_list,
+ aggregation_function: Callable[[Any, Any, Any], Any] = coalesce_by_creating_list,
) -> Dict[Any, Any]:
"""Merge N dicts into one dict containing the union of all keys /
values in the input dicts. When keys collide, apply the
['scott', '555-1212', '123 main st.', '12345']
"""
- r = ([], [])
+ r: Tuple[List[Any], List[Any]] = ([], [])
for (k, v) in d.items():
r[0].append(k)
r[1].append(v)