import re

from django.forms.widgets import TextInput, MultiWidget, Select


class CreditCardNumberWidget(TextInput):

    def render(self, name, value, attrs=None):
        if value:
            value = re.sub('[\s-]', '', value)
            if len(value) == 16:
                value = ' '.join([value[0:4], value[4:8],
                                  value[8:12], value[12:16]])
            elif len(value) == 15:
                value = ' '.join([value[0:4], value[4:10], value[10:15]])
            elif len(value) == 14:
                value = ' '.join([value[0:4], value[4:10], value[10:14]])
        return super(CreditCardNumberWidget, self).render(name, value, attrs)


# Credit Card Expiry Fields from:
# http://www.djangosnippets.org/snippets/907/
class CreditCardExpiryWidget(MultiWidget):
    """MultiWidget for representing credit card expiry date."""

    template_name = 'payments/credit_card_expiry_widget.html'

    def decompress(self, value):
        if value:
            return [value.month, value.year]
        else:
            return [None, None]


class SensitiveTextInput(TextInput):
    template_name = 'payments/sensitive_text_input.html'


class SensitiveSelect(Select):
    template_name = 'payments/sensitive_select.html'