feat/data-api (#65)

* fix: merge mistake
serratus-bio · Aug 28, 2023 · b38ad38 · b38ad38
1 parent 7195a30
commit b38ad38
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/route/data.py b/route/data.py
@@ -89,6 +89,11 @@ def data_query(arguments):
         arguments['_offset'] = 0
 
     data_session = sessionmaker(bind=SQLAlchemyEngine)()
+
+    data_entities = list_attributes_of_model(arguments['view'])
+    if('_columns' in arguments):
+        data_entities = list(map(lambda x: getattr(arguments['view'], x), filter(lambda x: hasattr(arguments['view'], x), arguments['_columns'])))
+
     data_query = (
         data_session
             .query(arguments['view'])