from flask_wtf import FlaskForm
from flask_wtf.file import FileAllowed
from wtforms import \
    StringField, \
    PasswordField, \
    TextAreaField, \
    SelectField, \
    BooleanField, \
    FileField
from wtforms import validators
from werkzeug.datastructures import FileStorage

from .models import User
from .content import content_dict as content
from .core import models

form_content = content['estimate']['form']


# ESTIMATION FORM #############################################################

class EstimateForm(FlaskForm):

    # email = StringField(
    #     label=form_content['email']['label'],
    #     description=form_content['email']['description'],
    #     validators=[
    #         validators.DataRequired(),
    #         validators.Email(),
    #     ],
    # )

    first_name = StringField(
        label=form_content['first_name']['label'],
        description=form_content['first_name']['description'],
        validators=[
            validators.Optional(),
            validators.Length(max=1024),
        ],
        render_kw={
            "placeholder": form_content['first_name']['placeholder']
        },
    )
    last_name = StringField(
        label=form_content['last_name']['label'],
        description=form_content['last_name']['description'],
        validators=[
            validators.Optional(),
            validators.Length(max=1024),
        ],
        render_kw={
            "placeholder": form_content['last_name']['placeholder']
        },
    )
    institution = StringField(
        label=form_content['institution']['label'],
        description=form_content['institution']['description'],
        validators=[
            validators.Optional(),
            validators.Length(max=1024),
        ],
    )
    use_train_below_km = SelectField(
        label=form_content['use_train_below_km']['label'],
        description=form_content['use_train_below_km']['description'],
        default=300,
        choices=[
            (0, 'Do not use train'),
            (300, '300 km'),
            (500, '500 km'),
            (700, '700 km'),
            (1000, '1000 km'),
        ],
        coerce=int,
    )
    comment = TextAreaField(
        label=form_content['comment']['label'],
        description=form_content['comment']['description'],
        validators=[
            validators.Optional(),
            validators.Length(max=2048),
        ],
    )
    origin_addresses = TextAreaField(
        label=form_content['origin_addresses']['label'],
        description=form_content['origin_addresses']['description'],
        validators=[
            validators.DataRequired(),
        ],
        render_kw={
            "placeholder": form_content['origin_addresses']['placeholder']
        },
    )
    destination_addresses = TextAreaField(
        label=form_content['destination_addresses']['label'],
        description=form_content['destination_addresses']['description'],
        validators=[
            validators.DataRequired(),
        ],
        render_kw={
            "placeholder": form_content['destination_addresses']['placeholder']
        },
    )
    origin_addresses_file = FileField(
        label=form_content['origin_addresses_file']['label'],
        description=form_content['origin_addresses_file']['description'],
        validators=[
            # We disabled validators because they bug with multiple FileFields
            # validators.Optional(),
            # FileAllowed(
            #     ['csv', 'xls', 'xlsx'],
            #     form_content['origin_addresses_file']['error']
            # )
        ],
    )
    destination_addresses_file = FileField(
        label=form_content['destination_addresses_file']['label'],
        description=form_content['destination_addresses_file']['description'],
        validators=[
            # We disabled validators because they bug with multiple FileFields
            # validators.Optional(),
            # FileAllowed(
            #     ['csv', 'xls', 'xlsx'],
            #     form_content['destination_addresses_file']['error']
            # )
        ],
    )

    upload_set = ['csv', 'xls', 'xlsx']

    # compute_optimal_destination = BooleanField(
    #     label=form_content['compute_optimal_destination']['label'],
    #     description=form_content['compute_optimal_destination']['description'],
    #     default=False,
    #     validators=[
    #         validators.Optional(),
    #     ],
    # )
    # use_atmosfair_rfi = BooleanField(
    #     label=form_content['use_atmosfair_rfi']['label'],
    #     description=form_content['use_atmosfair_rfi']['description'],
    #     default=False,
    #     validators=[
    #         validators.Optional(),
    #     ],
    # )

    def validate(self):
        check_validate = super(EstimateForm, self).validate()

        # Do our validators pass?
        if not check_validate:
            return False

        uses_at_least_one_model = False
        for model in models:
            use_model = getattr(self, 'use_model_%s' % model.slug)
            #print("Model data", model.slug, use_model.data)
            if use_model.data:
                uses_at_least_one_model = True

        if not uses_at_least_one_model:
            last_model = getattr(self, 'use_model_%s' % models[-1].slug)
            last_model.errors.append(
                "Please select at least one plane model, "
                "<em>even for train-only estimations.</em>"
            )
            return False

        # Check uploaded files' extensions, if any
        # We have to do this "by hand" because of a bug in flask wtf
        if isinstance(self.origin_addresses_file.data, FileStorage):
            fn = self.origin_addresses_file.data.filename.lower()
            if fn and not any(fn.endswith('.' + x) for x in self.upload_set):
                self.origin_addresses_file.errors.append(
                    form_content['origin_addresses_file']['error']
                )
                return False
        if isinstance(self.destination_addresses_file.data, FileStorage):
            fn = self.destination_addresses_file.data.filename.lower()
            if fn and not any(fn.endswith('.' + x) for x in self.upload_set):
                self.destination_addresses_file.errors.append(
                    form_content['destination_addresses_file']['error']
                )
                return False

        return True


# Add the models' checkboxes to the above Form
for model in models:
    setattr(  # setattr() takes no keyword arguments -.-
        EstimateForm,
        'use_model_%s' % model.slug,
        BooleanField(
            label=model.name,
            # description=model.short_description,
            default=model.selected,
            validators=[
                validators.Optional(),
            ],
        )
    )


# LOGIN FORM ##################################################################

class LoginForm(FlaskForm):
    username = StringField(u'Username', validators=[validators.required()])
    password = PasswordField(u'Password', validators=[validators.optional()])

    def validate(self):
        check_validate = super(LoginForm, self).validate()

        # Do our validators pass?
        if not check_validate:
            return False

        # Does the user even exist?
        user = User.query.filter_by(username=self.username.data).first()
        if not user:
            self.username.errors.append('Invalid username or password')
            return False

        # Do the passwords match?
        if not user.check_password(self.password.data):
            self.username.errors.append('Invalid username or password')
            return False

        return True