diff options
| -rw-r--r-- | scripts/sampledata.py | 50 |
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"))) |