from django import forms
from django.forms import ModelForm
import itertools

from .models import User, ContractorFinancialInfo
from common.models import Location
from projects.models import Project, BuildingClassfication, ConstructionType
from specializations.models import Specialization


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

        if kwargs.get('instance'):
            if kwargs.get('instance').is_contractor():
                self.fields['contractor_specializations'].queryset = kwargs.get('instance').contractor_specializations.all()

    class Meta:
        model = User
        
        fields = (
            'first_name',
            'last_name',
            'patronym',
            'location',
            'date_of_birth',
            'website',
            'skype',
            'avatar',
            'contractor_specializations',
            'cro',
            'phone',
            'gender',
        )

        widgets = {
            'date_of_birth': forms.SelectDateWidget(years=range(1940, 2015)),
            # 'contractor_specializations': forms.Select(),
        }


class ContractorFilterForm(forms.Form):
    # PROJECT_ORDER_CHOICES = ( # "Упорядочить по"...
    #     ('name', 'названию'),
    #     ('budget', 'цене'),
    #     ('created', 'дате размещения'),
    #     ('views', 'просмотрам'),
    # )
    # 
    # order_by = forms.ChoiceField(required=False, choices=PROJECT_ORDER_CHOICES)
    # last_order_by = forms.ChoiceField(required=False, choices=PROJECT_ORDER_CHOICES)
    # reverse_order = forms.BooleanField(required=False)
    
    specialization = forms.ModelChoiceField(
        queryset=Specialization.objects.root_nodes()[0].get_descendants(),
        required=False,
    )
    
    location = forms.ModelChoiceField(
        queryset=Location.objects.root_nodes()[0].get_descendants(),
        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)
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class ContractorFinancicalInfoForm(ModelForm):
    def __init__(self, *args, **kwargs):
        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 Meta:
        model = ContractorFinancialInfo
        
        fields = (
            'fio',
            'date_of_birth',
            'phone',
            'residency',
            'legal_status',
            'passport_series',
            'passport_number',
            'subdivision_code',
            'passport_issued_by',
            'passport_issue_date',
            'inn',
            'yandex_money',
            'credit_card_number',
            'passport_scan',
        )

        widgets = {
            'residency': forms.RadioSelect(),
            'legal_status': forms.RadioSelect(),
        }