def handle(self, domain, **options):
csv = options.get('csv')
couch_counts = map_counter_doc_types(_get_couchdb_counts(domain))
sql_counts = map_counter_doc_types(_get_sql_counts(domain))
es_counts = map_counter_doc_types(get_es_counts_by_doc_type(domain))
all_doc_types = set(couch_counts) | set(sql_counts) | set(es_counts)
output_rows = []
for doc_type in sorted(all_doc_types, key=lambda d: d.lower()):
couch = couch_counts.get(doc_type, '')
sql = sql_counts.get(doc_type, '')
es = es_counts.get(doc_type, '')
output_rows.append((doc_type, couch, sql, es))
if csv:
row_formatter = CSVRowFormatter()
else:
row_formatter = TableRowFormatter(
[50, 20, 20, 20],
_get_row_color
)
SimpleTableWriter(self.stdout, row_formatter).write_table(
['Doc Type', 'Couch', 'SQL', 'ES'], output_rows
)
def handle(self, domain, **options):
csv = options.get('csv')
couch_counts = map_counter_doc_types(_get_couchdb_counts(domain))
sql_counts = map_counter_doc_types(_get_sql_counts(domain))
es_counts = map_counter_doc_types(get_es_counts_by_doc_type(domain))
all_doc_types = set(couch_counts) | set(sql_counts) | set(es_counts)
output_rows = []
for doc_type in sorted(all_doc_types, key=lambda d: d.lower()):
couch = couch_counts.get(doc_type, '')
sql = sql_counts.get(doc_type, '')
es = es_counts.get(doc_type, '')
output_rows.append((doc_type, couch, sql, es))
if csv:
row_formatter = CSVRowFormatter()
else:
row_formatter = TableRowFormatter(
[50, 20, 20, 20],
_get_row_color
)
SimpleTableWriter(self.stdout, row_formatter).write_table(
['Doc Type', 'Couch', 'SQL', 'ES'], output_rows
)
def get_doc_counts_for_domain(domain):
"""
:param domain:
:return: List of tuples: ``(doc_type, primary_db_count, es_count)``
"""
primary_db_counts = map_counter_doc_types(_get_primary_db_counts(domain))
es_counts = map_counter_doc_types(
get_es_counts_by_doc_type(domain, (es.CaseES, es.FormES)))
es_counts.update(get_es_user_counts_by_doc_type(domain))
all_doc_types = set(primary_db_counts) | set(es_counts)
output_rows = []
for doc_type in sorted(all_doc_types, key=lambda d: d.lower()):
output_rows.append(
(doc_type, primary_db_counts[doc_type], es_counts[doc_type]))
return output_rows
def get_doc_counts_for_domain(domain):
"""
:param domain:
:return: List of tuples: ``(doc_type, primary_db_count, es_count)``
"""
primary_db_counts = map_counter_doc_types(_get_primary_db_counts(domain))
es_counts = map_counter_doc_types(
get_es_counts_by_doc_type(domain, (es.CaseES, es.FormES))
)
es_counts.update(get_es_user_counts_by_doc_type(domain))
all_doc_types = set(primary_db_counts) | set(es_counts)
output_rows = []
for doc_type in sorted(all_doc_types, key=lambda d: d.lower()):
output_rows.append((
doc_type,
primary_db_counts[doc_type],
es_counts[doc_type]
))
return output_rows
