1cfe1aff13
* root: initial rename * web: rename custom element prefix * root: rename external functions with pb_ prefix * root: fix formatting * root: replace domain with goauthentik.io * proxy: update path * root: rename remaining prefixes * flows: rename file extension * root: pbadmin -> akadmin * docs: fix image filenames * lifecycle: ignore migration files * ci: copy default config from current source before loading last tagged * *: new sentry dsn * tests: fix missing python3.9-dev package * root: add additional migrations for service accounts created by outposts * core: mark system-created service accounts with attribute * policies/expression: fix pb_ replacement not working * web: fix last linting errors, add lit-analyse * policies/expressions: fix lint errors * web: fix sidebar display on screens where not all items fit * proxy: attempt to fix proxy pipeline * proxy: use go env GOPATH to get gopath * lib: fix user_default naming inconsistency * docs: add upgrade docs * docs: update screenshots to use authentik * admin: fix create button on empty-state of outpost * web: fix modal submit not refreshing SiteShell and Table * web: fix height of app-card and height of generic icon * web: fix rendering of subtext * admin: fix version check error not being caught * web: fix worker count not being shown * docs: update screenshots * root: new icon * web: fix lint error * admin: fix linting error * root: migrate coverage config to pyproject
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
"""Tasks API"""
|
|
from importlib import import_module
|
|
|
|
from django.contrib import messages
|
|
from django.http.response import Http404
|
|
from django.utils.translation import gettext_lazy as _
|
|
from drf_yasg2.utils import swagger_auto_schema
|
|
from rest_framework.decorators import action
|
|
from rest_framework.fields import CharField, DateTimeField, IntegerField, ListField
|
|
from rest_framework.permissions import IsAdminUser
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
from rest_framework.serializers import Serializer
|
|
from rest_framework.viewsets import ViewSet
|
|
|
|
from authentik.lib.tasks import TaskInfo
|
|
|
|
|
|
class TaskSerializer(Serializer):
|
|
"""Serialize TaskInfo and TaskResult"""
|
|
|
|
task_name = CharField()
|
|
task_description = CharField()
|
|
task_finish_timestamp = DateTimeField(source="finish_timestamp")
|
|
|
|
status = IntegerField(source="result.status.value")
|
|
messages = ListField(source="result.messages")
|
|
|
|
def create(self, request: Request) -> Response:
|
|
raise NotImplementedError
|
|
|
|
def update(self, request: Request) -> Response:
|
|
raise NotImplementedError
|
|
|
|
|
|
class TaskViewSet(ViewSet):
|
|
"""Read-only view set that returns all background tasks"""
|
|
|
|
permission_classes = [IsAdminUser]
|
|
|
|
@swagger_auto_schema(responses={200: TaskSerializer(many=True)})
|
|
def list(self, request: Request) -> Response:
|
|
"""List current messages and pass into Serializer"""
|
|
return Response(TaskSerializer(TaskInfo.all().values(), many=True).data)
|
|
|
|
@action(detail=True, methods=["post"])
|
|
# pylint: disable=invalid-name
|
|
def retry(self, request: Request, pk=None) -> Response:
|
|
"""Retry task"""
|
|
task = TaskInfo.by_name(pk)
|
|
if not task:
|
|
raise Http404
|
|
try:
|
|
task_module = import_module(task.task_call_module)
|
|
task_func = getattr(task_module, task.task_call_func)
|
|
task_func.delay(*task.task_call_args, **task.task_call_kwargs)
|
|
messages.success(
|
|
self.request,
|
|
_(
|
|
"Successfully re-scheduled Task %(name)s!"
|
|
% {"name": task.task_name}
|
|
),
|
|
)
|
|
return Response(
|
|
{
|
|
"successful": True,
|
|
}
|
|
)
|
|
except ImportError:
|
|
# if we get an import error, the module path has probably changed
|
|
task.delete()
|
|
return Response({"successful": False})
|