form.py

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField, IntegerField, SelectField, TextAreaField
from wtforms.validators import DataRequired, Email, EqualTo, Length, Optional
import bson.objectid


class SignupForm(FlaskForm):
    fname = StringField('First Name',
                       validators=[DataRequired()])
    lname = StringField('Last Name', validators = [DataRequired()])
    roll = IntegerField('Roll Number', validators = [DataRequired()])
    year = SelectField('Year of Study', choices = [("SY", "Second Year")])
    branch = SelectField('Branch', choices = [("Comps", "Computer Engineering")])
    division = SelectField('Division', choices = [("A", 'A'),  ("B", 'B')])
    id = StringField('Email(A code will be sent to this email for verification)',
                        validators=[Length(min = 6),
                                    Email(message='Enter a valid email.'),
                                    DataRequired()])
    password = PasswordField('Password',
                             validators=[DataRequired(),
                                         Length(min=6, message='Select a stronger password. Minimum length of password should be 6')])
    confirm = PasswordField('Confirm Your Password',
                            validators=[DataRequired(),
                                        EqualTo('password', message='Passwords must match.')])
    submit = SubmitField('Register')


class LoginForm(FlaskForm):
    id = StringField('Email', validators=[DataRequired(),
                                             Email(message='Enter a valid email.')])
    password = PasswordField('Password', validators=[DataRequired()])
    submit = SubmitField('Log In')
    
class CodeForm(FlaskForm):
    code = IntegerField('Code(that was sent to your email)', validators = [DataRequired()])
    submit = SubmitField('Submit')
    
class Feedback(FlaskForm):
    name = SelectField('Select faculty', choices=[], coerce=str)
    subject = StringField('Subject of email(feedback)', validators=[DataRequired()])
    message = TextAreaField('Feedback message', validators=[DataRequired()])
    submit = SubmitField('Submit')

class Notify(FlaskForm):
    name = SelectField('Select student', choices=[], coerce=str)
    subject = StringField('Subject of email(notification)', validators=[DataRequired()])
    message = TextAreaField('Notification', validators=[DataRequired()])
    submit = SubmitField('Submit')
    
class Teacher(FlaskForm):
    subject = SelectField('Select subject', choices=[], validators=[DataRequired()])
    year = SelectField('Select year', choices=[("SY", "SY")], validators=[DataRequired()])
    branch = SelectField('Select branch', choices=[("Comps", "COMP")], validators=[DataRequired()])
    division = SelectField('Select division', choices=[("A", "A"), ("B", "B")], validators=[DataRequired()])
    submit = SubmitField('Get OTP')

class OTPform(FlaskForm):
    code = IntegerField('Enter OTP', validators = [DataRequired()])
    submit = SubmitField('Submit')