mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.9 KiB
52 lines
1.9 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.dispatch import receiver
|
|
from django.db.models.signals import post_save, pre_save
|
|
from django.conf import settings
|
|
from django.db.utils import ProgrammingError, OperationalError
|
|
|
|
from jumpserver.utils import current_request
|
|
from .models import Setting
|
|
from .utils import get_logger
|
|
from .signals import django_ready, ldap_auth_enable
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
@receiver(post_save, sender=Setting, dispatch_uid="my_unique_identifier")
|
|
def refresh_settings_on_changed(sender, instance=None, **kwargs):
|
|
logger.debug("Receive setting item change")
|
|
logger.debug(" - refresh setting: {}".format(instance.name))
|
|
if instance:
|
|
instance.refresh_setting()
|
|
|
|
|
|
@receiver(django_ready, dispatch_uid="my_unique_identifier")
|
|
def refresh_all_settings_on_django_ready(sender, **kwargs):
|
|
logger.debug("Receive django ready signal")
|
|
logger.debug(" - fresh all settings")
|
|
try:
|
|
Setting.refresh_all_settings()
|
|
except (ProgrammingError, OperationalError):
|
|
pass
|
|
|
|
|
|
@receiver(ldap_auth_enable, dispatch_uid="my_unique_identifier")
|
|
def ldap_auth_on_changed(sender, enabled=True, **kwargs):
|
|
if enabled:
|
|
logger.debug("Enable LDAP auth")
|
|
if settings.AUTH_LDAP_BACKEND not in settings.AUTHENTICATION_BACKENDS:
|
|
settings.AUTHENTICATION_BACKENDS.insert(0, settings.AUTH_LDAP_BACKEND)
|
|
|
|
else:
|
|
logger.debug("Disable LDAP auth")
|
|
if settings.AUTH_LDAP_BACKEND in settings.AUTHENTICATION_BACKENDS:
|
|
settings.AUTHENTICATION_BACKENDS.remove(settings.AUTH_LDAP_BACKEND)
|
|
|
|
|
|
@receiver(pre_save, dispatch_uid="my_unique_identifier")
|
|
def on_create_set_created_by(sender, instance=None, **kwargs):
|
|
if hasattr(instance, 'created_by') and not instance.created_by:
|
|
if current_request and current_request.user.is_authenticated:
|
|
instance.created_by = current_request.user.name
|