Converted device fields to use FlexibleModelChoiceField; misc cleanup

This commit is contained in:
Jeremy Stretch
2017-06-07 15:51:11 -04:00
parent 08315fdab0
commit 2512e01df4
3 changed files with 36 additions and 24 deletions

View File

@@ -7,7 +7,7 @@ from django import forms
from django.db.models import Count
from dcim.models import Device
from utilities.forms import BootstrapMixin, BulkEditForm, CSVDataField, FilterChoiceField, SlugField
from utilities.forms import BootstrapMixin, BulkEditForm, FilterChoiceField, FlexibleModelChoiceField, SlugField
from .models import Secret, SecretRole, UserKey
@@ -66,10 +66,10 @@ class SecretForm(BootstrapMixin, forms.ModelForm):
class SecretCSVForm(forms.ModelForm):
device = forms.ModelChoiceField(
device = FlexibleModelChoiceField(
queryset=Device.objects.all(),
to_field_name='name',
help_text='Device name',
help_text='Device name or ID',
error_messages={
'invalid_choice': 'Device not found.',
}
@@ -89,6 +89,9 @@ class SecretCSVForm(forms.ModelForm):
class Meta:
model = Secret
fields = ['device', 'role', 'name', 'plaintext']
help_texts = {
'name': 'Name or username',
}
def save(self, *args, **kwargs):
s = super(SecretCSVForm, self).save(*args, **kwargs)