mirror of https://github.com/jumpserver/jumpserver
86 lines
2.2 KiB
Python
86 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from django.db.models.signals import post_save, post_delete
|
|
from django.db.utils import ProgrammingError
|
|
from django.dispatch import receiver
|
|
from django.utils.functional import LazyObject
|
|
|
|
from assets.models import Asset
|
|
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 db_port_manager, 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_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(django_ready)
|
|
def check_db_port_mapper(sender, **kwargs):
|
|
logger.info('Init db port mapper')
|
|
try:
|
|
db_port_manager.check()
|
|
except (ProgrammingError,) as e:
|
|
pass
|
|
|
|
|
|
@receiver(post_save, sender=Asset)
|
|
def on_db_app_created(sender, instance: Asset, created, **kwargs):
|
|
if not instance.category != 'database':
|
|
return
|
|
if not created:
|
|
return
|
|
db_port_manager.add(instance)
|
|
|
|
|
|
@receiver(post_delete, sender=Asset)
|
|
def on_db_app_delete(sender, instance, **kwargs):
|
|
if not instance.category != 'database':
|
|
return
|
|
db_port_manager.pop(instance)
|
|
|
|
|
|
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()
|