Files
superset2/superset/exceptions.py
Rob DiCiuccio 4d329071a1 feat(SIP-39): Async query support for charts (#11499)
* Generate JWT in Flask app

* Refactor chart data API query logic, add JWT validation and async worker

* Add redis stream implementation, refactoring

* Add chart data cache endpoint, refactor QueryContext caching

* Typing, linting, refactoring

* pytest fixes and openapi schema update

* Enforce caching be configured for async query init

* Async query processing for explore_json endpoint

* Add /api/v1/async_event endpoint

* Async frontend for dashboards [WIP]

* Chart async error message support, refactoring

* Abstract asyncEvent middleware

* Async chart loading for Explore

* Pylint fixes

* asyncEvent middleware -> TypeScript, JS linting

* Chart data API: enforce forced_cache, add tests

* Add tests for explore_json endpoints

* Add test for chart data cache enpoint (no login)

* Consolidate set_and_log_cache and add STORE_CACHE_KEYS_IN_METADATA_DB flag

* Add tests for tasks/async_queries and address PR comments

* Bypass non-JSON result formats for async queries

* Add tests for redux middleware

* Remove debug statement

Co-authored-by: Ville Brofeldt <33317356+villebro@users.noreply.github.com>

* Skip force_cached if no queryObj

* SunburstViz: don't modify self.form_data

* Fix failing annotation test

* Resolve merge/lint issues

* Reduce polling delay

* Fix new getClientErrorObject reference

* Fix flakey unit tests

* /api/v1/async_event: increment redis stream ID, add tests

* PR feedback: refactoring, configuration

* Fixup: remove debugging

* Fix typescript errors due to redux upgrade

* Update UPDATING.md

* Fix failing py tests

* asyncEvent_spec.js -> asyncEvent_spec.ts

* Refactor flakey Python 3.7 mock assertions

* Fix another shared state issue in Py tests

* Use 'sub' claim in JWT for user_id

* Refactor async middleware config

* Fixup: restore FeatureFlag boolean type

Co-authored-by: Ville Brofeldt <33317356+villebro@users.noreply.github.com>
2020-12-10 20:21:56 -08:00

114 lines
2.9 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 typing import Any, Dict, List, Optional
from flask_babel import gettext as _
from superset.errors import ErrorLevel, SupersetError, SupersetErrorType
class SupersetException(Exception):
status = 500
message = ""
def __init__(
self, message: str = "", exception: Optional[Exception] = None,
) -> None:
if message:
self.message = message
self._exception = exception
super().__init__(self.message)
@property
def exception(self) -> Optional[Exception]:
return self._exception
class SupersetTimeoutException(SupersetException):
status = 408
def __init__(
self,
error_type: SupersetErrorType,
message: str,
level: ErrorLevel,
extra: Optional[Dict[str, Any]],
) -> None:
super(SupersetTimeoutException, self).__init__(message)
self.error = SupersetError(
error_type=error_type, message=message, level=level, extra=extra
)
class SupersetSecurityException(SupersetException):
status = 401
def __init__(
self, error: SupersetError, payload: Optional[Dict[str, Any]] = None
) -> None:
super(SupersetSecurityException, self).__init__(error.message)
self.error = error
self.payload = payload
class SupersetVizException(SupersetException):
status = 400
def __init__(self, errors: List[SupersetError]) -> None:
super(SupersetVizException, self).__init__(str(errors))
self.errors = errors
class NoDataException(SupersetException):
status = 400
class NullValueException(SupersetException):
status = 400
class SupersetTemplateException(SupersetException):
pass
class SpatialException(SupersetException):
pass
class CertificateException(SupersetException):
message = _("Invalid certificate")
class DatabaseNotFound(SupersetException):
status = 400
class QueryObjectValidationError(SupersetException):
status = 400
class CacheLoadError(SupersetException):
status = 404
class DashboardImportException(SupersetException):
pass
class SerializationError(SupersetException):
pass