mirror of
https://github.com/apache/superset.git
synced 2026-04-07 10:31:50 +00:00
110 lines
3.8 KiB
Python
110 lines
3.8 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.
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
from contextlib import contextmanager
|
|
from functools import wraps
|
|
from typing import Any, Callable, Dict, Iterator, TYPE_CHECKING, Union
|
|
|
|
from flask import current_app, Response
|
|
|
|
from superset import is_feature_enabled
|
|
from superset.dashboards.commands.exceptions import DashboardAccessDeniedError
|
|
from superset.utils import core as utils
|
|
from superset.utils.dates import now_as_float
|
|
|
|
if TYPE_CHECKING:
|
|
from superset.stats_logger import BaseStatsLogger
|
|
|
|
|
|
@contextmanager
|
|
def stats_timing(stats_key: str, stats_logger: BaseStatsLogger) -> Iterator[float]:
|
|
"""Provide a transactional scope around a series of operations."""
|
|
start_ts = now_as_float()
|
|
try:
|
|
yield start_ts
|
|
except Exception as ex:
|
|
raise ex
|
|
finally:
|
|
stats_logger.timing(stats_key, now_as_float() - start_ts)
|
|
|
|
|
|
def arghash(args: Any, kwargs: Dict[str, Any]) -> int:
|
|
"""Simple argument hash with kwargs sorted."""
|
|
sorted_args = tuple(
|
|
x if hasattr(x, "__repr__") else x for x in [*args, *sorted(kwargs.items())]
|
|
)
|
|
return hash(sorted_args)
|
|
|
|
|
|
def debounce(duration: Union[float, int] = 0.1) -> Callable[..., Any]:
|
|
"""Ensure a function called with the same arguments executes only once
|
|
per `duration` (default: 100ms).
|
|
"""
|
|
|
|
def decorate(f: Callable[..., Any]) -> Callable[..., Any]:
|
|
last: Dict[str, Any] = {"t": None, "input": None, "output": None}
|
|
|
|
def wrapped(*args: Any, **kwargs: Any) -> Any:
|
|
now = time.time()
|
|
updated_hash = arghash(args, kwargs)
|
|
if (
|
|
last["t"] is None
|
|
or now - last["t"] >= duration
|
|
or last["input"] != updated_hash
|
|
):
|
|
result = f(*args, **kwargs)
|
|
last["t"] = time.time()
|
|
last["input"] = updated_hash
|
|
last["output"] = result
|
|
return result
|
|
return last["output"]
|
|
|
|
return wrapped
|
|
|
|
return decorate
|
|
|
|
|
|
def on_security_exception(self: Any, ex: Exception) -> Response:
|
|
return self.response(403, **{"message": utils.error_msg_from_exception(ex)})
|
|
|
|
|
|
# noinspection PyPackageRequirements
|
|
def check_dashboard_access(
|
|
on_error: Callable[..., Any] = on_security_exception
|
|
) -> Callable[..., Any]:
|
|
def decorator(f: Callable[..., Any]) -> Callable[..., Any]:
|
|
@wraps(f)
|
|
def wrapper(self: Any, *args: Any, **kwargs: Any) -> Any:
|
|
from superset.models.dashboard import Dashboard
|
|
|
|
dashboard = Dashboard.get(str(kwargs["dashboard_id_or_slug"]))
|
|
if is_feature_enabled("DASHBOARD_RBAC"):
|
|
try:
|
|
current_app.appbuilder.sm.raise_for_dashboard_access(dashboard)
|
|
except DashboardAccessDeniedError as ex:
|
|
return on_error(self, ex)
|
|
except Exception as exception:
|
|
raise exception
|
|
|
|
return f(self, *args, dashboard=dashboard, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|