53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import logging
|
|
|
|
|
|
def extract_search_filters(result_set):
|
|
"""
|
|
Returned object should look something like:
|
|
|
|
{
|
|
"singles": [], # non tree tags
|
|
"entry_1": [ 'tag1', 'tag2' ],
|
|
"entry_2": [ 'tag1', 'tag2' ],
|
|
}
|
|
"""
|
|
filter_dict = {
|
|
'singles': set(),
|
|
}
|
|
for item in result_set:
|
|
# import ipdb; ipdb.set_trace()
|
|
try:
|
|
# extract tags
|
|
tags = item.tags.all()
|
|
for tag in tags:
|
|
if len(tag.name.split('/')) == 1:
|
|
filter_dict['singles'].add(tag.name)
|
|
else:
|
|
# set penultimate tag as header
|
|
chunks = tag.name.split('/')
|
|
header = chunks[-2]
|
|
name = chunks[-1]
|
|
# check if
|
|
entry = filter_dict.get(header)
|
|
if entry is None:
|
|
filter_dict[header] = set()
|
|
filter_dict[header].add(name)
|
|
# extract attributes
|
|
attributes = item.attributes.all()
|
|
for tag in attributes:
|
|
if len(tag.name.split('/')) == 1:
|
|
filter_dict['singles'].add(tag.name)
|
|
else:
|
|
# set penultimate tag as header
|
|
chunks = tag.name.split('/')
|
|
header = chunks[-2]
|
|
name = chunks[-1]
|
|
# check if
|
|
entry = filter_dict.get(header)
|
|
if entry is None:
|
|
filter_dict[header] = set()
|
|
filter_dict[header].add(name)
|
|
except Exception as e:
|
|
logging.error(f'Extacting filters for {item}')
|
|
return filter_dict
|