Software Público Brasileiro

# coding=utf-8
import re
from django import forms
from django.utils.translation import ugettext_lazy as _
from rolepermissions.shortcuts import assign_role
from django.contrib.auth import update_session_auth_hash
from django.core.validators import validate_email
from django.core.exceptions import ValidationError
from os.path import join
from resubmit.widgets import ResubmitFileWidget
from PIL import Image
import os
from amadeus import settings

from .models import User

class Validation(forms.ModelForm):
	MIN_PASS_LENGTH = 8
	MAX_UPLOAD_SIZE = 2*1024*1024

	def clean_email(self):
		email = self.cleaned_data.get('email', '')

		try:
			validate_email( email )
			v_email = re.compile('[\w.%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,4}')
			if v_email.fullmatch(email) is None:
				self._errors['email'] = [_('You must insert an email address')]

				return ValueError
			return email
		except ValidationError:
			self._errors['email'] = [_('You must insert an email address')]

			return ValueError

	def clean_image(self):
		image = self.cleaned_data.get('image', False)

		if image:
			if hasattr(image, '_size'):
				if image._size > self.MAX_UPLOAD_SIZE:
					self._errors['image'] = [_("The image is too large. It should have less than 2MB.")]

					return ValueError

		return image

	def clean_new_password(self):
		password = self.cleaned_data.get('new_password')

		if self.is_edit and len(password) == 0:
			return password

		if len(password) == 0:
			self._errors['new_password'] = [_('You must define a password.')]

			return ValueError

		return password

	def clean_password2(self):
		password = self.cleaned_data.get("new_password", None)
		password2 = self.cleaned_data.get("password2", None)

		if self.is_edit and len(password) == 0:
			return password2

		if not password is None and password != ValueError:
			if not password2 is None and password != password2:
				self._errors['password2'] = [_('The confirmation password is incorrect.')]

				return ValueError

		return password2

class RegisterUserForm(Validation):
	new_password = forms.CharField(label=_('Password'), widget = forms.PasswordInput(render_value = True, attrs = {'placeholder': _('Password *')}))
	password2 = forms.CharField(label = _('Confirm Password'), widget = forms.PasswordInput(render_value = True, attrs = {'placeholder': _('Confirm Password *')}))

	is_edit = False

	#Cropping image
	x = forms.FloatField(widget=forms.HiddenInput(),required=False)
	y = forms.FloatField(widget=forms.HiddenInput(),required=False)
	width = forms.FloatField(widget=forms.HiddenInput(),required=False)
	height = forms.FloatField(widget=forms.HiddenInput(),required=False)


	def save(self, commit=True):
		super(RegisterUserForm, self).save(commit=False)
		self.deletepath = ""

		x = self.cleaned_data.get('x')
		y = self.cleaned_data.get('y')
		w = self.cleaned_data.get('width')
		h = self.cleaned_data.get('height')

		if self.instance.image :
			image = Image.open(self.instance.image)
			if not x is None:
				cropped_image = image.crop((x, y, w+x, h+y))
				resized_image = cropped_image.resize((200, 200), Image.ANTIALIAS)

				folder_path = join(settings.MEDIA_ROOT, 'users')
				#check if the folder already exists
				if not os.path.isdir(folder_path):
					os.makedirs(folder_path)

				if ("users" not in self.instance.image.path):
					self.deletepath = self.instance.image.path

				resized_image.save(self.instance.image.path)

		self.instance.set_password(self.cleaned_data['new_password'])

		self.instance.save()
		if (self.deletepath):
			os.remove(self.deletepath)
		return self.instance

	class Meta:
		model = User
		fields = ['email', 'username', 'last_name', 'social_name', 'image', 'show_email', 'x', 'y', 'width', 'height',]
		widgets = {
			'email': forms.EmailInput(attrs = {'placeholder': _('Email *')}),
			'username': forms.TextInput(attrs = {'placeholder': _('Name *')}),
			'last_name': forms.TextInput(attrs = {'placeholder': _('Last Name *')}),
			'social_name': forms.TextInput(attrs = {'placeholder': _('Social Name')}),
			'image': ResubmitFileWidget(attrs={'accept':'image/*'}),
		}

class ProfileForm(Validation):
	is_edit = True
	#Cropping image
	x = forms.FloatField(widget=forms.HiddenInput(),required=False)
	y = forms.FloatField(widget=forms.HiddenInput(),required=False)
	width = forms.FloatField(widget=forms.HiddenInput(),required=False)
	height = forms.FloatField(widget=forms.HiddenInput(),required=False)


	def save(self, commit=True):
		super(ProfileForm, self).save(commit=False)
		self.deletepath = ""
		x = self.cleaned_data.get('x')
		y = self.cleaned_data.get('y')
		w = self.cleaned_data.get('width')
		h = self.cleaned_data.get('height')

		if self.instance.image:
			image = Image.open(self.instance.image)
			if not x is None:
				cropped_image = image.crop((x, y, w+x, h+y))
				resized_image = cropped_image.resize((200, 200), Image.ANTIALIAS)

				folder_path = join(settings.MEDIA_ROOT, 'users')
				#check if the folder already exists
				if not os.path.isdir(folder_path):
					os.makedirs(folder_path)

				if ("users" not in self.instance.image.path):
					self.deletepath = self.instance.image.path

				resized_image.save(self.instance.image.path)

		self.instance.save()
		if (self.deletepath):
			os.remove(self.deletepath)
		return self.instance

	class Meta:
		model = User
		fields = ['email', 'username', 'last_name', 'social_name', 'description', 'show_email', 'image']
		widgets = {
			'email': forms.EmailInput,
			'description': forms.Textarea,
			'username': forms.TextInput(attrs = {'readonly': 'readonly'}),
			'last_name': forms.TextInput(attrs = {'readonly': 'readonly'}),
			'image': ResubmitFileWidget(attrs={'accept':'image/*'}),
		}

class UserForm(Validation):
	is_edit = False

	def __init__(self, *args, **kwargs):
		is_update = kwargs.pop('is_edit', False)

		super(UserForm, self).__init__(*args, **kwargs)

		self.is_edit = is_update

	new_password = forms.CharField(label = _('Password'), widget = forms.PasswordInput(render_value = True), required = False)
	password2 = forms.CharField(label = _('Confirm Password'), widget = forms.PasswordInput(render_value = True), required = False)

	#Cropping image
	x = forms.FloatField(widget=forms.HiddenInput(),required=False)
	y = forms.FloatField(widget=forms.HiddenInput(),required=False)
	width = forms.FloatField(widget=forms.HiddenInput(),required=False)
	height = forms.FloatField(widget=forms.HiddenInput(),required=False)

	def save(self, commit=True):
		super(UserForm, self).save(commit=False)
		self.deletepath = ""

		x = self.cleaned_data.get('x')
		y = self.cleaned_data.get('y')
		w = self.cleaned_data.get('width')
		h = self.cleaned_data.get('height')

		if self.instance.image :
			image = Image.open(self.instance.image)
			if not x is None:
				cropped_image = image.crop((x, y, w+x, h+y))
				resized_image = cropped_image.resize((200, 200), Image.ANTIALIAS)

				folder_path = join(settings.MEDIA_ROOT, 'users')
				#check if the folder already exists
				if not os.path.isdir(folder_path):
					os.makedirs(folder_path)

				if ("users" not in self.instance.image.path):
					self.deletepath = self.instance.image.path

				resized_image.save(self.instance.image.path)


		if not self.is_edit or self.cleaned_data['new_password'] != '':
			self.instance.set_password(self.cleaned_data['new_password'])

		self.instance.save()
		if (self.deletepath):
			os.remove(self.deletepath)
		return self.instance

	class Meta:
		model = User
		fields = ['email', 'username', 'last_name', 'social_name', 'description', 'show_email', 'image', 'is_staff', 'is_active',]
		widgets = {
			'email': forms.EmailInput,
			'description': forms.Textarea,
			'image': ResubmitFileWidget(attrs={'accept':'image/*'}),
		}

class ChangePassForm(Validation):
	def __init__(self, *args, **kwargs):
		self.user = kwargs.pop('user', None)
		self.request = kwargs.pop('request', None)
		super(ChangePassForm, self).__init__(*args, **kwargs)

	is_edit = False

	new_password = forms.CharField(label=_('New Password'), widget = forms.PasswordInput(render_value=True), required = True)
	password2 = forms.CharField(label = _('Confirm Password'), widget = forms.PasswordInput(render_value=True), required = True)

	def clean_password(self):
		password = self.cleaned_data.get('password', None)

		if not self.user.check_password(password):
			self._errors['password'] = [_('The value inputed does not match with your actual password.')]

			return ValueError

		return password

	def save(self, commit=True):
		super(ChangePassForm, self).save(commit=False)

		self.instance.set_password(self.cleaned_data['new_password'])

		update_session_auth_hash(self.request, self.instance)

		self.instance.save()

		return self.instance

	class Meta:
		model = User
		fields = ['password']
		labels = {
			'password': _('Actual Password')
		}
		widgets = {
			'password': forms.PasswordInput
		}

class PassResetRequest(forms.Form):
	email = forms.EmailField(label = _('Email'), max_length = 254, widget = forms.EmailInput(attrs = {'placeholder': _('Email') + ' *'}))

	def clean_email(self):
		email = self.cleaned_data.get('email', '')

		try:
			validate_email( email )
			v_email = re.compile('[\w.%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,4}')
			if v_email.fullmatch(email) is None:
				self._errors['email'] = [_('You must insert an email address')]

				return ValueError
			return email
		except ValidationError:
			self._errors['email'] = [_('You must insert a valid email address')]

			return ValueError

class SetPasswordForm(Validation):
	is_edit = False

	new_password = forms.CharField(label=_('New Password'), widget = forms.PasswordInput(render_value=True), required = True)
	password2 = forms.CharField(label = _('Confirm Password'), widget = forms.PasswordInput(render_value=True), required = True)

	class Meta:
		model = User
		fields = []