mirror of https://github.com/jumpserver/jumpserver
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
from django.db.models.signals import post_save, post_delete
|
|
from django.dispatch import receiver
|
|
from django.utils.functional import LazyObject
|
|
|
|
from common.signals import django_ready
|
|
from common.utils import get_logger
|
|
from common.utils.connection import RedisPubSub
|
|
from orgs.utils import tmp_to_builtin_org
|
|
from ..models import Applet, AppletHost
|
|
from ..utils import DBPortManager
|
|
|
|
db_port_manager: DBPortManager
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
@receiver(post_save, sender=AppletHost)
|
|
def on_applet_host_create(sender, instance, created=False, **kwargs):
|
|
if not created:
|
|
return
|
|
applets = Applet.objects.all()
|
|
instance.applets.set(applets)
|
|
with tmp_to_builtin_org(system=1):
|
|
instance.generate_accounts()
|
|
|
|
applet_host_change_pub_sub.publish(True)
|
|
|
|
|
|
@receiver(post_delete, sender=AppletHost)
|
|
def on_applet_host_delete(sender, instance, **kwargs):
|
|
applet_host_change_pub_sub.publish(True)
|
|
|
|
|
|
@receiver(post_save, sender=Applet)
|
|
def on_applet_create(sender, instance, created=False, **kwargs):
|
|
if not created:
|
|
return
|
|
hosts = AppletHost.objects.all()
|
|
instance.hosts.set(hosts)
|
|
|
|
applet_host_change_pub_sub.publish(True)
|
|
|
|
|
|
@receiver(post_delete, sender=Applet)
|
|
def on_applet_delete(sender, instance, **kwargs):
|
|
applet_host_change_pub_sub.publish(True)
|
|
|
|
|
|
class AppletHostPubSub(LazyObject):
|
|
def _setup(self):
|
|
self._wrapped = RedisPubSub('fm.applet_host_change')
|
|
|
|
|
|
@receiver(django_ready)
|
|
def subscribe_applet_host_change(sender, **kwargs):
|
|
logger.debug("Start subscribe for expire node assets id mapping from memory")
|
|
|
|
def on_change(message):
|
|
from terminal.connect_methods import ConnectMethodUtil
|
|
ConnectMethodUtil.refresh_methods()
|
|
|
|
applet_host_change_pub_sub.subscribe(on_change)
|
|
|
|
|
|
applet_host_change_pub_sub = AppletHostPubSub()
|