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

Tweak validation for custom date fields

Browse Source
This commit is contained in:
Jeremy Stretch
2020-12-15 16:00:18 -05:00
parent 34643f536e
commit d64fc261e2
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/extras/models/customfields.py
View File

@ -1,6 +1,6 @@
import re import re
from collections import OrderedDict from collections import OrderedDict
from datetime import datetime from datetime import datetime, date
from django import forms from django import forms
from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.models import ContentType
@ -317,10 +317,11 @@ class CustomField(models.Model):
# Validate date # Validate date
if self.type == CustomFieldTypeChoices.TYPE_DATE: if self.type == CustomFieldTypeChoices.TYPE_DATE:
try: if type(value) is not date:
datetime.strptime(value, '%Y-%m-%d') try:
except ValueError: datetime.strptime(value, '%Y-%m-%d')
raise ValidationError("Date values must be in the format YYYY-MM-DD.") except ValueError:
raise ValidationError("Date values must be in the format YYYY-MM-DD.")
# Validate selected choice # Validate selected choice
if self.type == CustomFieldTypeChoices.TYPE_SELECT: if self.type == CustomFieldTypeChoices.TYPE_SELECT: