Source code for localflavor.sk.forms

"""Slovak-specific form helpers."""

from django.forms.fields import RegexField, Select
from django.utils.translation import gettext_lazy as _

from .sk_districts import DISTRICT_CHOICES
from .sk_regions import REGION_CHOICES


class SKRegionSelect(Select):
    """A select widget widget with list of Slovak regions as choices."""

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


class SKDistrictSelect(Select):
    """A select widget with list of Slovak districts as choices."""

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


class SKPostalCodeField(RegexField):
    """
    A form field that validates its input as Slovak postal code.

    Valid form is XXXXX or XXX XX, where X represents integer.
    """

    default_error_messages = {
        'invalid': _('Enter a postal code in the format XXXXX or XXX XX.'),
    }

    def __init__(self, **kwargs):
        super().__init__(r'^\d{5}$|^\d{3} \d{2}$', **kwargs)

    def clean(self, value):
        """
        Validates the input and returns a string that contains only numbers.

        Returns an empty string for empty values.
        """
        value = super().clean(value)
        if value in self.empty_values:
            return value
        return value.replace(' ', '')