add list2dict in utils.py, add its unittest

Author: supercoderhawk

pysenal/utils/utils.py

@@ -1,4 +1,5 @@
 # -*- coding: UTF-8 -*-
+import copy
 from typing import Iterable
 
 
@@ -24,3 +25,26 @@ def get_chunk(l, n):
 
         if chunk:
             yield chunk
+
+
+def list2dict(l, key, pop_key=False):
+    """
+    convert list of dict to dict
+    :param l: list of dict
+    :param key: key name, which value of dict in list as returned dict key
+    :param pop_key: whether pop the key in list of
+    :return: assembled dict
+    """
+    if type(l) not in {list, tuple}:
+        raise TypeError('input must in list or tuple type')
+    d = {}
+    for item in l:
+        if not isinstance(item, dict):
+            raise TypeError('item {0} is not dict'.format(item))
+        if key not in item:
+            raise KeyError('key is not in item')
+        new_item = copy.deepcopy(item)
+        if pop_key:
+            new_item.pop(key)
+        d[item[key]] = new_item
+    return d

tests/utils/test_utils.py

@@ -35,3 +35,24 @@ def test_get_chunk():
     assert next(chunk_g3, [10, 100, 200])
     with pytest.raises(StopIteration):
         next(chunk_g3)
+
+
+def test_list2dict():
+    l1 = [{'pid': '1', 'title': 'haha'}, {'pid': '2', 'title': 'lalala'}]
+    expected_l1 = {'1': {'pid': '1', 'title': 'haha'}, '2': {'pid': '2', 'title': 'lalala'}}
+    ret_l1 = list2dict(l1, 'pid')
+    l1[0]['pid'] = '11'
+    l1[1]['pid'] = '22'
+
+    assert ret_l1 == expected_l1
+    with pytest.raises(KeyError):
+        list2dict(l1, 'patent_id')
+    l1.append(1)
+    with pytest.raises(TypeError):
+        list2dict(l1, 'pid')
+
+    l2 = [{'name': 'a', 'count': 11}, {'name': 'b', 'count': 2}]
+    expected_l2 = {'a': {'count': 11}, 'b': {'count': 2}}
+    assert list2dict(l2, 'name', pop_key=True) == expected_l2
+    with pytest.raises(TypeError):
+        list2dict(1, 'pid')