Source code for localflavor.py_.forms

"""
PY-specific Form helpers.
"""

from django.forms.fields import Select

from .py_department import DEPARTMENT_CHOICES, DEPARTMENT_ROMAN_CHOICES


class PyDepartmentSelect(Select):
    """
    A Select widget with a list of Paraguayan departments as choices.
    """
    def __init__(self, attrs=None):
        super(PyDepartmentSelect, self).__init__(attrs, choices=DEPARTMENT_CHOICES)


class PyNumberedDepartmentSelect(Select):
    """
    A Select widget with a roman numbered list of Paraguayan departments as choices.
    """
    def __init__(self, attrs=None):
        super(PyNumberedDepartmentSelect, self).__init__(attrs, choices=DEPARTMENT_ROMAN_CHOICES)