""" Checkout forms """
from django import forms
from store_order.models import Order
import re


def strip_non_numbers(data):
    """ gets rid of all non-number characters """
    non_numbers = re.compile(r'\D')
    return non_numbers.sub('', data)


# Currently not used, can remove if we don't need in future
class CheckoutForm(forms.ModelForm):
    """ checkout form class to collect user billing and
    shipping information for placing an order """

    def __init__(self, *args, **kwargs):
        super(CheckoutForm, self).__init__(*args, **kwargs)
        # override default attributes
        for field in self.fields:
            self.fields[field].widget.attrs['size'] = '30'

        self.fields['shipping_state'].widget.attrs['size'] = '3'

        self.fields['billing_state'].widget.attrs['size'] = '3'
        self.fields['billing_zip'].widget.attrs['size'] = '6'

    class Meta:
        model = Order
        exclude = ('status', 'ip_address', 'user', 'transaction_id')

    def clean_phone(self):
        """ Validate the phone field """
        phone = self.cleaned_data['phone']
        stripped_phone = strip_non_numbers(phone)
        if len(stripped_phone) < 10:
            error_str = 'Enter a valid phone number with area code.'\
                '(e.g. 555-555-5555)'
            raise forms.ValidationError(error_str)
        return self.cleaned_data['phone']