From 98de7bf9649115b67a13018454a2c3766be1fc12 Mon Sep 17 00:00:00 2001
From: Muriithi Frederick Muriuki
Date: Fri, 19 Jan 2018 10:38:04 +0300
Subject: 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.
---
 wqflask/utility/elasticsearch_tools.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'wqflask')

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)
-- 
cgit v1.2.3