Add a function that fetches all the traits given the right params

* scripts/sampledata.py (fetch_all_traits): New function.

1 files changed, 12 insertions, 1 deletions

diff --git a/scripts/sampledata.py b/scripts/sampledata.py
index e1b38f3c..fb3668d6 100644
--- a/scripts/sampledata.py
+++ b/scripts/sampledata.py
@@ -6,6 +6,7 @@ from wqflask import app
 
 from wqflask.show_trait import show_trait
 from wqflask.database import database_connection
+from wqflask.search_results import SearchResultPage
 
 
 class UserSessionSimulator():
@@ -47,4 +48,14 @@ def dump_sample_data(dataset_name, trait_id):
             return sample_data
 
 
-print(json.dumps(dump_sample_data("BXDPublish", "10007")))
+def fetch_all_traits(species, group, type_, dataset):
+    with app.app_context():
+        g.user_session = UserSessionSimulator(None)
+        for result in SearchResultPage({
+                "species": species,
+                "group": group,
+                "type": type_,
+                "dataset": dataset,
+                "search_terms_or": "*",
+        }).trait_list:
+            yield result.get('name')