wip

Signed-off-by: Marc 'risson' Schmitt <marc.schmitt@risson.space>

authentik/enterprise/providers/ssf/tasks.py

@@ -4,7 +4,6 @@ from uuid import UUID
 from django.http import HttpRequest
 from django.utils.timezone import now
 from dramatiq.actor import actor
-from dramatiq.composition import group
 from requests.exceptions import RequestException
 from structlog.stdlib import get_logger
 

authentik/events/tasks.py

@@ -1,6 +1,7 @@
 """Event notification tasks"""
 
 from uuid import UUID
+
 from django.db.models.query_utils import Q
 from dramatiq.actor import actor
 from guardian.shortcuts import get_anonymous_user

authentik/stages/authenticator_duo/api.py

@@ -159,7 +159,7 @@ class AuthenticatorDuoStageViewSet(UsedByMixin, ModelViewSet):
                 },
                 status=400,
             )
-        result = duo_import_devices.send(str(stage.pk)).get_result()
+        result = duo_import_devices.send(stage.pk).get_result()
         return Response(data=result, status=200 if result["error"] == "" else 400)
 
 

authentik/stages/authenticator_duo/tasks.py

@@ -1,5 +1,7 @@
 """duo tasks"""
 
+from uuid import UUID
+
 from dramatiq.actor import actor
 from structlog.stdlib import get_logger
 
@@ -10,7 +12,7 @@ LOGGER = get_logger()
 
 
 @actor(store_results=True)
-def duo_import_devices(stage_pk: str):
+def duo_import_devices(stage_pk: UUID):
     """Import duo devices"""
     created = 0
     stage: AuthenticatorDuoStage = AuthenticatorDuoStage.objects.filter(pk=stage_pk).first()

authentik/stages/authenticator_webauthn/tasks.py

@@ -14,7 +14,6 @@ from authentik.stages.authenticator_webauthn.models import (
     WebAuthnDeviceType,
 )
 from authentik.tasks.middleware import CurrentTask
-from authentik.tasks.models import Task, TaskStatus
 
 CACHE_KEY_MDS_NO = "goauthentik.io/stages/authenticator_webauthn/mds_no"
 AAGUID_BLOB_PATH = Path(__file__).parent / "mds" / "aaguid.json"
@@ -32,7 +31,7 @@ def mds_ca() -> bytes:
 @actor
 def webauthn_mds_import(force=False):
     """Background task to import FIDO Alliance MDS blob and AAGUIDs into database"""
-    self: Task = CurrentTask.get_task()
+    self = CurrentTask.get_task()
     with open(MDS_BLOB_PATH, mode="rb") as _raw_blob:
         blob = parse_blob(_raw_blob.read(), mds_ca())
     to_create_update = [
@@ -87,7 +86,4 @@ def webauthn_mds_import(force=False):
             unique_fields=["aaguid"],
         )
 
-    self.set_status(
-        TaskStatus.SUCCESSFUL,
-        "Successfully imported FIDO Alliance MDS blobs and AAGUIDs.",
-    )
+    self.info("Successfully imported FIDO Alliance MDS blobs and AAGUIDs.")

authentik/stages/email/tasks.py

@@ -1,7 +1,6 @@
 """email stage tasks"""
 
 from email.utils import make_msgid
-from smtplib import SMTPException
 from typing import Any
 
 from django.core.mail import EmailMultiAlternatives
@@ -17,7 +16,6 @@ from authentik.stages.authenticator_email.models import AuthenticatorEmailStage
 from authentik.stages.email.models import EmailStage
 from authentik.stages.email.utils import logo_data
 from authentik.tasks.middleware import CurrentTask
-from authentik.tasks.models import Task, TaskStatus
 
 LOGGER = get_logger()
 
@@ -56,29 +54,23 @@ def send_mail(
     email_stage_pk: str | None = None,
 ):
     """Send Email for Email Stage. Retries are scheduled automatically."""
-    self: Task = CurrentTask.get_task()
-    # TODO: fix me
-    # self.save_on_success = False
+    self = CurrentTask.get_task()
     message_id = make_msgid(domain=DNS_NAME)
     self.set_uid(slugify(message_id.replace(".", "_").replace("@", "_")))
-    try:
     if not stage_class_path or not email_stage_pk:
         stage = EmailStage(use_global_settings=True)
     else:
         stage_class = path_to_class(stage_class_path)
         stages = stage_class.objects.filter(pk=email_stage_pk)
         if not stages.exists():
-                self.set_status(
-                    TaskStatus.WARNING,
-                    "Email stage does not exist anymore. Discarding message.",
-                )
+            self.warning("Email stage does not exist anymore. Discarding message.")
             return
         stage: EmailStage | AuthenticatorEmailStage = stages.first()
     try:
         backend = stage.backend
     except ValueError as exc:
         LOGGER.warning("failed to get email backend", exc=exc)
-            self.set_error(exc)
+        self.error(exc)
         return
     backend.open()
     # Since django's EmailMessage objects are not JSON serialisable,
@@ -112,11 +104,4 @@ def send_mail(
         from_email=message_object.from_email,
         to_email=message_object.to,
     ).save()
-        self.set_status(
-            TaskStatus.SUCCESSFUL,
-            "Successfully sent Mail.",
-        )
-    except (SMTPException, ConnectionError, OSError) as exc:
-        LOGGER.debug("Error sending email, retrying...", exc=exc)
-        self.set_error(exc)
-        raise exc
+    self.info("Successfully sent mail.")