Commit 4d782f00b0354d02795b453b582d43fd8e003bcc
1 parent
baf2c430
Exists in
master
and in
5 other branches
import date to validate date register and update profile #7
Showing
1 changed file
with
26 additions
and
25 deletions
Show diff stats
users/forms.py
| 1 | 1 | # coding=utf-8 |
| 2 | -import os | |
| 2 | +import os, re | |
| 3 | +from datetime import date | |
| 4 | +from pycpfcnpj import cpfcnpj | |
| 3 | 5 | from django.conf import settings |
| 4 | 6 | from django import forms |
| 5 | 7 | from django.utils.translation import ugettext_lazy as _ |
| ... | ... | @@ -8,40 +10,39 @@ from django.contrib.auth.forms import UserCreationForm |
| 8 | 10 | from core.forms import RegisterUserForm |
| 9 | 11 | from .models import User |
| 10 | 12 | |
| 11 | - | |
| 12 | -class ProfileForm(forms.ModelForm): | |
| 13 | - | |
| 14 | - def save(self, commit=True): | |
| 15 | - super(ProfileForm, self).save(commit=False) | |
| 16 | - | |
| 17 | - self.instance.set_password(self.cleaned_data['password']) | |
| 18 | - self.instance.save() | |
| 19 | - | |
| 20 | - return self.instance | |
| 21 | - | |
| 22 | - class Meta: | |
| 23 | - model = User | |
| 24 | - fields = ['username', 'name', 'email', 'password', 'birth_date', 'city', 'state', 'gender', 'cpf', 'phone', 'image'] | |
| 25 | - widgets = { | |
| 26 | - 'password':forms.PasswordInput | |
| 27 | - } | |
| 28 | - | |
| 29 | 13 | class UserForm(RegisterUserForm): |
| 30 | 14 | |
| 31 | 15 | class Meta: |
| 32 | 16 | model = User |
| 33 | 17 | fields = ['username', 'name', 'email', 'birth_date', 'city', 'state', 'gender', 'type_profile', 'cpf', 'phone', 'image', 'is_staff', 'is_active'] |
| 34 | 18 | |
| 35 | -class EditUserForm(forms.ModelForm): | |
| 19 | +class UpdateUserForm(forms.ModelForm): | |
| 20 | + | |
| 21 | + def validate_cpf(self, cpf): | |
| 22 | + cpf = ''.join(re.findall('\d', str(cpf))) | |
| 23 | + | |
| 24 | + if cpfcnpj.validate(cpf): | |
| 25 | + return True | |
| 26 | + return False | |
| 27 | + | |
| 28 | + def clean_cpf(self): | |
| 29 | + cpf = self.cleaned_data['cpf'] | |
| 30 | + if not self.validate_cpf(cpf): | |
| 31 | + raise forms.ValidationError(_('Please enter a valid CPF')) | |
| 32 | + return cpf | |
| 33 | + | |
| 34 | + def clean_birth_date(self): | |
| 35 | + birth_date = self.cleaned_data['birth_date'] | |
| 36 | + if birth_date >= date.today(): | |
| 37 | + raise forms.ValidationError(_('Please enter a valid date')) | |
| 38 | + return birth_date | |
| 36 | 39 | |
| 37 | 40 | class Meta: |
| 38 | 41 | model = User |
| 39 | - fields = ['username', 'name', 'email', 'birth_date', 'city', 'state', 'gender', 'cpf', 'phone', 'image'] | |
| 42 | + fields = ['username', 'name', 'email', 'city', 'state', 'birth_date', 'gender', 'type_profile', 'cpf', 'phone', 'image', 'is_staff', 'is_active'] | |
| 40 | 43 | |
| 41 | -# Ailson | |
| 42 | -class UpdateUserForm(forms.ModelForm): | |
| 43 | - company_logo = forms.ImageField(label=_('Company Logo'),required=False, error_messages = {'invalid':_("Image files only")}) | |
| 44 | +class UpdateProfileForm(UpdateUserForm): | |
| 44 | 45 | |
| 45 | 46 | class Meta: |
| 46 | 47 | model = User |
| 47 | - fields = ['username', 'name', 'email', 'city', 'state', 'birth_date', 'gender', 'cpf', 'phone', 'image'] | |
| 48 | 48 | \ No newline at end of file |
| 49 | + fields = ['username', 'name', 'email', 'birth_date', 'city', 'state', 'gender', 'cpf', 'phone', 'image'] | ... | ... |