#51: Use updated include filter

ral-facilities · Sep 11, 2019 · ce6beb7 · ce6beb7
1 parent 958da33
commit ce6beb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: