from django import forms
from mptt.forms import TreeNodeChoiceField

from common.models import LiveImageUpload
from projects.models import BuildingClassfication
from specializations.models import Specialization
from .models import WorkSell, WorkSellPhoto


class ContractorWorkSellTrashForm(forms.Form):
    pk = forms.ModelChoiceField(queryset=WorkSell.objects.none())

    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)

        self.fields['pk'].queryset = self.request.user.work_sell.all()


class WorkSellForm(forms.ModelForm):
    # Define a form field manually for a reverse model vield:

    # photos = forms.ModelMultipleChoiceField(
    #     queryset=WorkSellPhoto.objects.none(),
    #     widget=forms.CheckboxSelectMultiple,
    #     required=False,
    # )
    #
    # live_images = forms.ModelMultipleChoiceField(
    #     queryset=LiveImageUpload.objects.all(),
    #     widget=forms.CheckboxSelectMultiple,
    #     required=False,
    # )
    #
    # building_classification = TreeNodeChoiceField(
    #     BuildingClassfication.objects.exclude(name='_root'),
    #     widget=forms.Select(attrs={
    #         'class': 'selectpicker'
    #     }),
    #     required=False,
    #     level_indicator='  ',
    # )

    class Meta:
        model = WorkSell

        fields = (
            'building_classification',
            'construction_type',
            'specialization',
            'specializations',
            'contractor',
            'location',
            'budget',
            'currency',
            'description',
            'el_format',
            'name',
            'term',
            'term_type',
        )

        # widgets = {
        #     'construction_type': forms.Select(attrs={'class': 'selectpicker'}),
        #     'currency': forms.Select(attrs={'class': 'selectpicker'}),
        #     'term_type': forms.Select(attrs={'class': 'selectpicker'}),
        #     'work_type': forms.Select(attrs={'class': 'selectpicker -project-work-type-select-field'}),
        # }

    # def clean(self):
    #     print("CLEAN", self.cleaned_data)
    #     data = self.cleaned_data.get('specialization')
    #     print("before data = ", data)
    #     # print('spec = ', data.split(','))
    #     # self.cleaned_data['specialization'] = data.split(',')
    #     super().clean()

    # def clean_specialization(self):
    #     data = self.cleaned_data['specialization']
    #     print("before data = ", data)
    #     data = data.split(',')
    #     print("after data = ", data)
    #     return data

    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)

        # self.fields['photos'].queryset = self.instance.photos.all()


class WorkSellFilterForm(forms.ModelForm):
    keywords = forms.CharField(required=False, max_length=255)

    class Meta:
        model = WorkSell
        fields = (
            'location',
            'building_classification',
            'construction_type',
            'specialization',
        )

        widgets = {
            'building_classification': forms.Select(attrs={'class': 'selectpicker'}),
            'construction_type': forms.Select(attrs={'class': 'selectpicker'}),
        }

    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)

        self.fields['specialization'].required = False

        self.fields['specialization'].queryset = Specialization.objects.root_nodes()[0].get_descendants()