# -*- coding: utf-8 -*- import phonenumbers from django import forms from django.utils.safestring import mark_safe from django.utils.translation import ugettext_lazy as _ from servo.forms.base import BaseModelForm, DatepickerInput from servo.models import Customer class CustomerForm(BaseModelForm): class Meta: model = Customer widgets = { 'groups': forms.CheckboxSelectMultiple } exclude = [] def clean_name(self): name = self.cleaned_data.get('name') return name.strip() def clean(self): cd = super(CustomerForm, self).clean() phone = cd.get('phone') country = cd.get('country') if len(phone) < 1: return cd try: phonenumbers.parse(phone, country) except phonenumbers.NumberParseException: msg = _('Enter a valid phone number') self._errors["phone"] = self.error_class([msg]) return cd class CustomerSearchForm(forms.Form): name__icontains = forms.CharField( required=False, label=_('Name contains') ) email__icontains = forms.CharField( required=False, label=_('Email contains') ) street_address__icontains = forms.CharField( required=False, label=_('Address contains') ) checked_in_start = forms.DateField( required=False, label=_('Checked in between'), widget=DatepickerInput(attrs={'class': "input-small"}) ) checked_in_end = forms.DateField( required=False, label=mark_safe(' '), widget=DatepickerInput(attrs={'class': "input-small"}) ) def clean(self): cd = super(CustomerSearchForm, self).clean() for k, v in cd.items(): if v not in ['', None]: return cd raise forms.ValidationError(_('Please specify at least one parameter')) class CustomerUploadForm(forms.Form): datafile = forms.FileField(label=_('CSV file')) skip_dups = forms.BooleanField( initial=False, required=False, label=_('Skip duplicates'), help_text=_('Skip customers with existing email addresses') ) def clean_datafile(self): d = self.cleaned_data.get('datafile') if not d.content_type.startswith('text'): raise forms.ValidationError(_('Data file should be in text format')) return d