authentik.enterprise.providers.google_workspace.tasks

 1"""Google Provider tasks"""
 2
 3from django.utils.translation import gettext_lazy as _
 4from dramatiq.actor import actor
 5
 6from authentik.enterprise.providers.google_workspace.models import GoogleWorkspaceProvider
 7from authentik.lib.sync.outgoing.tasks import SyncTasks
 8
 9sync_tasks = SyncTasks(GoogleWorkspaceProvider)
10
11
12@actor(description=_("Sync Google Workspace provider objects."))
13def google_workspace_sync_objects(*args, **kwargs):
14    return sync_tasks.sync_objects(*args, **kwargs)
15
16
17@actor(description=_("Full sync for Google Workspace provider."))
18def google_workspace_sync(provider_pk: int, *args, **kwargs):
19    """Run full sync for Google Workspace provider"""
20    return sync_tasks.sync(provider_pk, google_workspace_sync_objects)
21
22
23@actor(description=_("Sync a direct object (user, group) for Google Workspace provider."))
24def google_workspace_sync_direct(*args, **kwargs):
25    return sync_tasks.sync_signal_direct(*args, **kwargs)
26
27
28@actor(
29    description=_("Dispatch deletions for an object (user, group) for Google Workspace providers.")
30)
31def google_workspace_sync_delete_dispatch(*args, **kwargs):
32    return sync_tasks.sync_signal_delete_dispatch(google_workspace_sync_delete, *args, **kwargs)
33
34
35@actor(description=_("Delete an object (user, group) for Google Workspace provider."))
36def google_workspace_sync_delete(*args, **kwargs):
37    return sync_tasks.sync_signal_delete(*args, **kwargs)
38
39
40@actor(
41    description=_(
42        "Dispatch syncs for a direct object (user, group) for Google Workspace providers."
43    )
44)
45def google_workspace_sync_direct_dispatch(*args, **kwargs):
46    return sync_tasks.sync_signal_direct_dispatch(google_workspace_sync_direct, *args, **kwargs)
47
48
49@actor(description=_("Sync a related object (memberships) for Google Workspace provider."))
50def google_workspace_sync_m2m(*args, **kwargs):
51    return sync_tasks.sync_signal_m2m(*args, **kwargs)
52
53
54@actor(
55    description=_(
56        "Dispatch syncs for a related object (memberships) for Google Workspace providers."
57    )
58)
59def google_workspace_sync_m2m_dispatch(*args, **kwargs):
60    return sync_tasks.sync_signal_m2m_dispatch(google_workspace_sync_m2m, *args, **kwargs)