import os

from django import forms
from apps.main.models import Device, Experiment
from .models import JARSConfiguration, JARSFilter
from .widgets import SpectralWidget
from apps.main.forms import add_empty_choice

def create_choices_from_model(model, filter_id=None):

    #instance = globals()[model]
    choices = model.objects.all().values_list('pk', 'name')
    choices = add_empty_choice(choices,label="New Choice")
    return choices

class JARSConfigurationForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(JARSConfigurationForm, self).__init__(*args, **kwargs)
        instance = getattr(self, 'instance', None)

        if instance and instance.pk:
            devices = Device.objects.filter(device_type__name='jars')
            self.fields['device'].widget.choices = [(device.id, device) for device in devices]
        self.fields['spectral_number'].widget.attrs['readonly'] = True
        self.fields['spectral'].widget = SpectralWidget()

    class Meta:
        model = JARSConfiguration
        exclude = ('type', 'parameters', 'status', 'filter_parms', 'author', 'hash', 'filter')

class JARSFilterForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(JARSFilterForm, self).__init__(*args, **kwargs)
        instance = getattr(self, 'instance', None)

        self.fields['f_decimal'].widget.attrs['readonly'] = True

        if 'initial' in kwargs:
            self.fields['filter_template'] = forms.ChoiceField(
                choices = create_choices_from_model(JARSFilter),
                initial = kwargs['initial']['id']
            )
            # self.fields['name'].initial = kwargs['initial']['id']
            
                # filter_id = kwargs['initial']['filter_id']

                # if filter_id == 0:
                #     for value in self.fields:
                #         if value != 'name':
                #             self.fields.pop(value)
                #     self.fields['name'].label = "Filter Template Name"
                # else:
                #     self.fields['name'] = forms.ChoiceField(choices=create_choices_from_model(JARSFilter, kwargs['initial']['filter_id']))
                #     jars_filter = JARSFilter.objects.get(pk=kwargs['initial']['filter_id'])
                #     labels = [f.name for f in jars_filter._meta.get_fields()]

                #     for label in ['id']:
                #         labels.remove(label)
                #     for label in labels:
                #         self.fields['name'].initial = kwargs['initial']['filter_id']
                #         self.fields[label].initial = getattr(jars_filter,label)
                #     self.fields['name'].label = "Filter Template Name"

    class Meta:
        model = JARSFilter
        exclude = ('name', )

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

        self.fields['f_decimal'].widget.attrs['readonly'] = True

    class Meta:
        model = JARSFilter
        exclude = ()

class ExtFileField(forms.FileField):
    """
    Same as forms.FileField, but you can specify a file extension whitelist.

    >>> from django.core.files.uploadedfile import SimpleUploadedFile
    >>>
    >>> t = ExtFileField(ext_whitelist=(".pdf", ".txt"))
    >>>
    >>> t.clean(SimpleUploadedFile('filename.pdf', 'Some File Content'))
    >>> t.clean(SimpleUploadedFile('filename.txt', 'Some File Content'))
    >>>
    >>> t.clean(SimpleUploadedFile('filename.exe', 'Some File Content'))
    Traceback (most recent call last):
    ...
    ValidationError: [u'Not allowed filetype!']
    """
    def __init__(self, *args, **kwargs):
        extensions = kwargs.pop("extensions")
        self.extensions = [i.lower() for i in extensions]

        super(ExtFileField, self).__init__(*args, **kwargs)

    def clean(self, *args, **kwargs):
        data = super(ExtFileField, self).clean(*args, **kwargs)
        filename = data.name
        ext = os.path.splitext(filename)[1]
        ext = ext.lower()
        if ext not in self.extensions:
            raise forms.ValidationError('Not allowed file type: %s' % ext)


class JARSImportForm(forms.Form):

    file_name = ExtFileField(extensions=['.racp','.json'])