# -*- coding: utf-8 -*-
'''
Created on Aug 4, 2013

@author: ymh
'''
from django.conf import settings
from django.contrib.auth import get_user_model
from django.contrib.auth.forms import (UserChangeForm as AuthUserChangeForm, 
    UserCreationForm as AuthUserCreationForm)
from django.core.exceptions import ValidationError
from django.forms import Form
from django.forms.fields import ChoiceField, CharField
from django.utils.translation import ugettext as _
from .models import Record


User = get_user_model()

class UserCreationform(AuthUserCreationForm):
    class Meta:
        model = User
        
    def clean_username(self):
        # Since User.username is unique, this check is redundant,
        # but it sets a nicer error message than the ORM. See #13147.
        username = self.cleaned_data["username"]
        try:
            User.objects.get(username=username)
        except User.DoesNotExist:
            return username
        raise ValidationError(self.error_messages['duplicate_username'])    
    

class UserChangeForm(AuthUserChangeForm):
    language = ChoiceField(label=_("language"), choices=[(k,_(v)) for k,v in settings.LANGUAGES], initial=settings.LANGUAGE_CODE[:2])
    class Meta:
        model = User


class RecordFilterForm(Form):
    title = CharField(required=True, min_length=1)
    def get_filter_qs(self, qs=None):
        if qs is None:
            qs = Record.objects.select_related("language").prefetch_related('titles')  # @UndefinedVariable
        t = self.cleaned_data.get('title',None)
        if t:
            qs = qs.filter(titles__title__icontains=t)
        return qs