'''

@author: olivier.massot, 2018
'''
from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User
from martor.fields import MartorFormField

from main.models import Story, Epic


class RegisterForm(UserCreationForm):
                            
    class Meta:
        model = User
        fields = ('username', 'first_name', 'last_name', 'email')
        
    def save(self, commit=True):
        user = super(RegisterForm, self).save(commit=False)
        user.first_name = self.cleaned_data["first_name"]
        user.last_name = self.cleaned_data["last_name"]
        user.email = self.cleaned_data["email"]
        if commit:
            user.save()

class ProfileForm(forms.ModelForm):
    class Meta:
        model = User
        fields = ('username', 'first_name', 'last_name', 'email')

class EpicForm(forms.ModelForm):
    
    description = MartorFormField(label="Description")
 
    class Meta:
        model = Epic
        fields = ('project', 'name', 'size', 'value', 'description')
        
class StoryForm(forms.ModelForm):
    
    description = MartorFormField(label="Description")
 
    class Meta:
        model = Story
        widgets = {'epic': forms.HiddenInput()}
        fields = ('epic', 'name', 'description', 'assignees', 'sprints')
        
    def __init__(self, *args, **kwargs):
        super(StoryForm, self).__init__(*args, **kwargs)
        self.fields['assignees'].required = False
        self.fields['sprints'].required = False