feat: add global task framework (#36368)

superset/commands/tasks/exceptions.py

@@ -0,0 +1,106 @@
+# 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."
+    )