From ce6beb7182ae87eb326351ebc7e5ef313aec572d Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Wed, 11 Sep 2019 07:55:35 +0100 Subject: [PATCH] #51: Use updated include filter --- common/database_helpers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/common/database_helpers.py b/common/database_helpers.py index 03e155c2..5166e39a 100644 --- a/common/database_helpers.py +++ b/common/database_helpers.py @@ -330,8 +330,8 @@ def get_filtered_read_query_results(filter_handler, filters, query): results = query.get_all_results() if query.include_related_entities: for query_filter in filters: - if list(query_filter)[0].lower() == "include": - return list(map(lambda x: x.to_nested_dict(query_filter["include"]), results)) + if type(query_filter) is IncludeFilter: + return list(map(lambda x: x.to_nested_dict(query_filter.included_filters), results)) return list(map(lambda x: x.to_dict(), results)) finally: