def get_analytics_for_domain(sqs, data_domain):
# State Analytic
if data_domain == SEARCH_DATA_DOMAIN_PEOPLE:
states_raw = sqs.facet_counts().get('fields', {}).get('loc_state', [])
if len(states_raw) > 5:
states_raw = states_raw[:5]
states = [state for state in states_raw if state[1] > 0]
return get_state_bar_graph(states)
# Sector Analytic
elif data_domain == SEARCH_DATA_DOMAIN_RECORD:
orgs_raw = sqs.facet_counts().get('fields', {}).get('sector_id', [])
orgs = \
[(ORG_TYPE_SECTOR_MAPPING.get(i[0], 'Unknown'), i[1]) for i in orgs_raw if i[1] > 0]
return get_sector_analytics(orgs)
# Not Supported
else:
_log.error('Search Data Domain: %s is not supported' %data_domain)
def get_sector_choices():
sector_choices = [(x, ORG_TYPE_SECTOR_MAPPING.get('%s' % x)) for x in range(1, 7)]
sector_choices = sorted(sector_choices, key=itemgetter(1))
return sector_choices
