github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

feat: 授权规则创建

pull/8931/head
Jiangjie.Bai 2022-09-29 20:41:40 +08:00
parent 76747642c4
commit 14f48c459c
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/perms/serializers/permission.py
View File

@ -87,6 +87,18 @@ class AssetPermissionSerializer(BulkOrgResourceModelSerializer):
'is_valid': {'label': _('Is valid')}, 'is_valid': {'label': _('Is valid')},
} }
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.set_actions_field()
def set_actions_field(self):
actions = self.fields.get('actions')
if not actions:
return
choices = actions._choices
actions._choices = choices
actions.default = list(choices.keys())
@classmethod @classmethod
def setup_eager_loading(cls, queryset): def setup_eager_loading(cls, queryset):
""" Perform necessary eager loading of data. """ """ Perform necessary eager loading of data. """
@ -110,7 +122,7 @@ class AssetPermissionSerializer(BulkOrgResourceModelSerializer):
instance.user_groups.add(*user_groups_to_set) instance.user_groups.add(*user_groups_to_set)
# 资产 # 资产
assets_to_set = Asset.objects.filter( assets_to_set = Asset.objects.filter(
Q(ip__in=kwargs.get('assets_display')) | Q(address__in=kwargs.get('assets_display')) |
Q(name__in=kwargs.get('assets_display')) Q(name__in=kwargs.get('assets_display'))
).distinct() ).distinct()
instance.assets.add(*assets_to_set) instance.assets.add(*assets_to_set)