def get_ledger_section_entry_combinations(domain): """Get all section / entry combinations in a domain. :returns: a generator of namedtuples with fields ``section_id``, ``entry_id``, ``doc_count`` """ terms = [ AggregationTerm('section_id', 'section_id'), AggregationTerm('entry_id', 'entry_id'), ] query = LedgerES().domain(domain) return NestedTermAggregationsHelper(base_query=query, terms=terms).get_data()
def get_app_submission_breakdown_es(domain_name, monthspan): terms = [ AggregationTerm('app_id', 'app_id'), AggregationTerm('device_id', 'form.meta.deviceID'), AggregationTerm('user_id', 'form.meta.userID'), AggregationTerm('username', 'form.meta.username'), ] query = FormES().domain(domain_name).submitted( gte=monthspan.startdate, lt=monthspan.computed_enddate, ) return NestedTermAggregationsHelper(base_query=query, terms=terms).get_data()
def get_aggregated_ledger_values(domain, case_ids, section_id, entry_ids=None): # todo: figure out why this causes circular import query = LedgerES().domain(domain).section(section_id).case(case_ids) if entry_ids: query = query.entry(entry_ids) terms = [ AggregationTerm('entry_id', 'entry_id'), ] return NestedTermAggregationsHelper( base_query=query, terms=terms, inner_most_aggregation=SumAggregation('balance', 'balance'), ).get_data()
def get_app_submission_breakdown_es(domain_name, monthspan, user_ids=None): # takes > 1 m to load at 50k worker scale terms = [ AggregationTerm('app_id', 'app_id'), AggregationTerm('device_id', 'form.meta.deviceID'), AggregationTerm('user_id', 'form.meta.userID'), AggregationTerm('username', 'form.meta.username'), ] query = FormES(es_instance_alias=ES_EXPORT_INSTANCE).domain(domain_name).submitted( gte=monthspan.startdate, lt=monthspan.computed_enddate, ) if user_ids is not None: query = query.user_id(user_ids) return NestedTermAggregationsHelper(base_query=query, terms=terms).get_data()
def test_nested_terms_helper(self): json_output = { "query": { "filtered": { "filter": { "and": [{ "match_all": {} }] }, "query": { "match_all": {} } } }, "aggs": { "app_id": { "terms": { "field": "app_id", "size": SIZE_LIMIT, }, "aggs": { "user_id": { "terms": { "field": "user_id", "size": SIZE_LIMIT, }, "aggs": { "balance": { "sum": { "field": "balance" } } } } } } }, "size": SIZE_LIMIT } base_query = HQESQuery('cases') query = NestedTermAggregationsHelper( base_query=base_query, terms=[ AggregationTerm('app_id', 'app_id'), AggregationTerm('user_id', 'user_id') ], inner_most_aggregation=SumAggregation('balance', 'balance')).query self.checkQuery(query, json_output)
def initialize_ledger_combinations(apps, schema_editor): terms = [ AggregationTerm('domain', 'domain'), AggregationTerm('section_id', 'section_id'), AggregationTerm('entry_id', 'entry_id'), ] combos = [ a for a in NestedTermAggregationsHelper(base_query=LedgerES(), terms=terms).get_data() ] for combo in combos: LedgerSectionEntry.objects.get_or_create( domain=combo.domain, section_id=combo.section_id, entry_id=combo.entry_id, )