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: 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