From ae1d0901ecb32fc1db83c74f25dfc9e49f0090a2 Mon Sep 17 00:00:00 2001
From: Maxime Beauchemin <maximebeauchemin@gmail.com>
Date: Wed, 7 Feb 2018 12:39:04 -0800
Subject: [PATCH] Lyft-specific views

(cherry picked from commit 1a7362708cf7953c1cb61e8e3c492e3522659ea7)
(cherry picked from commit 6931e777931efe7189d5db02ee612d74c03ce86a)
---
 superset/views/__init__.py |  1 +
 superset/views/lyft.py     | 46 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+)
 create mode 100644 superset/views/lyft.py

diff --git a/superset/views/__init__.py b/superset/views/__init__.py
index ab93a55ce7c13..0d7c058054452 100644
--- a/superset/views/__init__.py
+++ b/superset/views/__init__.py
@@ -3,3 +3,4 @@
 from . import core  # noqa
 from . import sql_lab  # noqa
 from . import annotations # noqa
+from . import lyft # noqa
diff --git a/superset/views/lyft.py b/superset/views/lyft.py
new file mode 100644
index 0000000000000..0eb7a901367ee
--- /dev/null
+++ b/superset/views/lyft.py
@@ -0,0 +1,46 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import os
+import logging
+from flask import Response, request
+from flask_babel import gettext as __
+
+from superset import app, appbuilder, utils
+import superset.models.core as models
+from superset.views.core import Superset
+
+config = app.config
+stats_logger = config.get('STATS_LOGGER')
+log_this = models.Log.log_this
+DAR = models.DatasourceAccessRequest
+
+
+ALL_DATASOURCE_ACCESS_ERR = __(
+    'This endpoint requires the `all_datasource_access` permission')
+DATASOURCE_MISSING_ERR = __('The datasource seems to have been deleted')
+ACCESS_REQUEST_MISSING_ERR = __(
+    'The access requests seem to have been deleted')
+USER_MISSING_ERR = __('The user seems to have been deleted')
+DATASOURCE_ACCESS_ERR = __("You don't have access to this datasource")
+SECRET_KEY = os.getenv("CREDENTIALS_SUPERSET_SECRET_KEY") or None
+
+
+def json_success(json_msg, status=200):
+    return Response(json_msg, status=status, mimetype='application/json')
+
+
+class Lyft(Superset):
+
+    def datasource_access(self, datasource, user=None):
+        if SECRET_KEY is None:
+            logging.error('No secret loaded')
+            return False
+
+        tom_request_key = request.headers.get('TOM_ACCESS_KEY')
+        return tom_request_key == SECRET_KEY
+
+
+appbuilder.add_view_no_menu(Lyft)