From 9d4121c3b72cd9d509463850e1b00e1ebf28c693 Mon Sep 17 00:00:00 2001 From: ibuler Date: Thu, 15 Jul 2021 10:51:35 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/assets/signals_handler/system_user.py | 23 +++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/apps/assets/signals_handler/system_user.py b/apps/assets/signals_handler/system_user.py index 49892828d..9e1ee2045 100644 --- a/apps/assets/signals_handler/system_user.py +++ b/apps/assets/signals_handler/system_user.py @@ -42,6 +42,7 @@ def on_system_user_assets_change(instance, action, model, pk_set, **kwargs): org_id = instance.org_id + # 关联创建的 authbook 没有系统用户id with tmp_to_root_org(): authbooks = AuthBook.objects.filter( asset_id__in=asset_ids, @@ -50,19 +51,19 @@ def on_system_user_assets_change(instance, action, model, pk_set, **kwargs): if action == POST_ADD: authbooks.update(org_id=org_id) - save_action_mapper = { - 'pre_add': pre_save, - 'post_add': post_save, - 'pre_remove': pre_delete, - 'post_remove': post_delete - } + save_action_mapper = { + 'pre_add': pre_save, + 'post_add': post_save, + 'pre_remove': pre_delete, + 'post_remove': post_delete + } - for ab in authbooks: - ab.org_id = org_id + for ab in authbooks: + ab.org_id = org_id - post_action = save_action_mapper[action] - logger.debug('Send AuthBook post save signal: {} -> {}'.format(action, ab.id)) - post_action.send(sender=AuthBook, instance=ab, created=True) + save_action = save_action_mapper[action] + logger.debug('Send AuthBook post save signal: {} -> {}'.format(action, ab.id)) + save_action.send(sender=AuthBook, instance=ab, created=True) if action == POST_ADD: for system_user_id in system_user_ids: