This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/sources/ldap/tasks.py

42 lines
1.3 KiB
Python
Raw Normal View History

2019-10-11 11:43:35 +00:00
"""LDAP Sync tasks"""
from time import time
from django.core.cache import cache
2020-10-16 12:17:47 +00:00
from ldap3.core.exceptions import LDAPException
2020-10-16 12:17:47 +00:00
from passbook.lib.tasks import MonitoredTask, TaskResult, TaskResultStatus
from passbook.root.celery import CELERY_APP
from passbook.sources.ldap.models import LDAPSource
from passbook.sources.ldap.sync import LDAPSynchronizer
@CELERY_APP.task()
2020-10-16 12:17:47 +00:00
def ldap_sync_all():
"""Sync all sources"""
for source in LDAPSource.objects.filter(enabled=True):
2020-10-16 12:17:47 +00:00
ldap_sync.delay(source.pk)
2020-09-14 21:35:01 +00:00
2020-10-16 12:17:47 +00:00
@CELERY_APP.task(bind=True, base=MonitoredTask)
def ldap_sync(self: MonitoredTask, source_pk: int):
2020-09-14 21:35:01 +00:00
"""Sync a single source"""
source: LDAPSource = LDAPSource.objects.get(pk=source_pk)
2020-10-16 12:17:47 +00:00
try:
syncer = LDAPSynchronizer(source)
user_count = syncer.sync_users()
group_count = syncer.sync_groups()
syncer.sync_membership()
cache_key = source.state_cache_prefix("last_sync")
cache.set(cache_key, time(), timeout=60 * 60)
self.set_status(
TaskResult(
TaskResultStatus.SUCCESSFUL,
[f"Synced {user_count} users", f"Synced {group_count} groups"],
uid=source.name,
2020-10-16 12:17:47 +00:00
)
)
except LDAPException as exc:
self.set_status(
TaskResult(TaskResultStatus.ERROR, uid=source.name).with_error(exc)
)