okcashpro/okshop

from django import forms
from django.contrib.auth.models import User
from django.core.validators import validate_email, validate_slug, MinLengthValidator
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Submit, Layout, Div

class RegisterForm(forms.Form):
    username = forms.CharField(label='Username', max_length=150,
                               validators=[validate_slug], required=True)
    email = forms.CharField(label='Email', max_length=1024,
                            validators=[validate_email], required=True)
    password = forms.CharField(widget=forms.PasswordInput,
                               validators=[MinLengthValidator(8)],
                               label='Password', required=True)
    passwordconfirm = forms.CharField(widget=forms.PasswordInput,
                                      label='Confirm password', required=True)

    def clean(self):
        cleaned_data = super().clean()
        if cleaned_data.get('password') != cleaned_data.get('passwordconfirm'):
            raise forms.ValidationError("Passwords don't match!")
        if User.objects.filter(username=cleaned_data.get('username')).count() > 0:
            raise forms.ValidationError("Username already in use!")
        if User.objects.filter(email=cleaned_data.get('email')).count() > 0:
            raise forms.ValidationError("Email already in use!")

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.helper = FormHelper()
        self.helper.add_input(Submit('submit', 'Register'))
        self.helper.form_class = 'form-horizontal'
        self.helper.label_class = 'col-lg-2'
        self.helper.field_class = 'col-lg-8'


class ReviewForm(forms.Form):
    title = forms.CharField(label='Title', max_length=150, required=False)
    rating = forms.FloatField(label='Rating',
                              widget=forms.NumberInput(attrs={'min': 1, 'max': 5, 'step': 1}),
                              required=True)
    review = forms.CharField(label='Review',
                             widget=forms.Textarea(attrs={'class': 'vresize'}),
                             required=True)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.helper = FormHelper()
        self.helper.form_class = 'modal-body'
        self.helper.layout = Layout(
            Div(
                'title',
                'rating',
                'review',
                css_class='modal-body'
            ),
            Div(
                Submit('submit', 'Post Review'),
                css_class='modal-footer'
            )
        )


    def clean_rating(self):
        data = self.cleaned_data
        if not 1 <= data['rating'] <= 5:
            raise forms.ValidationError("Rating should be between 1 and 5")
        return int(data['rating'])


class KeyForm(forms.Form):
    title = forms.CharField(max_length=200)
    description = forms.CharField(widget=forms.Textarea(attrs={'class': 'vresize'}))
    keys = forms.CharField(required=False, label="Keys to add",
                           widget=forms.Textarea(attrs={'class': 'vresize'}),
                           help_text='One key per line. '
                                     'You won\'t be able to remove these')
    is_link = forms.BooleanField(label="Are these keys links?",
                                 help_text='Setting this will make it so that'
                                           ' the keys are clickable',
                                 required=False)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.helper = FormHelper()
        self.helper.layout = Layout(
            Div(
                'title',
                'description',
                'keys',
                'is_link',
                css_class='panel-body'
            ),
            Div(
                Submit('submit', 'Add keys'),
                css_class='panel-footer'
            )
        )