✨ feat: useListMergeByKey

Author: mic1on

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "usepy"
-version = "0.2.6"
+version = "0.2.7"
 description = "usepy"
 homepage = "https://usepy.code05.com/"
 authors = ["miclon <jcnd@163.com>"]

src/usepy/__init__.py

@@ -28,6 +28,7 @@
     'useListSort',
     'useListUnique',
     'useListify',
+    'useListMergeByKey',
     'usePath',
     'useRandomString',
     'useRandomUUID',
@@ -96,6 +97,7 @@ def find_module_alias():
     'useListSome': 'usepy.core.useList:useListSome',
     'useListSort': 'usepy.core.useList:useListSort',
     'useListUnique': 'usepy.core.useList:useListUnique',
+    'useListMergeByKey': 'usepy.core.useList:useListMergeByKey',
     'useRandomString': 'usepy.core.useRandom:useRandomString',
     'useRandomUUID': 'usepy.core.useRandom:useRandomUUID',
     'useStringMiddle': 'usepy.core.useString:useStringMiddle',

src/usepy/core/useList.py

@@ -302,3 +302,21 @@ def useListUnique(array: List, fn: Optional[Callable] = None):
     :return: 去重后数组
     """
     return list(useList.dedupe(array, fn))
+
+
+def useListMergeByKey(collection1: List, collection2: List, key: str):
+    """
+    数组中对象按key合并
+    """
+    if not collection1 or not collection2:
+        return collection1 or collection2
+    if key not in collection1[0] or key not in collection2[0]:
+        return collection1 + collection2
+    array1_dict = useList.objs_to_obj(collection1, key)
+    array2_dict = useList.objs_to_obj(collection2, key)
+    for _key in array2_dict:
+        if _key in array1_dict:
+            array1_dict[_key].update(array2_dict[_key])
+        else:
+            array1_dict[_key] = array2_dict[_key]
+    return list(array1_dict.values())

tests/test_core/test_useCleanHtml.py

@@ -20,8 +20,9 @@ def tag_stripper():
     yield ts
     ts.close()
 
+
 def test_close_method(tag_stripper):
     extracted_data = tag_stripper.get_data()
     assert extracted_data == "<h1>Title</h1>"
     tag_stripper.close()
-    assert tag_stripper.get_data() == ""
+    assert tag_stripper.get_data() == ""

tests/test_core/test_useList.py

@@ -1,7 +1,7 @@
 import pytest
 
 from usepy import useList, useListFilter, useListFlatten, useListDifference, useListEvery, useListSome, useListSort, \
-    useListUnique
+    useListUnique, useListMergeByKey
 
 
 def test_split():
@@ -118,3 +118,15 @@ def test_useListEvery(array, fn, expected):
 )
 def test_useListSome(array, fn, expected):
     assert useListSome(array, fn) == expected
+
+
+@pytest.mark.parametrize(
+    "array1,array2,key,expected",
+    [
+        ([{"id": 1, "age": 18}], [{"id": 1, "name": "miclon"}], "id", [{"id": 1, "age": 18, "name": "miclon"}]),
+        ([{"id": 1, "age": 18}], [{"name": "miclon"}], "id", [{'id': 1, "age": 18}, {'name': 'miclon'}]),
+        ([{"id": 1, "age": 18}], [], "id", [{"id": 1, "age": 18}]),
+    ]
+)
+def test_useListMergeByKey(array1, array2, key, expected):
+    assert useListMergeByKey(array1, array2, key) == expected