7249f84e18
* dev: remove len for empty comparison * dev: using in instead of multiple ors * dev: assign expression to empty variables * dev: use f-string * dev: remove list comprehension and use generators * dev: remove assert from paginator * dev: use is for identity comparison with singleton * dev: remove unnecessary else statements * dev: fix does not exists error for both project and workspace * dev: remove reimports * dev: iterate a dictionary * dev: remove unused commented code * dev: remove redefinition * dev: remove unused imports * dev: remove unused imports * dev: remove unnecessary f strings * dev: remove unused variables * dev: use literal structure to create the data structure * dev: add empty lines at the end of the file * dev: remove user middleware * dev: remove unnecessary default None
164 lines
7.8 KiB
Python
164 lines
7.8 KiB
Python
def resolve_keys(group_keys, value):
|
|
"""resolve keys to a key which will be used for
|
|
grouping
|
|
|
|
Args:
|
|
group_keys (string): key which will be used for grouping
|
|
value (obj): data value
|
|
|
|
Returns:
|
|
string: the key which will be used for
|
|
"""
|
|
keys = group_keys.split(".")
|
|
for key in keys:
|
|
value = value.get(key, None)
|
|
return value
|
|
|
|
|
|
def group_results(results_data, group_by, sub_group_by=False):
|
|
"""group results data into certain group_by
|
|
|
|
Args:
|
|
results_data (obj): complete results data
|
|
group_by (key): string
|
|
|
|
Returns:
|
|
obj: grouped results
|
|
"""
|
|
if sub_group_by:
|
|
main_responsive_dict = dict()
|
|
|
|
if sub_group_by == "priority":
|
|
main_responsive_dict = {
|
|
"urgent": {},
|
|
"high": {},
|
|
"medium": {},
|
|
"low": {},
|
|
"none": {},
|
|
}
|
|
|
|
for value in results_data:
|
|
main_group_attribute = resolve_keys(sub_group_by, value)
|
|
group_attribute = resolve_keys(group_by, value)
|
|
if isinstance(main_group_attribute, list) and not isinstance(group_attribute, list):
|
|
if len(main_group_attribute):
|
|
for attrib in main_group_attribute:
|
|
if str(attrib) not in main_responsive_dict:
|
|
main_responsive_dict[str(attrib)] = {}
|
|
if str(group_attribute) in main_responsive_dict[str(attrib)]:
|
|
main_responsive_dict[str(attrib)][str(group_attribute)].append(value)
|
|
else:
|
|
main_responsive_dict[str(attrib)][str(group_attribute)] = []
|
|
main_responsive_dict[str(attrib)][str(group_attribute)].append(value)
|
|
else:
|
|
if str(None) not in main_responsive_dict:
|
|
main_responsive_dict[str(None)] = {}
|
|
|
|
if str(group_attribute) in main_responsive_dict[str(None)]:
|
|
main_responsive_dict[str(None)][str(group_attribute)].append(value)
|
|
else:
|
|
main_responsive_dict[str(None)][str(group_attribute)] = []
|
|
main_responsive_dict[str(None)][str(group_attribute)].append(value)
|
|
|
|
elif isinstance(group_attribute, list) and not isinstance(main_group_attribute, list):
|
|
if str(main_group_attribute) not in main_responsive_dict:
|
|
main_responsive_dict[str(main_group_attribute)] = {}
|
|
if len(group_attribute):
|
|
for attrib in group_attribute:
|
|
if str(attrib) in main_responsive_dict[str(main_group_attribute)]:
|
|
main_responsive_dict[str(main_group_attribute)][str(attrib)].append(value)
|
|
else:
|
|
main_responsive_dict[str(main_group_attribute)][str(attrib)] = []
|
|
main_responsive_dict[str(main_group_attribute)][str(attrib)].append(value)
|
|
else:
|
|
if str(None) in main_responsive_dict[str(main_group_attribute)]:
|
|
main_responsive_dict[str(main_group_attribute)][str(None)].append(value)
|
|
else:
|
|
main_responsive_dict[str(main_group_attribute)][str(None)] = []
|
|
main_responsive_dict[str(main_group_attribute)][str(None)].append(value)
|
|
|
|
elif isinstance(group_attribute, list) and isinstance(main_group_attribute, list):
|
|
if len(main_group_attribute):
|
|
for main_attrib in main_group_attribute:
|
|
if str(main_attrib) not in main_responsive_dict:
|
|
main_responsive_dict[str(main_attrib)] = {}
|
|
if len(group_attribute):
|
|
for attrib in group_attribute:
|
|
if str(attrib) in main_responsive_dict[str(main_attrib)]:
|
|
main_responsive_dict[str(main_attrib)][str(attrib)].append(value)
|
|
else:
|
|
main_responsive_dict[str(main_attrib)][str(attrib)] = []
|
|
main_responsive_dict[str(main_attrib)][str(attrib)].append(value)
|
|
else:
|
|
if str(None) in main_responsive_dict[str(main_attrib)]:
|
|
main_responsive_dict[str(main_attrib)][str(None)].append(value)
|
|
else:
|
|
main_responsive_dict[str(main_attrib)][str(None)] = []
|
|
main_responsive_dict[str(main_attrib)][str(None)].append(value)
|
|
else:
|
|
if str(None) not in main_responsive_dict:
|
|
main_responsive_dict[str(None)] = {}
|
|
if len(group_attribute):
|
|
for attrib in group_attribute:
|
|
if str(attrib) in main_responsive_dict[str(None)]:
|
|
main_responsive_dict[str(None)][str(attrib)].append(value)
|
|
else:
|
|
main_responsive_dict[str(None)][str(attrib)] = []
|
|
main_responsive_dict[str(None)][str(attrib)].append(value)
|
|
else:
|
|
if str(None) in main_responsive_dict[str(None)]:
|
|
main_responsive_dict[str(None)][str(None)].append(value)
|
|
else:
|
|
main_responsive_dict[str(None)][str(None)] = []
|
|
main_responsive_dict[str(None)][str(None)].append(value)
|
|
else:
|
|
main_group_attribute = resolve_keys(sub_group_by, value)
|
|
group_attribute = resolve_keys(group_by, value)
|
|
|
|
if str(main_group_attribute) not in main_responsive_dict:
|
|
main_responsive_dict[str(main_group_attribute)] = {}
|
|
|
|
if str(group_attribute) in main_responsive_dict[str(main_group_attribute)]:
|
|
main_responsive_dict[str(main_group_attribute)][str(group_attribute)].append(value)
|
|
else:
|
|
main_responsive_dict[str(main_group_attribute)][str(group_attribute)] = []
|
|
main_responsive_dict[str(main_group_attribute)][str(group_attribute)].append(value)
|
|
|
|
return main_responsive_dict
|
|
|
|
else:
|
|
response_dict = {}
|
|
|
|
if group_by == "priority":
|
|
response_dict = {
|
|
"urgent": [],
|
|
"high": [],
|
|
"medium": [],
|
|
"low": [],
|
|
"none": [],
|
|
}
|
|
|
|
for value in results_data:
|
|
group_attribute = resolve_keys(group_by, value)
|
|
if isinstance(group_attribute, list):
|
|
if len(group_attribute):
|
|
for attrib in group_attribute:
|
|
if str(attrib) in response_dict:
|
|
response_dict[str(attrib)].append(value)
|
|
else:
|
|
response_dict[str(attrib)] = []
|
|
response_dict[str(attrib)].append(value)
|
|
else:
|
|
if str(None) in response_dict:
|
|
response_dict[str(None)].append(value)
|
|
else:
|
|
response_dict[str(None)] = []
|
|
response_dict[str(None)].append(value)
|
|
else:
|
|
if str(group_attribute) in response_dict:
|
|
response_dict[str(group_attribute)].append(value)
|
|
else:
|
|
response_dict[str(group_attribute)] = []
|
|
response_dict[str(group_attribute)].append(value)
|
|
|
|
return response_dict
|