Merge pull request #1 from jgoetzinger/topic/dictionary_functions

Helper function to split a list into a dictionary, from @jgoetzinger

Author: Emantor

labgridhelper/barebox.py

@@ -1,5 +1,5 @@
 from labgrid.driver import BareboxDriver
-
+from labgridhelper.dict import split_to_dict
 
 def get_commands(command):
     """Returns the available commands of a running Barebox bootloader
@@ -26,11 +26,4 @@ def get_globals(command):
     """
     assert isinstance(command, BareboxDriver)
     out = command.run_check("global")
-    out = map(lambda x: x[2:], out)
-    global_variables = {}
-    for line in out:
-        sep = line.index(":")
-        key = line[:sep]
-        value = line[sep+2:]
-        global_variables[key] = value
-    return global_variables
+    return split_to_dict(out, delimiter=":", strip_chars="* ")

labgridhelper/dict.py

@@ -0,0 +1,22 @@
+def split_to_dict(command_output, delimiter="=", strip_chars=""):
+    '''
+    returns a dictionary that splits the content of the command_output.
+    The key is the part before and the value is the part after the first
+    delimiter. Optional chars can be specified to be stript from key and
+    value.
+    Args:
+        command_output: output to split into dict
+        delimiter: character that seperates key and value
+        strip_chars: characters to strip from key and value
+    Returns:
+        output_dict: dictionary that contains the pairs of key and value
+    '''
+    output_dict = {}
+    for line in command_output:
+        try:
+            if delimiter in line: # lines without a delimiter are ignored
+                (key, value) = line.split(delimiter, 1)
+                output_dict[key.strip(strip_chars)] = value.strip(strip_chars)
+        except ValueError:
+            pass
+    return output_dict