fix(perms): 修复删除 资产或授权引起的bug

apps/assets/signals_handler/node_assets_amount.py

@@ -7,7 +7,7 @@ from django.db.models.signals import (
     m2m_changed
 )
 
-from orgs.utils import ensure_in_real_or_default_org
+from orgs.utils import ensure_in_real_or_default_org, tmp_to_org
 from common.const.signals import PRE_ADD, POST_REMOVE, PRE_CLEAR
 from common.utils import get_logger
 from assets.models import Asset, Node, compute_parent_key
@@ -34,6 +34,7 @@ def on_node_asset_change(sender, action, instance, reverse, pk_set, **kwargs):
 
     operator = mapper[action]
 
+    with tmp_to_org(instance.org):
         if reverse:
             node: Node = instance
             asset_pk_set = set(pk_set)

apps/perms/signals_handler/refresh_perms.py

@@ -5,7 +5,7 @@ from django.dispatch import receiver
 
 from users.models import User
 from assets.models import Asset
-from orgs.utils import current_org
+from orgs.utils import current_org, tmp_to_org
 from common.utils import get_logger
 from common.exceptions import M2MReverseNotAllowed
 from common.const.signals import POST_ADD, POST_REMOVE, POST_CLEAR
@@ -18,7 +18,8 @@ logger = get_logger(__file__)
 
 @receiver(m2m_changed, sender=User.groups.through)
 def on_user_groups_change(sender, instance, action, reverse, pk_set, **kwargs):
-    if action.startswith('post'):
+    if not action.startswith('post'):
+        return
     if reverse:
         group_ids = [instance.id]
         user_ids = pk_set
@@ -27,7 +28,9 @@ def on_user_groups_change(sender, instance, action, reverse, pk_set, **kwargs):
         user_ids = [instance.id]
 
     exists = AssetPermission.user_groups.through.objects.filter(usergroup_id__in=group_ids).exists()
-        if exists:
+    if not exists:
+        return
+    with tmp_to_org(instance.org):
         org_ids = [current_org.id]
         UserGrantedTreeRefreshController.add_need_refresh_orgs_for_users(org_ids, user_ids)
 
@@ -35,6 +38,7 @@ def on_user_groups_change(sender, instance, action, reverse, pk_set, **kwargs):
 @receiver([pre_delete], sender=AssetPermission)
 def on_asset_perm_pre_delete(sender, instance, **kwargs):
     # 授权删除之前，查出所有相关用户
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_by_asset_perm_ids([instance.id])
 
 
@@ -44,6 +48,7 @@ def on_asset_perm_pre_save(sender, instance, **kwargs):
         old = AssetPermission.objects.get(id=instance.id)
 
         if old.is_valid != instance.is_valid:
+            with tmp_to_org(instance.org):
                 UserGrantedTreeRefreshController.add_need_refresh_by_asset_perm_ids([instance.id])
     except AssetPermission.DoesNotExist:
         pass
@@ -51,7 +56,9 @@ def on_asset_perm_pre_save(sender, instance, **kwargs):
 
 @receiver([post_save], sender=AssetPermission)
 def on_asset_perm_post_save(sender, instance, created, **kwargs):
-    if created:
+    if not created:
+        return
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_by_asset_perm_ids([instance.id])
 
 
@@ -64,7 +71,10 @@ def on_permission_nodes_changed(sender, instance, action, reverse, **kwargs):
     if reverse:
         raise M2MReverseNotAllowed
 
-    if need_rebuild_mapping_node(action):
+    if not need_rebuild_mapping_node(action):
+        return
+
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_by_asset_perm_ids([instance.id])
 
 
@@ -73,28 +83,38 @@ def on_permission_assets_changed(sender, instance, action, reverse, pk_set, mode
     if reverse:
         raise M2MReverseNotAllowed
 
-    if need_rebuild_mapping_node(action):
+    if not need_rebuild_mapping_node(action):
+        return
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_by_asset_perm_ids([instance.id])
 
 
 @receiver(m2m_changed, sender=AssetPermission.users.through)
-def on_asset_permission_users_changed(sender, action, reverse, pk_set, **kwargs):
+def on_asset_permission_users_changed(sender, action, reverse, instance, pk_set, **kwargs):
     if reverse:
         raise M2MReverseNotAllowed
 
-    if need_rebuild_mapping_node(action):
+    if not need_rebuild_mapping_node(action):
+        return
+
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_orgs_for_users(
             [current_org.id], pk_set
         )
 
 
 @receiver(m2m_changed, sender=AssetPermission.user_groups.through)
-def on_asset_permission_user_groups_changed(sender, action, pk_set, reverse, **kwargs):
+def on_asset_permission_user_groups_changed(sender, instance, action, pk_set, reverse, **kwargs):
     if reverse:
         raise M2MReverseNotAllowed
 
-    if need_rebuild_mapping_node(action):
+    if not need_rebuild_mapping_node(action):
-        user_ids = User.groups.through.objects.filter(usergroup_id__in=pk_set).distinct().values_list('user_id', flat=True)
+        return
+
+    user_ids = User.groups.through.objects.filter(usergroup_id__in=pk_set) \
+        .values_list('user_id', flat=True) \
+        .distinct()
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_orgs_for_users(
             [current_org.id], user_ids
         )
@@ -112,4 +132,5 @@ def on_node_asset_change(action, instance, reverse, pk_set, **kwargs):
         asset_pk_set = [instance.id]
         node_pk_set = pk_set
 
+    with tmp_to_org(instance.org):
         UserGrantedTreeRefreshController.add_need_refresh_on_nodes_assets_relate_change(node_pk_set, asset_pk_set)