added type hint to parameter idexer; formatted code; used keyword arg…

…uments in calls to _es.count()

src/utils/stats.py

@@ -1,22 +1,23 @@
+from biothings.utils.es import ESIndexer
 
-def update_stats(idxer,assembly):
-    stats = compute_stats(idxer,assembly)
+
+def update_stats(idxer: ESIndexer, assembly):
+    stats = compute_stats(idxer, assembly)
     # now update _meta.stats
     m = idxer.get_mapping_meta()
-    m["_meta"].get("stats",{}).update(stats)
+    m["_meta"].get("stats", {}).update(stats)
     idxer.update_mapping_meta(m)
     return m["_meta"]["stats"]
 
 
-def compute_stats(idxer,assembly):
-    stats = {}
+def compute_stats(idxer: ESIndexer, assembly):
+    stats = dict()
     stats["total"] = idxer.count()
-    for k in [assembly,"observed","vcf"]:
-        stats[k] = count_field(idxer,k)
+    for k in [assembly, "observed", "vcf"]:
+        stats[k] = count_field(idxer, k)
     return stats
 
 
-def count_field(idxer,field):
-    q = {"query":{"exists":{"field":field}}}
-    return idxer._es.count(idxer._index,idxer._doc_type,q)["count"]
-
+def count_field(idxer: ESIndexer, field):
+    q = {"query": {"exists": {"field": field}}}
+    return idxer._es.count(index=idxer._index, doc_type=idxer._doc_type, body=q)["count"]