Create script to dumple sample data given the dataset and trait id

* scripts/sampledata.py: New file.

1 files changed, 50 insertions, 0 deletions

diff --git a/scripts/sampledata.py b/scripts/sampledata.py
new file mode 100644
index 00000000..e1b38f3c
--- /dev/null
+++ b/scripts/sampledata.py
@@ -0,0 +1,50 @@
+import json
+
+# Required Evils!
+from flask import g
+from wqflask import app
+
+from wqflask.show_trait import show_trait
+from wqflask.database import database_connection
+
+
+class UserSessionSimulator():
+    def __init__(self, user_id):
+        self._user_id = user_id
+
+    @property
+    def user_id(self):
+        return self._user_id
+
+
+def dump_sample_data(dataset_name, trait_id):
+    """Givena DATASET_NAME e.g. 'BXDPublish' and a TRAIT_ID
+    e.g. '10007', dump the sample data as json object"""
+    with database_connection() as conn, conn.cursor() as cursor:
+        sample_data = {"headers": ["Name", "Value", "SE"], "data": []}
+
+        with app.app_context():
+            g.user_session = UserSessionSimulator(None)
+            data = show_trait.ShowTrait(
+                cursor, user_id=None,
+                kw={
+                    "trait_id": "10007",
+                    "dataset": "BXDPublish",
+                }
+            )
+            attributes = data.js_data.get("attributes")
+            for id_ in attributes:
+                sample_data["headers"].append(attributes[id_].name)
+            for sample in data.js_data.get("sample_lists")[0]:
+                sample_data["data"].append(
+                    [
+                        sample.name,
+                        sample.value or 'x',
+                        sample.variance or 'x',
+                        *[str(sample.extra_attributes.get(str(key), "x"))
+                          for key in attributes],
+                    ])
+            return sample_data
+
+
+print(json.dumps(dump_sample_data("BXDPublish", "10007")))