mirror of
https://github.com/apache/superset.git
synced 2026-04-09 19:35:21 +00:00
b35b1d7633
Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: Daniel Gaspar <danielvazgaspar@gmail.com>
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
import urllib
|
|
from typing import Any
|
|
from urllib.parse import urlparse
|
|
|
|
from flask import current_app as app, has_request_context, url_for
|
|
|
|
|
|
def get_url_host(user_friendly: bool = False) -> str:
|
|
if user_friendly:
|
|
return app.config["WEBDRIVER_BASEURL_USER_FRIENDLY"]
|
|
return app.config["WEBDRIVER_BASEURL"]
|
|
|
|
|
|
def headless_url(path: str, user_friendly: bool = False) -> str:
|
|
return urllib.parse.urljoin(get_url_host(user_friendly=user_friendly), path)
|
|
|
|
|
|
def get_url_path(view: str, user_friendly: bool = False, **kwargs: Any) -> str:
|
|
in_request_context = has_request_context()
|
|
|
|
# When already in a request context, Flask's url_for respects SCRIPT_NAME from
|
|
# the WSGI environment, so the prefix is already included. Only add APPLICATION_ROOT
|
|
# prefix when creating a new request context.
|
|
if in_request_context:
|
|
url = url_for(view, **kwargs)
|
|
else:
|
|
with app.test_request_context():
|
|
url = url_for(view, **kwargs)
|
|
app_root = app.config.get("APPLICATION_ROOT", "/")
|
|
if app_root != "/" and not url.startswith(app_root):
|
|
url = app_root.rstrip("/") + url
|
|
|
|
return headless_url(url, user_friendly=user_friendly)
|
|
|
|
|
|
def modify_url_query(url: str, **kwargs: Any) -> str:
|
|
"""
|
|
Replace or add parameters to a URL.
|
|
"""
|
|
parts = list(urllib.parse.urlsplit(url))
|
|
params = urllib.parse.parse_qs(parts[3])
|
|
for k, v in kwargs.items():
|
|
if not isinstance(v, list):
|
|
v = [v]
|
|
params[k] = v
|
|
|
|
parts[3] = "&".join(
|
|
f"{k}={urllib.parse.quote(str(v[0]))}" for k, v in params.items()
|
|
)
|
|
return urllib.parse.urlunsplit(parts)
|
|
|
|
|
|
def is_secure_url(url: str) -> bool:
|
|
"""
|
|
Validates if a URL is secure (uses HTTPS).
|
|
|
|
:param url: The URL to validate.
|
|
:return: True if the URL uses HTTPS (secure), False if it uses HTTP (non-secure).
|
|
"""
|
|
parsed_url = urlparse(url)
|
|
return parsed_url.scheme == "https"
|