mirror of https://github.com/jumpserver/jumpserver
27 lines
778 B
Python
27 lines
778 B
Python
|
from django.db.models.signals import pre_save, post_save
|
||
|
|
from django.dispatch import receiver
|
||
|
|
|
||
|
|
from assets.models import Asset
|
||
|
|
from common.decorator import on_transaction_commit
|
||
|
|
from common.utils import get_logger
|
||
|
|
from .automations.push_account.manager import PushAccountManager
|
||
|
|
from .models import Account
|
||
|
|
|
||
|
|
logger = get_logger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@receiver(pre_save, sender=Account)
|
||
|
|
def on_account_pre_create(sender, instance, **kwargs):
|
||
|
|
# 升级版本号
|
||
|
|
instance.version += 1
|
||
|
|
# 即使在 root 组织也不怕
|
||
|
|
instance.org_id = instance.asset.org_id
|
||
|
|
|
||
|
|
|
||
|
|
@receiver(post_save, sender=Asset)
|
||
|
|
@on_transaction_commit
|
||
|
|
def on_asset_create(sender, instance, created=False, **kwargs):
|
||
|
|
if not created:
|
||
|
|
return
|
||
|
|
# PushAccountManager.trigger_by_asset_create(instance)
|