Custom Widget using MultiWidget in Django

class AddressWidget(forms.MultiWidget):
    """
    Custom widget for Custom Field AddressField
    """
    def __init__(self, country_choices, city_choices, locality_choices, attrs=None):
        widgets = (forms.TextInput(), forms.TextInput(), forms.Select(attrs=attrs, choices=country_choices), forms.Select(attrs=attrs, choices=city_choices),forms.Select(attrs=attrs, choices=locality_choices), forms.TextInput())
        super(AddressWidget, self).__init__(widgets, attrs)

    def decompress(self, value):
        if value:
            return value
        return [None,None,None,None,None,None]