Files
superset2/superset/commands/tasks/exceptions.py
2026-02-09 10:45:56 -08:00

107 lines
3.1 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 flask_babel import lazy_gettext as _
from superset.commands.exceptions import (
CommandException,
CommandInvalidError,
CreateFailedError,
ForbiddenError,
UpdateFailedError,
)
class TaskNotFoundError(CommandException):
"""Task not found."""
status = 404
message = "Task not found."
class TaskInvalidError(CommandInvalidError):
"""Task parameters are invalid."""
message = _("Task parameters are invalid.")
class TaskCreateFailedError(CreateFailedError):
"""Task creation failed."""
message = _("Task could not be created.")
class TaskUpdateFailedError(UpdateFailedError):
"""Task update failed."""
message = _("Task could not be updated.")
class TaskAbortFailedError(CommandException):
"""Task abortion failed."""
status = 422
message = _("Task could not be aborted.")
class TaskNotAbortableError(CommandException):
"""
Task cannot be aborted.
Raised when attempting to abort an in-progress task that has not
registered an abort handler (is_abortable is not True).
"""
status = 400
message = _(
"Task is not abortable. The task is in progress but has not "
"registered an abort handler."
)
class TaskForbiddenError(ForbiddenError):
"""Task operation forbidden."""
message = _("Changing this task is forbidden")
class TaskPermissionDeniedError(ForbiddenError):
"""Task operation not permitted for current user."""
def __init__(self, message: str | None = None):
super().__init__()
if message:
self.message = message
else:
self.message = _("You do not have permission to perform this operation")
class GlobalTaskFrameworkDisabledError(CommandException):
"""
Raised when a GTF task is called or scheduled but GTF is disabled.
This exception is raised at call/schedule time (not decoration time) to allow
modules with @task-decorated functions to be imported safely when GTF is disabled.
The check is deferred until someone actually tries to execute a task.
"""
message = _(
"The Global Task Framework is not enabled. "
"Set GLOBAL_TASK_FRAMEWORK=True in your feature flags to use @task. "
"See https://superset.apache.org/docs/configuration/async-queries-celery "
"for configuration details."
)