17 lines
537 B
Python
17 lines
537 B
Python
from countries.models import Country
|
|
from django.core.exceptions import ValidationError
|
|
from django.core.validators import BaseValidator
|
|
from django.utils.deconstruct import deconstructible
|
|
|
|
|
|
@deconstructible
|
|
class CountryCodeValidator(BaseValidator):
|
|
def __init__(self, message=None):
|
|
self.message = message or "There is no such country"
|
|
|
|
def __call__(self, value):
|
|
try:
|
|
Country.objects.get(alpha2=value)
|
|
except Country.DoesNotExist:
|
|
raise ValidationError(self.message) from None
|