mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Implemented ConnectionStatusCSVField

Browse Source
This commit is contained in:
Jeremy Stretch
2017-06-05 16:12:51 -04:00
parent 08c8bd3049
commit ec9d0d4008
1 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
netbox/dcim/forms.py
View File

@ -50,12 +50,25 @@ def get_device_by_name_or_pk(name):
return device return device
def validate_connection_status(value): class ConnectionStatusCSVField(forms.CharField):
""" """
Custom validator for connection statuses. value must be either "planned" or "connected" (case-insensitive). This field accepts either "planned" or "connected" as a connection status for CSV imports.
""" """
if value.lower() not in ['planned', 'connected']: def __init__(self, *args, **kwargs):
raise ValidationError('Invalid connection status ({}); must be either "planned" or "connected".'.format(value)) super(ConnectionStatusCSVField, self).__init__(*args, **kwargs)
if not self.help_text:
self.help_text = 'Connection status ("planned" or "connected")'
def clean(self, value):
value = super(ConnectionStatusCSVField, self).clean(value)
try:
return {
'planned': CONNECTION_STATUS_PLANNED,
'connected': CONNECTION_STATUS_CONNECTED,
}[value.lower()]
except KeyError:
raise ValidationError(
'Invalid connection status ({}); must be either "planned" or "connected".'.format(value))
class DeviceComponentForm(BootstrapMixin, forms.Form): class DeviceComponentForm(BootstrapMixin, forms.Form):
@ -968,11 +981,11 @@ class ConsoleConnectionCSVForm(forms.ModelForm):
console_port = forms.CharField( console_port = forms.CharField(
help_text='Console port name' help_text='Console port name'
) )
status = forms.CharField(validators=[validate_connection_status]) connection_status = ConnectionStatusCSVField()
class Meta: class Meta:
model = ConsolePort model = ConsolePort
fields = ['console_server', 'cs_port', 'device', 'console_port'] fields = ['console_server', 'cs_port', 'device', 'console_port', 'connection_status']
def clean_console_port(self): def clean_console_port(self):
@ -1222,11 +1235,11 @@ class PowerConnectionCSVForm(forms.ModelForm):
power_port = forms.CharField( power_port = forms.CharField(
help_text='Power port name' help_text='Power port name'
) )
status = forms.CharField(validators=[validate_connection_status]) connection_status = ConnectionStatusCSVField()
class Meta: class Meta:
model = PowerPort model = PowerPort
fields = ['pdu', 'power_outlet', 'device', 'power_port'] fields = ['pdu', 'power_outlet', 'device', 'power_port', 'connection_status']
def clean_power_port(self): def clean_power_port(self):
@ -1622,10 +1635,7 @@ class InterfaceConnectionCSVForm(forms.ModelForm):
interface_b = forms.CharField( interface_b = forms.CharField(
help_text='Interface name' help_text='Interface name'
) )
connection_status = forms.CharField( connection_status = ConnectionStatusCSVField()
help_text='Connection status',
validators=[validate_connection_status]
)
class Meta: class Meta:
model = InterfaceConnection model = InterfaceConnection