elasticsearchclient.py 文件源码

def get_efo_info_from_code(self, efo_codes, **kwargs):
        params = SearchParams(**kwargs)

        if not isinstance(efo_codes, list):
            efo_codes = [efo_codes]

        query_body = addict.Dict()
        query_body.query.ids["values"] = efo_codes
        query_body.size = params.size
        if params.facets == 'true':
            query_body.aggregations.significant_therapeutic_areas.significant_terms.field = "therapeutic_labels.keyword"
            query_body.aggregations.significant_therapeutic_areas.significant_terms.size = 25
            query_body.aggregations.significant_therapeutic_areas.significant_terms.min_doc_count = 2

        if params.fields:
            query_body._source = params.fields

        if efo_codes:
            res = self._cached_search(index=self._index_efo,
                                      doc_type=self._docname_efo,
                                      body=query_body.to_dict()
                                      )
            return PaginatedResult(res, params)