dnmgmt/whoisdb/validators.py

41 lines
1.1 KiB
Python
Raw Normal View History

2017-02-15 02:35:46 +01:00
from django.core import validators
2017-03-21 02:36:07 +01:00
from django.core.exceptions import ValidationError
2017-02-15 02:35:46 +01:00
from django.utils import six
from django.utils.deconstruct import deconstructible
from django.utils.translation import ugettext_lazy as _
2017-02-23 22:28:06 +01:00
import re
2017-03-21 02:36:07 +01:00
import ipaddress
2017-02-23 22:28:06 +01:00
2017-02-24 03:43:59 +01:00
2017-02-15 02:35:46 +01:00
@deconstructible
class HandleValidator(validators.RegexValidator):
2017-03-22 04:06:24 +01:00
regex = r'^(?:[A-Z]+[0-9]*(-[A-Z]+)|AUTO)'
2017-02-24 03:43:59 +01:00
message = _(
'Enter a valid handle (all uppercase)'
)
flags = re.ASCII if six.PY3 else 0
@deconstructible
class HandleValidatorWithSuffix(validators.RegexValidator):
flags = re.ASCII if six.PY3 else 0
def __init__(self, suffix):
2017-03-22 04:06:24 +01:00
self.regex = r'^(?:[A-Z]+[0-9]*-%s|AUTO)' % re.escape(suffix)
2017-02-24 03:43:59 +01:00
self.message = _(
2017-03-22 04:06:24 +01:00
'Enter a valid handle with suffix %s (all uppercase), e.g. FOO3-%s' % (suffix, suffix)
2017-02-24 03:43:59 +01:00
)
2017-02-15 02:35:46 +01:00
2017-02-24 03:43:59 +01:00
super(HandleValidatorWithSuffix, self).__init__()
2017-03-21 02:36:07 +01:00
2017-03-22 04:06:24 +01:00
2017-03-21 02:36:07 +01:00
def IP46CIDRValidator(value):
2017-03-22 04:06:24 +01:00
if not re.match(r"[0-9a-fA-F:.]+/[0-9]+", value):
2017-03-21 02:36:07 +01:00
raise ValidationError("Address needs to be a subnet in the format of ip/cidr")
try:
ipaddress.ip_network(value)
except ValueError as e:
raise ValidationError(str(e))