mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-16 07:28:59 +00:00
f3dc9b886b
* [Update] 拆分filter org * [Update] 修改session支持protocol搜索 * [Bugfix] 修复判断问题 * [Update] 支持收藏资产 * [update] 修改org resource queryset * [Update] 修改form serializer 对应的多对多字段 * [Bugfix] 修复其他组织取消收藏的bug * [Update] 去掉debug信息 * [Update] 修改remote app get queryset * [Update] 修改remote app get queryset * [Update] 修改没有授权时显示情况 * [Bugfix] 修复组织管理员查看用户权限失败问题 * [Update] 优化forms assets queryset设置
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django import forms
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from ..models import Label, Asset
|
|
|
|
__all__ = ['LabelForm']
|
|
|
|
|
|
class LabelForm(forms.ModelForm):
|
|
assets = forms.ModelMultipleChoiceField(
|
|
queryset=Asset.objects.none(), label=_('Asset'), required=False,
|
|
widget=forms.SelectMultiple(
|
|
attrs={'class': 'select2', 'data-placeholder': _('Select assets')}
|
|
)
|
|
)
|
|
|
|
class Meta:
|
|
model = Label
|
|
fields = ['name', 'value', 'assets']
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.set_fields_queryset()
|
|
|
|
def set_fields_queryset(self):
|
|
assets_field = self.fields.get('assets')
|
|
|
|
# 没有data代表是渲染表单, 有data代表是提交创建/更新表单
|
|
if not self.data:
|
|
# 有instance 代表渲染更新表单, 否则是创建表单
|
|
# 前端渲染优化, 防止过多资产, 设置assets queryset为none
|
|
if self.instance:
|
|
assets_field.initial = self.instance.assets.all()
|
|
assets_field.queryset = self.instance.assets.all()
|
|
else:
|
|
assets_field.queryset = Asset.objects.none()
|
|
else:
|
|
assets_field.queryset = Asset.objects.all()
|
|
|
|
def save(self, commit=True):
|
|
label = super().save(commit=commit)
|
|
assets = self.cleaned_data['assets']
|
|
label.assets.set(assets)
|
|
return label
|