[gentoo-commits] proj/R_overlay:master commit in: roverlay/util/

["André Erdmann" <dywi@mailerd.de>]

commit:     2cc79dc0280b25cfabdd8c5166a35f61ff17935c
Author:     André Erdmann <dywi <AT> mailerd <DOT> de>
AuthorDate: Thu Sep 12 16:30:15 2013 +0000
Commit:     André Erdmann <dywi <AT> mailerd <DOT> de>
CommitDate: Thu Sep 12 16:30:15 2013 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/R_overlay.git;a=commit;h=2cc79dc0

roverlay/util/dictwalk: dictmerge()

Convert an iterable of iterables with non-unique keys into a dict with one list
per key.

E.g.
[ ( "a", 1, 2 ), ( "a", 2, 3 ), ( "b", 1, 2 ) ]
-> dict( "a": [(1,2),(2,3)], "b": [(1,2)] )

---
 roverlay/util/dictwalk.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/roverlay/util/dictwalk.py b/roverlay/util/dictwalk.py
index 52a8c61..0813d11 100644
--- a/roverlay/util/dictwalk.py
+++ b/roverlay/util/dictwalk.py
@@ -7,6 +7,27 @@
 import roverlay.util.namespace
 import roverlay.util.objects
 
+def dictmerge_inplace ( d, iterable, get_key=None, get_value=None ):
+   keyfunc = ( lambda x: x[0]  ) if get_key   is None else get_key
+   valfunc = ( lambda x: x[1:] ) if get_value is None else get_value
+
+   for item in iterable:
+      key = keyfunc ( item )
+      val = valfunc ( item )
+      entry = d.get ( key )
+      if entry is None:
+         d[key] = [ val ]
+      else:
+         #assert isinstance ( entry, list )
+         entry.append ( val )
+# --- end of dictmerge_inplace (...) ---
+
+def dictmerge ( iterable, dict_cls=dict, get_key=None, get_value=None ):
+   ret = dict_cls()
+   dictmerge_inplace ( ret, iterable, get_key=get_key, get_value=get_value )
+   return ret
+# --- end of dictmerge (...) ---
+
 
 def dictwalk_create_parent_v ( root, path, dict_create=None, cautious=True ):
    """Creates a dict tree structure using keys the given path. The last path