app.admin.forms

 1"""Flask-WTF admin form definitions for Vision Hub application."""
 2from flask_wtf import FlaskForm
 3from wtforms import StringField, PasswordField, RadioField, SubmitField, SelectField, DateField, TextAreaField, URLField, FormField, FieldList, BooleanField, SelectMultipleField
 4from wtforms.validators import ValidationError, DataRequired, EqualTo, Optional, URL, Length
 5import sqlalchemy as sa
 6
 7from app import db
 8from app.models import User, Department, Role
 9
10
11def validate_email(self, field):
12    """Validator to check if the email is valid."""
13    value = field.data
14    if '@' not in value or '.' not in value.split('@')[-1]:
15        raise ValidationError('Invalid email address.')
16
17 
18def validate_username(self, field):
19    """Validator to check if the username is already registered."""
20    existing = db.session.scalar(
21        sa.select(User).where(User.username == field.data)
22    )
23    if existing and existing.id != getattr(self, 'user_id', None):
24        raise ValidationError('Email address already registered.')
25
26
27class CreateUserForm(FlaskForm):
28    """Form for creating a new user in the admin interface."""
29    firstName = StringField('First Name', validators=[DataRequired()])
30    surname = StringField('Surname', validators=[DataRequired()])
31    username = StringField('Email', validators=[DataRequired(), validate_email, validate_username])
32    password = PasswordField('Password', validators=[DataRequired(), Length (min=8, max=30)])
33    password2 = PasswordField('Confirm Password', validators=[DataRequired(), EqualTo('password')])
34    role = SelectField('Role', coerce=int, validators=[DataRequired()])
35    is_onboarding = RadioField('Does staff member need onboarding?', choices=[('yes', 'Yes'), ('no', 'No')],validators=[DataRequired()])
36    manager= SelectField('Manager', coerce=int, validators=[Optional()])
37    department = SelectField('Department', coerce=int, validators=[DataRequired()])
38    job_title = StringField('Position', validators=[DataRequired()])
39    submit = SubmitField('Register')
40
41
42class EditUserForm(FlaskForm):
43    """Form for editing an existing user in the admin interface."""
44    first_name = StringField('First Name', validators=[DataRequired()])
45    surname = StringField('Surname', validators=[DataRequired()])
46    username = StringField('Email', validators=[DataRequired(), validate_email, validate_username])
47    password = PasswordField('Password Reset', validators=[Optional(), Length (min=8, max=30)])
48    password2 = PasswordField('Confirm New Password', validators=[Optional(), EqualTo('password')])
49    role = SelectField('Role', coerce=int, validators=[DataRequired()])
50    is_onboarding = RadioField('Does staff member need onboarding?', choices=[('yes', 'Yes'), ('no', 'No')],validators=[DataRequired()])
51    manager= SelectField('Manager', coerce=int, validators=[Optional()])
52    department = SelectField('Department', coerce=int, validators=[DataRequired()])
53    job_title = StringField('Position', validators=[DataRequired()])
54    dateStarted = DateField('Start Date', format='%d-%m-%y', validators=[Optional()])
55    submit = SubmitField('Confirm')             
56
57
58class OptionForm(FlaskForm):
59    """Sub-form for creating or editing an option in a question."""
60    option_text = StringField("Option Text", validators=[DataRequired(), Length(max=500)])
61    is_correct = BooleanField("Correct Answer")
62
63    class Meta:
64        csrf = False
65
66class QuestionForm(FlaskForm):
67    """Sub-form for a quiz question; contains four OptionForm fields."""
68    question_text = TextAreaField("Question Text", validators=[DataRequired(), Length(max=1000)])
69    option1 = FormField(OptionForm)
70    option2 = FormField(OptionForm)
71    option3 = FormField(OptionForm)
72    option4 = FormField(OptionForm)
73
74    class Meta:
75        csrf = False
76
77class CreateTrainingModuleForm(FlaskForm):
78    """Admin form to build a TrainingModule with questions and pathways."""
79    module_title = StringField("Module Title", validators=[DataRequired(), Length(max=150)])
80    module_description = TextAreaField("Description", validators=[DataRequired()])
81    module_instructions = TextAreaField("Instructions", validators=[DataRequired()])
82    video_url = URLField("Video URL (Optional)", validators=[Optional(), URL(require_tld=True), Length(max=300)])
83    pathways = SelectMultipleField("Assign to Pathways", coerce=int, validators=[DataRequired()])
84    questions = FieldList(FormField(QuestionForm), min_entries=1, max_entries=20)
85    submit = SubmitField("Create")