From f4decec103c0550e54c823726fc1998a9e187207 Mon Sep 17 00:00:00 2001 From: Zhouyang Lian Date: Mon, 13 Jul 2020 12:12:22 -0400 Subject: [PATCH] fix --- datanator_query_python/util/mongo_util.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/datanator_query_python/util/mongo_util.py b/datanator_query_python/util/mongo_util.py index 86927f4..8c185f9 100644 --- a/datanator_query_python/util/mongo_util.py +++ b/datanator_query_python/util/mongo_util.py @@ -180,7 +180,8 @@ def build_taxon_object(self, _id, _format="tax_id"): obj = self.client["datanator-test"]["taxon_tree"].find_one({_format: _id}, projection={"canon_anc_ids": 1, "canon_anc_names": 1, - "tax_name": 1}) + "tax_name": 1, + "tax_id": 1}) if obj is None: return {} else: @@ -190,7 +191,7 @@ def build_taxon_object(self, _id, _format="tax_id"): for _id, name in zip(canon_anc_ids, canon_anc_names): canon_ancestors.append({"ncbi_taxonomy_id": _id, "name": name}) - return {"ncbi_taxonomy_id": taxon_id, + return {"ncbi_taxonomy_id": obj["tax_id"], "name": obj["tax_name"], "canon_ancestors": canon_ancestors} \ No newline at end of file