forms.py

from starlette_wtf import StarletteForm
from wtforms import Form, StringField, PasswordField, SubmitField, validators, SelectMultipleField, widgets
from wtforms.validators import DataRequired, Email, EqualTo
from wtforms.widgets import PasswordInput


class MultiCheckboxField(SelectMultipleField):
    widget = widgets.ListWidget(prefix_label=False)
    option_widget = widgets.CheckboxInput()

class RegistrationForm(Form):
    username = StringField('Username', [validators.Length(min=4, max=25)])
    password = PasswordField('New Password', [
        validators.DataRequired(),
        validators.EqualTo('confirm', message='Passwords must match')
    ])
    confirm = PasswordField('Repeat Password')

    roles = MultiCheckboxField('Roles', coerce=int)
    submit = SubmitField('Create account')



class LoginForm(Form): 
    username = StringField('Username', [validators.DataRequired()])
    password = PasswordField('Password', [validators.DataRequired()])
    submit = SubmitField('Login')

class ChangePasswordForm(Form):
    current_password = PasswordField('Current Password', [validators.DataRequired()])
    new_password = PasswordField('New Password', [
        validators.DataRequired(),
        validators.Length(min=6, message="Password should be at least 6 characters long")
    ])
    confirm_password = PasswordField('Confirm New Password', [
        validators.EqualTo('new_password', message='Passwords must match')
    ])
    submit = SubmitField('Change Password')

        

class CreateAccountForm(StarletteForm):
    email = StringField(
        'Email address',
        validators=[
            DataRequired('Please enter your email address'),
            Email()
        ]
    )

    password = PasswordField(
        'Password',
        widget=PasswordInput(hide_value=False),
        validators=[
            DataRequired('Please enter your password'),
            EqualTo('password_confirm', message='Passwords must match')
        ]
    )

    password_confirm = PasswordField(
        'Confirm Password',
        widget=PasswordInput(hide_value=False),
        validators=[
            DataRequired('Please confirm your password')
        ]
    )
    submit = SubmitField('Create account')