forms.py

# Imports
from flask_wtf import FlaskForm
from wtforms import StringField,PasswordField,SubmitField,BooleanField,RadioField
from wtforms.validators import DataRequired,Length,Email,EqualTo

# User Forms
class MyRegistrationForm(FlaskForm):
    """
        Form used for registration of a normal user.
    """
    username = StringField('Username',validators=[DataRequired(),Length(min=5,max=20)])
    email = StringField('Email',validators=[DataRequired(),Email(message="Please provide valid email!")])
    password = PasswordField('Password',validators=[DataRequired(),Length(min = 6,max=60)])
    confirm_password = PasswordField('Confirm Password',validators=[DataRequired(),Length(min = 6,max=60),EqualTo('password',message="Password and confirm password should match!")])
    submit = SubmitField('Sign Up')

class MyLoginForm(FlaskForm):
    """
        Form used for sign in for any user.
    """
    email = StringField('Email',validators=[DataRequired(),Email(message="Please provide valid email!")])
    password = PasswordField('Password',validators=[DataRequired(),Length(min = 6,max=60)])
    remember = BooleanField('Remember Me')
    submit = SubmitField('Log In')

class UserSearchForm(FlaskForm):
    """
        Form used for search operation for a normal user.
    """
    book_name = StringField('Book Name')
    author_name = StringField('Author Name')
    section_name = StringField('Genre')
    submit = SubmitField('Search')

class LibrarianSearchForm(FlaskForm):
    """
        Form used for search operation for a librarian.
    """
    obj_name = StringField("Search term",validators=[DataRequired()])
    obj_type = RadioField('Search for',choices=[(x,x) for x in ['Book','Author','Section','User']],default='Book')
    submit = SubmitField('Search')

class EditBookForm(FlaskForm):
    """
        Form used for edit book operation for a librarian.
    """
    name = StringField('Book Name',validators=[DataRequired()])
    description = StringField('Description',validators=[DataRequired()])
    file_path = StringField('Content Address',validators=[DataRequired()]) 
    submit = SubmitField('Edit')
    
class EditAuthorForm(FlaskForm):
    """
        Form used for edit author operation for a librarian.
    """
    name = StringField('Author Name')
    description = StringField('Description')
    submit = SubmitField('Edit')

class EditSectionForm(FlaskForm):
    """
        Form used for edit section operation for a librarian.
    """
    name = StringField('Section Name')
    description = StringField('Description')
    submit = SubmitField('Edit')