From 213867b07def83f9a7e6af55e979ec465cad8bfa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Oriol=20L=C3=B3pez-Doriga?= Date: Wed, 19 Jul 2023 12:33:56 +0200 Subject: [PATCH] fixed dataset and cohorts response by id --- beacon/response/build_response.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/beacon/response/build_response.py b/beacon/response/build_response.py index fdf5b85..8f4adcc 100644 --- a/beacon/response/build_response.py +++ b/beacon/response/build_response.py @@ -139,6 +139,14 @@ def build_beacon_resultset_response_by_dataset(data, elif doc['individualId'] in dataset_dict['ids'][0]['individualIds']: dataset_id = dataset_dict['dataset'] response_dict[dataset_id].append(doc) + elif str(entity_schema) == 'DefaultSchemas.DATASETS': + if doc['id'] == dataset_dict['dataset']: + dataset_id = dataset_dict['dataset'] + response_dict[dataset_id].append(doc) + elif str(entity_schema) == 'DefaultSchemas.COHORTS': + if doc['id'] == dataset_dict['dataset']: + dataset_id = dataset_dict['dataset'] + response_dict[dataset_id].append(doc) else: if doc['id'] in dataset_dict['ids'][0]['biosampleIds']: dataset_id = dataset_dict['dataset']