from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired, FileAllowed
from wtforms import StringField, DecimalField
from wtforms.validators import DataRequired, Email


class MyForm(FlaskForm):
    firstname = StringField('First Name', validators=[DataRequired()])
    lastname = StringField('Last Name', validators=[DataRequired()])
    email = StringField('Email', validators=[DataRequired(), Email()])


class curieForm(FlaskForm):
    ligand = FileField('Ligand', validators=[
        FileRequired(),
        FileAllowed(['pdbqt', 'PDBQT only!'])
    ])
    target = FileField('Receptor / Target', validators=[
        FileRequired(),
        FileAllowed(['pdbqt', 'PDBQT only!'])
    ])
    description = StringField('Description',default="Curie Web Task")

    size_x = DecimalField('Size X',default=25.0)
    size_y = DecimalField('Size Y',default=25.0)
    size_z = DecimalField('Size Z',default=25.0)

    center_x = DecimalField('Center X',default=0)
    center_y = DecimalField('Center Y',default=0)
    center_z = DecimalField('Center Z',default=0)

    email = StringField('Email', validators=[DataRequired(), Email()])