This is an automated email from the ASF dual-hosted git repository. graceguo pushed a commit to branch test-email-profile in repository https://gitbox.apache.org/repos/asf/incubator-superset.git
commit 6679302490ac36b7d03e7499cf0fe6f36f6a753c Author: Grace <[email protected]> AuthorDate: Fri Nov 13 21:49:08 2020 -0800 use firefox proxy --- setup.cfg | 2 +- superset/utils/machine_auth.py | 31 +++++++++++++++++++++++++++++++ superset/views/core.py | 1 + 3 files changed, 33 insertions(+), 1 deletion(-) diff --git a/setup.cfg b/setup.cfg index 44f7e0d..eaf0844 100644 --- a/setup.cfg +++ b/setup.cfg @@ -30,7 +30,7 @@ combine_as_imports = true include_trailing_comma = true line_length = 88 known_first_party = superset -known_third_party =alembic,apispec,backoff,bleach,cachelib,celery,click,colorama,contextlib2,croniter,cryptography,dateutil,flask,flask_appbuilder,flask_babel,flask_caching,flask_compress,flask_login,flask_migrate,flask_sqlalchemy,flask_talisman,flask_testing,flask_wtf,freezegun,geohash,geopy,humanize,isodate,jinja2,markdown,markupsafe,marshmallow,msgpack,numpy,pandas,parameterized,parsedatetime,pathlib2,polyline,prison,pyarrow,pyhive,pytest,pytz,retry,selenium,setuptools,simplejson,slac [...] +known_third_party =alembic,apispec,backoff,bleach,cachelib,celery,click,colorama,contextlib2,croniter,cryptography,dateutil,flask,flask_appbuilder,flask_babel,flask_caching,flask_compress,flask_login,flask_migrate,flask_sqlalchemy,flask_talisman,flask_testing,flask_wtf,freezegun,geohash,geopy,humanize,isodate,jinja2,markdown,markupsafe,marshmallow,msgpack,numpy,pandas,parameterized,parsedatetime,pathlib2,polyline,prison,pyarrow,pyhive,pytest,pytz,requests,retry,selenium,setuptools,simple [...] multi_line_output = 3 order_by_type = false diff --git a/superset/utils/machine_auth.py b/superset/utils/machine_auth.py index 778f2a6..3d08475 100644 --- a/superset/utils/machine_auth.py +++ b/superset/utils/machine_auth.py @@ -17,11 +17,18 @@ import importlib import logging +from tempfile import NamedTemporaryFile from typing import Callable, Dict, TYPE_CHECKING +from urllib.parse import urlparse +import requests from flask import current_app, Flask, request, Response, session from flask_login import login_user +from selenium import webdriver +from selenium.webdriver import FirefoxOptions from selenium.webdriver.remote.webdriver import WebDriver +from selenium.webdriver.support import expected_conditions as EC +from selenium.webdriver.support.ui import WebDriverWait from werkzeug.http import parse_cookie from superset.utils.urls import headless_url @@ -49,8 +56,31 @@ class MachineAuthProvider: if self._auth_webdriver_func_override: return self._auth_webdriver_func_override(driver, user) + print("\n\n i am user:{}".format(user)) + with NamedTemporaryFile() as file: + file.write( + requests.get( + "https://github.com/bewisse/modheader_selenium/blob/master/firefox-modheader/modheader.xpi?raw=true"; + ).content + ) + + driver.install_addon(file.name) + + print("\n\n setting header \n\n") + driver.get( + "https://webdriver.bewisse.com/add?X-Internalauth-Username=svc_di_analytics_products"; + ) + WebDriverWait(driver, 1).until(EC.title_is("Done")) + print("\n\n installed") + # Setting cookies requires doing a request first driver.get(headless_url("/login/")) + if urlparse(driver.current_url).path == "/agree-to-access-policy": + print("\n\n agree-to-access-policy....") + driver.find_element_by_tag_name("button").click() + + print("\n\n i am url:{}\n\n".format(driver.current_url)) + print("\n\n i am page source:{}".format(driver.page_source)) if user: cookies = self.get_auth_cookies(user) @@ -59,6 +89,7 @@ class MachineAuthProvider: else: cookies = {} + print(f"\n\n i am cookies:{cookies}") for cookie_name, cookie_val in cookies.items(): driver.add_cookie(dict(name=cookie_name, value=cookie_val)) diff --git a/superset/views/core.py b/superset/views/core.py index 97eef1f..9c8fd8c 100755 --- a/superset/views/core.py +++ b/superset/views/core.py @@ -1669,6 +1669,7 @@ class Superset(BaseSupersetView): # pylint: disable=too-many-public-methods # set a default value to appease pylint update_log_payload: Callable[..., None] = lambda **kwargs: None, ) -> FlaskResponse: + print(f"\n\n got headers: {request.headers}\n\n") """Server side rendering for a dashboard""" session = db.session() qry = session.query(Dashboard)
