from django import forms
import itertools
import pydash as _; _.map = _.map_; _.filter = _.filter_

from .models import User, UserFinancialInfo, Team, ContractorResume, ContractorResumeFiles, GENDERS
from common.models import Location, LiveImageUpload
from projects.models import Project, Realty, BuildingClassfication, ConstructionType
from specializations.models import Specialization


class TeamForm(forms.ModelForm):
    class Meta:
        model = Team
        
        fields = (
            'name',
            'owner',
        )


class ContractorResumeForm(forms.ModelForm):
    class Meta:
        model = ContractorResume
        
        fields = (
            'text',
        )

class ContractorResumeFilesForm(forms.ModelForm):
    class Meta:
        model = ContractorResumeFiles

        fields = (
            'type',
            'resume',
        )


class UserProfileEditForm(forms.ModelForm):
    gender = forms.ChoiceField(
        choices=GENDERS,
        widget=forms.RadioSelect,
        required=False,
    )
    
    live_image = forms.ModelChoiceField(
        queryset=LiveImageUpload.objects.all(),
        required=False,
    )
    
    class Meta:
        model = User
        
        fields = (
            'contractor_specializations',
            'cro',
            'date_of_birth',
            'first_name',
            'gender',
            'last_name',
            'location',
            'patronym',
            'phone',
            'phone2',
            'skype',
            'website',
        )
        
        widgets = {
            'date_of_birth': forms.TextInput(attrs={'class': 'datepicker'}), # TODO: Use common format with jQueryUI Datepicker
        }
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class UserProfileBasicInfoEditForm(forms.ModelForm):
    live_image = forms.ModelChoiceField(
        queryset=LiveImageUpload.objects.all(),
        required=False,
    )
    
    class Meta:
        model = User
        
        fields = (
            'contractor_specializations',
            'first_name',
            'last_name',
            'location',
            'patronym',
        )
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class UserFinancialInfoEditForm(forms.ModelForm):
    class Meta:
        model = UserFinancialInfo
        
        fields = (
            'address',
            'credit_card_number',
            'date_of_birth',
            'fio',
            'inn',
            'legal_status',
            'passport_issue_date',
            'passport_issued_by',
            'passport_number',
            'passport_scan',
            'passport_series',
            'phone',
            'residency',
            'subdivision_code',
            'yandex_money',
        )
        
        widgets = {
            # 'date_of_birth': forms.TextInput(attrs={'class': 'datepicker'}),
            # 'passport_issue_date': forms.TextInput(attrs={'class': 'datepicker'}),
            'legal_status': forms.RadioSelect(),
            'residency': forms.RadioSelect(),
        }
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)
        
        self.fields['residency'].choices = self.fields['residency'].choices[1:]
        self.fields['legal_status'].choices = self.fields['legal_status'].choices[1:]
        
        # self.fields['residency'].empty_label = None
        # self.fields['residency'].widget.choices = self.fields['residency'].choices


class ContractorFilterForm(forms.Form):
    CONTRACTOR_ORDER_CHOICES = (
        ('', 'Сортировать по...'),
        ('name', 'Названию'),
        ('created', 'Дате размещения'),
        ('views', 'Просмотрам'),
        ('reviews', 'Отзывам'),
        ('rating', 'Рейтингу'),
        # ('project_number', 'Кол-ву выполн. проектов'),
    )
    
    PARTY_TYPES = (
        ('all', 'Все'),
        ('teams', 'Группы'),
        ('contractors', 'Исполнители'),
    )
    
    order_by = forms.ChoiceField(required=False, choices=CONTRACTOR_ORDER_CHOICES)
    last_order_by = forms.ChoiceField(required=False, choices=CONTRACTOR_ORDER_CHOICES)
    reverse_order = forms.BooleanField(required=False)


    try:  #TODO: dirty
        qs = Specialization.objects.root_nodes()[0].get_descendants()
    except:
        qs = Specialization.objects

    specialization = forms.ModelChoiceField(
        queryset=qs,
        required=False,
    )

    try:  # TODO: dirty
        qs = Location.objects.root_nodes()[0].get_descendants()
    except:
        qs = Location.objects

    location = forms.ModelChoiceField(
        queryset=qs,
        required=False,
    )
    
    building_classification = forms.ModelChoiceField(
        queryset=BuildingClassfication.objects,
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
        empty_label='',
    )
    
    work_type = forms.ChoiceField(
        choices=tuple(itertools.chain((('',''),), Project.WORK_TYPES)),
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
    )
    
    construction_type = forms.ModelChoiceField(
        queryset=ConstructionType.objects,
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
        empty_label='',
    )
    
    cro = forms.BooleanField(required=False)
    
    party_types = forms.ChoiceField(
        choices=PARTY_TYPES,
        required=False,
    )
    
    last_party_types = forms.ChoiceField(
        choices=PARTY_TYPES,
        required=False,
    )
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class CustomerProfileProjectRealtyForm(forms.Form):
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        self.customer = kwargs.pop('customer')
        
        super().__init__(*args, **kwargs)
        
        realties = _.uniq(tuple(p.realty for p in self.customer.projects.all()))
        
        self.fields['realty'] = forms.ChoiceField(
            widget=forms.Select(attrs={
                'class': 'selectpicker',
                'onchange': "$(this).closest('form').submit()",
            }),
            
            choices=(('', 'Все объекты'),) + tuple((r.pk, r.name) for r in realties),
            required=False,
        )


# import code; code.interact(local=dict(globals(), **locals()))