from django import forms
from .models import ABSConfiguration, ABSBeam
from .widgets import UpDataWidget, DownDataWidget, EditUpDataWidget, EditDownDataWidget
from apps.main.models import Configuration
import os

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

    class Meta:
        model = ABSConfiguration
        exclude = ('type', 'status', 'parameters', 'active_beam', 'module_status')


class ABSBeamAddForm(forms.Form):

    up_data = forms.CharField(widget=UpDataWidget, label='')
    down_data = forms.CharField(widget=DownDataWidget, label='')

    def __init__(self, *args, **kwargs):
        super(ABSBeamAddForm, self).__init__(*args, **kwargs)



class ABSBeamEditForm(forms.Form):

    up_data = forms.CharField(widget=EditUpDataWidget, label='')
    down_data = forms.CharField(widget=EditDownDataWidget, label='')

    def __init__(self, *args, **kwargs):
        super(ABSBeamEditForm, self).__init__(*args, **kwargs)

        if 'initial' in kwargs:
            if 'beam' in self.initial:
                self.fields['up_data'].initial  = self.initial['beam']
                self.fields['down_data'].initial  = self.initial['beam']


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 ABSImportForm(forms.Form):

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