summaryrefslogtreecommitdiff
path: root/payments/widgets.py
blob: 795906feeb6e1ee56670258141c2513b412544ac (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
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'