Add save_user() function

* On successful login via OAuth2, save the details of the user in
  elasticsearch store, to avoid hitting the external provider for the
  basic details.

1 files changed, 7 insertions, 0 deletions

diff --git a/wqflask/utility/elasticsearch_tools.py b/wqflask/utility/elasticsearch_tools.py
index bc7bb240..74db489b 100644
--- a/wqflask/utility/elasticsearch_tools.py
+++ b/wqflask/utility/elasticsearch_tools.py
@@ -20,3 +20,10 @@ def get_user_by_unique_column(column_name, column_value):
     except TransportError as te: 
         pass
     return user_details
+
+def save_user(user, user_id, index="users", doc_type="local"):
+    es = Elasticsearch([{
+        "host": ELASTICSEARCH_HOST
+        , "port": ELASTICSEARCH_PORT
+    }])
+    es.create(index, doc_type, body=user, id=user_id)