170 lines
4.0 KiB
Python
170 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Search data tests
|
|
iterator_handling.data_search
|
|
"""
|
|
|
|
from corelibs.debug_handling.dump_data import dump_data
|
|
from corelibs.iterator_handling.data_search import find_in_array_from_list, ArraySearchList
|
|
|
|
|
|
def main() -> None:
|
|
"""
|
|
Comment
|
|
"""
|
|
data = [
|
|
{
|
|
"lookup_value_p": "A01",
|
|
"lookup_value_c": "B01",
|
|
"replace_value": "R01",
|
|
},
|
|
{
|
|
"lookup_value_p": "A02",
|
|
"lookup_value_c": "B02",
|
|
"replace_value": "R02",
|
|
},
|
|
{
|
|
"lookup_value_p": "A03",
|
|
"lookup_value_c": "B03",
|
|
"replace_value": "R03",
|
|
},
|
|
]
|
|
test_foo = ArraySearchList(
|
|
key="lookup_value_p",
|
|
value="A01"
|
|
)
|
|
result = find_in_array_from_list(data, [test_foo])
|
|
print(f"Search A: {dump_data(test_foo)} -> {dump_data(result)}")
|
|
|
|
search: list[ArraySearchList] = [
|
|
{
|
|
"key": "lookup_value_p",
|
|
"value": "A01"
|
|
},
|
|
{
|
|
"key": "lookup_value_c",
|
|
"value": "B01"
|
|
},
|
|
]
|
|
result = find_in_array_from_list(data, search)
|
|
print(f"Search B: {dump_data(search)} -> {dump_data(result)}")
|
|
|
|
search: list[ArraySearchList] = [
|
|
{
|
|
"key": "lookup_value_p",
|
|
"value": "A01"
|
|
},
|
|
{
|
|
"key": "lookup_value_c",
|
|
"value": "B01"
|
|
},
|
|
{
|
|
"key": "lookup_value_c",
|
|
"value": "B02"
|
|
},
|
|
]
|
|
try:
|
|
result = find_in_array_from_list(data, search)
|
|
print(f"Search C: {dump_data(search)} -> {dump_data(result)}")
|
|
except KeyError as e:
|
|
print(f"Search C raised KeyError: {e}")
|
|
|
|
search: list[ArraySearchList] = [
|
|
{
|
|
"key": "lookup_value_p",
|
|
"value": "A01"
|
|
},
|
|
{
|
|
"key": "lookup_value_c",
|
|
"value": ["B01", "B02"]
|
|
},
|
|
]
|
|
try:
|
|
result = find_in_array_from_list(data, search)
|
|
print(f"Search D: {dump_data(search)} -> {dump_data(result)}")
|
|
except KeyError as e:
|
|
print(f"Search D raised KeyError: {e}")
|
|
|
|
search: list[ArraySearchList] = [
|
|
{
|
|
"key": "lookup_value_p",
|
|
"value": ["A01", "A03"]
|
|
},
|
|
{
|
|
"key": "lookup_value_c",
|
|
"value": ["B01", "B02"]
|
|
},
|
|
]
|
|
try:
|
|
result = find_in_array_from_list(data, search)
|
|
print(f"Search E: {dump_data(search)} -> {dump_data(result)}")
|
|
except KeyError as e:
|
|
print(f"Search E raised KeyError: {e}")
|
|
|
|
search: list[ArraySearchList] = [
|
|
{
|
|
"key": "lookup_value_p",
|
|
"value": "NOT FOUND"
|
|
},
|
|
]
|
|
try:
|
|
result = find_in_array_from_list(data, search)
|
|
print(f"Search F: {dump_data(search)} -> {dump_data(result)}")
|
|
except KeyError as e:
|
|
print(f"Search F raised KeyError: {e}")
|
|
|
|
data = [
|
|
{
|
|
"sd_user_id": "1593",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1592",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1596",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1594",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1595",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1861",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1862",
|
|
"email": "",
|
|
"employee_id": ""
|
|
},
|
|
{
|
|
"sd_user_id": "1860",
|
|
"email": "",
|
|
"employee_id": ""
|
|
}
|
|
]
|
|
result = find_in_array_from_list(data, [ArraySearchList(
|
|
key="sd_user_id",
|
|
value="1593"
|
|
)])
|
|
print(f"Search F: -> {dump_data(result)}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
# __END__
|