child_controller.py

import operator

import connexion

from swagger_server.models.child import Child  # noqa: E501
from swagger_server.models.error_model import ErrorModel  # noqa: E501
from swagger_server import util

import config
from controllers.query_controller import *


def list_child(lang, orphacode, hchid):  # noqa: E501
    """Search for a clinical entity's child(ren) by ORPHAcode and the unique identifier of a classification

    The result retrieves the clinical entity's ORPHAcode and preferred term as well as the unique identifier and the name of the queried classification, and the clinical entity's child(ren), specifying ORPHAcode and preferred term. # noqa: E501

    :param lang: Language
    :type lang: str
    :param orphacode: A unique and time-stable numerical identifier attributed randomly by the Orphanet database to each clinical entity upon its creation.
    :type orphacode: int
    :param hchid: A unique and time-stable numerical identifier attributed randomly by the Orphanet database to each classification upon its creation.
    :type hchid: int

    :rtype: Child
    """
    es = config.elastic_server

    index = "rdcode_orphaclassification_{}_{}".format(hchid, lang.lower())

    query = "{\"query\": {\"match\": {\"ORPHAcode\": \"" + str(orphacode) + "\"}}," \
            "\"_source\":[\"Date\"," \
                         "\"Classification.ID of the classification\"," \
                         "\"Classification.Name of the classification\"," \
                         "\"ORPHAcode\"," \
                         "\"Preferred term\"," \
                         "\"Child\"]}"

    response = single_res(es, index, query)

    # Test to return error
    if isinstance(response, str) or isinstance(response, tuple):
        return response
    else:
        if not response["Child"]:
            response_child = []
        else:
            code_list = ",".join(["\"" + str(code) + "\"" for code in response["Child"]])
            query = "{\"query\": {\"terms\": {\"ORPHAcode\": [" + code_list + "]}}," \
                    "\"_source\":[\"ORPHAcode\", \"Preferred term\"]}"

            response_child = multiple_res(es, index, query, 1000)
            # Test to return error
            if isinstance(response_child, str) or isinstance(response_child, tuple):
                return response_child
            response_child.sort(key=operator.itemgetter('ORPHAcode'))
        response["Child"] = response_child

        # return yaml if needed
        response = if_yaml(connexion.request.accept_mimetypes.best, response)
    return response