Add secret key support

CPSSD · Apr 27, 2016 · d21ccae · d21ccae
1 parent e8cdd40
commit d21ccae
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/visualisations/model_error/model_error.py b/visualisations/model_error/model_error.py
@@ -32,6 +32,9 @@ def db_get(collection, query, projection):
         'query': query,
         'projection': projection
     }
+    key = os.getenv('SECRETKEY')
+    if key is not None:
+        db_req['key'] = key
     str_bson_db_req = str(bson.BSON.encode(db_req))
     raw_db_result = gearman_client.submit_job('db-get', str_bson_db_req).result
     db_result = bson.BSON.decode(bson.BSON(raw_db_result))