forms.py

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, EmailField, SelectField, SelectMultipleField, IntegerField, TextAreaField
from wtforms.validators import DataRequired, Length, Email, NumberRange, Optional


diets = ['None', 'Gluten Free', 'Ketogenic', 'Vegetarian', 'Lacto-Vegetarian', 
        'Ovo-Vegetarian', 'Vegan', 'Pescatarian', 'Paleo', 'Primal',
        'Low FODMAP', 'Whole30']

intolerances = ['Dairy', 'Egg', 'Gluten', 'Grain', 'Peanut', 'Seafood',
                'Sesame', 'Shellfish', 'Soy', 'Sulfite', 'Tree Nut', 'Wheat']


class SignupForm(FlaskForm):
    """Form to sign up a user"""

    username = StringField('Username', validators=[DataRequired(), Length(max=20)])
    email = EmailField('Email', validators=[DataRequired(), Length(max=50), Email()])
    password = PasswordField('Password', validators=[DataRequired(), Length(min=6)])
    diet = SelectField('Diet (Optional)', choices = [(diet, diet) for diet in diets])
    intolerances = SelectMultipleField('Allergies/Intolerances (Optional)', choices=[(intolerance, intolerance) for intolerance in intolerances])
    exclude_ingredients = StringField('Ingredients to Exclude (Optional)')
   

class LoginForm(FlaskForm):
    """Form to log in a user"""

    username = StringField('Username', validators=[DataRequired(), Length(max=20)])
    password = PasswordField('Password', validators=[DataRequired(), Length(min=6)])


class EditProfileForm(FlaskForm):
    """Form to edit a user's profile"""

    username = StringField('Username', validators=[DataRequired(), Length(max=20)])
    email = EmailField('Email', validators=[DataRequired(), Length(max=50), Email()])
    password = PasswordField('Please enter your password to save your changes:', validators=[DataRequired(), Length(min=6)])
    diet = SelectField('Diet (Optional)', choices = [(diet, diet) for diet in diets])
    intolerances = SelectMultipleField('Allergies/Intolerances (Optional)', choices=[(intolerance, intolerance) for intolerance in intolerances])
    exclude_ingredients = StringField('Ingredients to Exclude (Optional)')


class RecipeForm(FlaskForm):
    """Form for a user to optionally filter out recipes"""

    diet = SelectField('Diet', choices = [(diet, diet) for diet in diets])
    intolerances = SelectMultipleField('Food Allergies/Intolerances', choices=[(intolerance, intolerance) for intolerance in intolerances])
    exclude_ingredients = StringField('Ingredients to Exclude (e.g. radish, onions, etc.)')
    food_type = StringField('Food Type and/or Meal Type (e.g. pasta, chicken, salad, dinner, etc.)')
    num_of_recipes = IntegerField('Number of recipes:', validators=[NumberRange(min=1, max=100), Optional()])


class NoteForm(FlaskForm):
    """Form to add notes to a saved recipe"""

    note = TextAreaField('Recipe Notes')