Source code for localflavor.dk.forms

"""Denmark specific Form helpers."""

from django.core.exceptions import ValidationError
from django.forms import fields, widgets
from django.utils.translation import gettext_lazy as _

from .dk_municipalities import DK_MUNICIPALITIES
from .dk_postalcodes import DK_POSTALCODES


def postal_code_validator(value):
    if value not in [entry[0] for entry in DK_POSTALCODES]:
        raise ValidationError(_('Enter a postal code in the format XXXX.'), code='invalid')


class DKPostalCodeField(fields.CharField):
    """An Input widget that uses a list of Danish postal codes as valid input."""

    default_validators = [postal_code_validator]


class DKMunicipalitySelect(widgets.Select):
    """A Select widget that uses a list of Danish municipalities (kommuner) as its choices."""

    def __init__(self, attrs=None):
        super().__init__(attrs, choices=DK_MUNICIPALITIES)