diff --git a/apps/assets/forms.py b/apps/assets/forms.py index b8de07e73..1d8990ef2 100644 --- a/apps/assets/forms.py +++ b/apps/assets/forms.py @@ -23,8 +23,6 @@ from django.utils.translation import gettext_lazy as _ # class AssetCreateForm(forms.ModelForm): - tags = forms.CharField(label=_('Tags'), widget=forms.TextInput(attrs={'id': 'tags'}), - required=False, help_text='Use `,` split') def __init__(self, *args, **kwargs): instance = kwargs.get('instance', None) @@ -32,19 +30,15 @@ class AssetCreateForm(forms.ModelForm): if instance: initial = kwargs.get('initial', {}) tags = instance.tags.all() - initial['tags'] = ",".join([tag.value for tag in tags]) + initial['tags'] = [t.pk for t in kwargs['instance'].tags.all()] print(kwargs.get('initial')) super(AssetCreateForm, self).__init__(*args, **kwargs) def _save_m2m(self): + super(AssetCreateForm, self)._save_m2m() tags = self.cleaned_data['tags'] - - if tags: - value_list = tags.split(',') - self.instance.tags.all().delete() - Tag.objects.bulk_create( - [Tag(value=value, asset=self.instance) for value in value_list] - ) + self.instance.tags.clear() + self.instance.tags.add(*tuple(tags)) class Meta: model = Asset @@ -52,11 +46,14 @@ class AssetCreateForm(forms.ModelForm): fields = [ 'hostname', 'ip', 'port', 'type', 'comment', 'admin_user', 'system_users', 'idc', 'groups', 'other_ip', 'remote_card_ip', 'mac_address', 'brand', 'cpu', 'memory', 'disk', 'os', 'cabinet_no', - 'cabinet_pos', 'number', 'status', 'env', 'sn', + 'cabinet_pos', 'number', 'status', 'env', 'sn','tags', ] + tags = forms.ModelMultipleChoiceField(queryset=Tag.objects.all()) widgets = { 'groups': forms.SelectMultiple(attrs={'class': 'select2', 'data-placeholder': _('Select asset groups')}), + 'tags': forms.SelectMultiple(attrs={'class': 'select2', + 'data-placeholder': _('Select asset groups')}), 'system_users': forms.SelectMultiple(attrs={'class': 'select2', 'data-placeholder': _('Select asset system users')}), 'admin_user': forms.Select(attrs={'class': 'select2', 'data-placeholder': _('Select asset admin user')}), @@ -64,6 +61,7 @@ class AssetCreateForm(forms.ModelForm): help_texts = { 'hostname': '* required', 'ip': '* required', + 'tags': '最多5个标签,单个标签最长8个汉字,按回车确认' } diff --git a/apps/assets/models.py b/apps/assets/models.py index 4726eb3ff..84b934184 100644 --- a/apps/assets/models.py +++ b/apps/assets/models.py @@ -323,6 +323,7 @@ class Asset(models.Model): is_active = models.BooleanField(default=True, verbose_name=_('Is active')) date_created = models.DateTimeField(auto_now=True, null=True, blank=True, verbose_name=_('Date added')) comment = models.TextField(max_length=128, null=True, blank=True, verbose_name=_('Comment')) + tags = models.ManyToManyField('Tag', verbose_name='标签集合', blank=True) def __unicode__(self): return '%(ip)s:%(port)s' % {'ip': self.ip, 'port': self.port} @@ -364,15 +365,17 @@ class Asset(models.Model): class Tag(models.Model): - value = models.CharField(max_length=64, verbose_name=_('VALUE')) - asset = models.ForeignKey(Asset, related_name='tags', on_delete=models.CASCADE, verbose_name=_('Asset')) + name = models.CharField('标签名', max_length=64) + created_time = models.DateTimeField('创建时间', auto_now_add=True) + created_by = models.CharField(max_length=32, null=True, blank=True, verbose_name=_('Created by')) + def __str__(self): + return self.name def __unicode__(self): - return self.value + return self.name class Meta: db_table = 'tag' - unique_together = ('value', 'asset') def init_all_models(): diff --git a/apps/assets/templates/assets/asset_create.html b/apps/assets/templates/assets/asset_create.html index 4df3377c5..3bfb0453f 100644 --- a/apps/assets/templates/assets/asset_create.html +++ b/apps/assets/templates/assets/asset_create.html @@ -25,6 +25,8 @@