256 lines
13 KiB
Python
256 lines
13 KiB
Python
from flask_wtf import FlaskForm
|
|
from flask_wtf.file import FileField, FileAllowed
|
|
from wtforms import (StringField, PasswordField, SelectField, TextAreaField,
|
|
DecimalField, BooleanField, IntegerField, HiddenField,
|
|
RadioField)
|
|
from wtforms.validators import (DataRequired, Email, Length, EqualTo,
|
|
Optional, NumberRange, ValidationError)
|
|
from app.models.user import User
|
|
|
|
|
|
# ── Auth ─────────────────────────────────────────────────────────────────────
|
|
|
|
class LoginForm(FlaskForm):
|
|
username = StringField('Username', validators=[DataRequired(), Length(min=3, max=100)])
|
|
password = PasswordField('Password', validators=[DataRequired()])
|
|
remember_me = BooleanField('Keep me logged in')
|
|
|
|
|
|
class ProfileForm(FlaskForm):
|
|
"""Self-service profile update form — available to all authenticated users."""
|
|
full_name = StringField('Full Name', validators=[Optional(), Length(max=150)])
|
|
email = StringField('Email', validators=[DataRequired(), Email(), Length(max=255)])
|
|
current_password = PasswordField('Current Password', validators=[Optional()])
|
|
new_password = PasswordField('New Password', validators=[Optional(), Length(min=6, max=100)])
|
|
confirm_password = PasswordField('Confirm New Password', validators=[EqualTo('new_password', message='Passwords must match.')])
|
|
|
|
def __init__(self, user=None, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
|
|
def validate_email(self, field):
|
|
q = User.query.filter_by(email=field.data).first()
|
|
if self.user and field.data != self.user.email and q:
|
|
raise ValidationError('Email already registered.')
|
|
elif not self.user and q:
|
|
raise ValidationError('Email already registered.')
|
|
|
|
def validate_current_password(self, field):
|
|
"""Require current password only when the user wants to set a new one."""
|
|
if self.new_password.data:
|
|
if not field.data:
|
|
raise ValidationError('Please enter your current password to set a new one.')
|
|
if self.user and not self.user.check_password(field.data):
|
|
raise ValidationError('Current password is incorrect.')
|
|
|
|
|
|
class UserForm(FlaskForm):
|
|
username = StringField('Username', validators=[DataRequired(), Length(min=3, max=100)])
|
|
full_name = StringField('Full Name', validators=[Optional(), Length(max=150)])
|
|
email = StringField('Email', validators=[DataRequired(), Email(), Length(max=255)])
|
|
password = PasswordField('Password', validators=[Optional(), Length(min=6, max=100)])
|
|
confirm_password = PasswordField('Confirm Password', validators=[Optional(), EqualTo('password')])
|
|
role = SelectField('Role', choices=[
|
|
('admin', 'Administrator'),
|
|
('director', 'Director'),
|
|
('inspector', 'Inspector'),
|
|
('project_manager', 'Project Manager'),
|
|
# 'customer' is intentionally excluded — customer accounts are managed via /customers
|
|
], validators=[DataRequired()])
|
|
|
|
def __init__(self, user=None, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
|
|
def validate_password(self, field):
|
|
"""Enforce minimum length only when a new password is actually provided."""
|
|
if field.data and len(field.data) < 6:
|
|
raise ValidationError('Password must be at least 6 characters.')
|
|
|
|
def validate_confirm_password(self, field):
|
|
"""Require confirmation to match only when a new password is provided."""
|
|
if self.password.data and field.data != self.password.data:
|
|
raise ValidationError('Passwords must match.')
|
|
|
|
def validate_username(self, field):
|
|
q = User.query.filter_by(username=field.data).first()
|
|
if self.user:
|
|
if field.data != self.user.username and q:
|
|
raise ValidationError('Username already exists.')
|
|
elif q:
|
|
raise ValidationError('Username already exists.')
|
|
|
|
def validate_email(self, field):
|
|
q = User.query.filter_by(email=field.data).first()
|
|
if self.user:
|
|
if field.data != self.user.email and q:
|
|
raise ValidationError('Email already registered.')
|
|
elif q:
|
|
raise ValidationError('Email already registered.')
|
|
|
|
|
|
# ── Facility / Area ──────────────────────────────────────────────────────────
|
|
|
|
class FacilityForm(FlaskForm):
|
|
name = StringField('Facility Name', validators=[DataRequired(), Length(max=255)])
|
|
address = TextAreaField('Address', validators=[Optional()])
|
|
contact_person = StringField('Contact Person', validators=[Optional(), Length(max=100)])
|
|
contact_phone = StringField('Contact Phone', validators=[Optional(), Length(max=20)])
|
|
project_id = SelectField('Contract', coerce=int, validators=[Optional()])
|
|
active = BooleanField('Active', default=True)
|
|
|
|
|
|
class AreaForm(FlaskForm):
|
|
name = StringField('Area Name', validators=[DataRequired(), Length(max=255)])
|
|
area_type = SelectField('Area Type', choices=[
|
|
('restroom','Restroom'), ('lobby','Lobby'), ('hallway','Hallway'),
|
|
('office','Office'), ('kitchen','Kitchen'), ('storage','Storage'),
|
|
('floor','Floor'),
|
|
('outdoor','Outdoor'), ('other','Other'),
|
|
], validators=[Optional()])
|
|
facility_id = SelectField('Facility', coerce=int, validators=[DataRequired()])
|
|
|
|
|
|
# ── Templates ────────────────────────────────────────────────────────────────
|
|
|
|
class InspectionTemplateForm(FlaskForm):
|
|
name = StringField('Template Name', validators=[DataRequired(), Length(max=255)])
|
|
description = TextAreaField('Description', validators=[Optional()])
|
|
frequency = SelectField('Inspection Frequency', choices=[
|
|
('daily','Daily'), ('weekly','Weekly'),
|
|
('monthly','Monthly'), ('quarterly','Quarterly'),
|
|
], validators=[DataRequired()])
|
|
|
|
|
|
class ChecklistItemForm(FlaskForm):
|
|
category = StringField('Category', validators=[DataRequired(), Length(max=100)])
|
|
item_description = TextAreaField('Item Description', validators=[DataRequired()])
|
|
scoring_type = SelectField('Scoring Type', choices=[
|
|
('pass_fail','Pass/Fail'), ('rating_5','5-Point Rating'), ('rating_10','10-Point Rating'),
|
|
], validators=[DataRequired()])
|
|
weight = DecimalField('Weight', validators=[Optional(), NumberRange(min=0.1, max=10.0)], default=1.00)
|
|
requires_photo = BooleanField('Requires Photo Evidence', default=False)
|
|
display_order = IntegerField('Display Order', validators=[Optional()], default=0)
|
|
|
|
|
|
# ── Inspections ──────────────────────────────────────────────────────────────
|
|
|
|
class StartInspectionForm(FlaskForm):
|
|
template_id = SelectField('Template', coerce=int, validators=[DataRequired()])
|
|
facility_id = SelectField('Facility', coerce=int, validators=[DataRequired()])
|
|
|
|
|
|
class ChecklistResultForm(FlaskForm):
|
|
"""Dynamically rendered per checklist item — base validators only."""
|
|
score = DecimalField('Score', validators=[Optional(), NumberRange(min=0, max=10)])
|
|
passed = HiddenField('Passed') # 'true' / 'false' / ''
|
|
comments = TextAreaField('Comments', validators=[Optional(), Length(max=1000)])
|
|
photo = FileField('Photo', validators=[
|
|
Optional(),
|
|
FileAllowed(['jpg','jpeg','png','gif'], 'Images only.')
|
|
])
|
|
|
|
|
|
# ── Issues ───────────────────────────────────────────────────────────────────
|
|
|
|
class IssueForm(FlaskForm):
|
|
area_id = SelectField('Area', coerce=int, validators=[DataRequired()])
|
|
severity = SelectField('Severity', choices=[
|
|
('low','Low'), ('medium','Medium'), ('high','High'), ('critical','Critical'),
|
|
], validators=[DataRequired()])
|
|
description = TextAreaField('Description', validators=[DataRequired(), Length(max=2000)])
|
|
photo = FileField('Photo Evidence', validators=[
|
|
Optional(),
|
|
FileAllowed(['jpg','jpeg','png','gif'], 'Images only.')
|
|
])
|
|
assigned_to = SelectField('Assign To', coerce=int, validators=[Optional()])
|
|
|
|
|
|
class IssueUpdateForm(FlaskForm):
|
|
status = SelectField('Status', choices=[
|
|
('open','Open'), ('in_progress','In Progress'),
|
|
('pending_verification','Pending Verification'), ('resolved','Resolved'),
|
|
], validators=[DataRequired()])
|
|
assigned_to = SelectField('Assign To', coerce=int, validators=[Optional()])
|
|
update_notes = TextAreaField('Update Notes', validators=[Optional(), Length(max=1000)])
|
|
result_notes = TextAreaField('Result Notes', validators=[Optional(), Length(max=2000)])
|
|
result_photos = FileField('Result Photos', validators=[
|
|
Optional(),
|
|
FileAllowed(['jpg','jpeg','png','gif'], 'Images only.')
|
|
])
|
|
|
|
# ── Projects ─────────────────────────────────────────────────────────────────
|
|
|
|
class ProjectForm(FlaskForm):
|
|
name = StringField('Contract Name', validators=[DataRequired(), Length(max=255)])
|
|
description = TextAreaField('Description', validators=[Optional()])
|
|
project_manager_id = SelectField('Project Manager', coerce=int, validators=[Optional()])
|
|
active = BooleanField('Active', default=True)
|
|
|
|
|
|
class CustomerAssignmentForm(FlaskForm):
|
|
user_id = SelectField('Customer User', coerce=int, validators=[DataRequired()])
|
|
facility_id = SelectField('Facility Scope', coerce=int, validators=[Optional()])
|
|
|
|
|
|
class CustomerUserForm(FlaskForm):
|
|
"""Create / edit a customer-role user account.
|
|
Used exclusively in the Customer Management UI.
|
|
Password is required on create; optional on edit.
|
|
"""
|
|
username = StringField('Username', validators=[DataRequired(), Length(min=3, max=100)])
|
|
full_name = StringField('Full Name', validators=[Optional(), Length(max=150)])
|
|
email = StringField('Email', validators=[DataRequired(), Email(), Length(max=255)])
|
|
password = PasswordField('Password', validators=[Optional(), Length(min=8)])
|
|
confirm_password = PasswordField('Confirm Password',
|
|
validators=[Optional(), EqualTo('password',
|
|
message='Passwords must match.')])
|
|
|
|
def __init__(self, user=None, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self._user = user # existing user instance (edit mode) or None (create mode)
|
|
|
|
def validate_username(self, field):
|
|
from app.models.user import User
|
|
existing = User.query.filter_by(username=field.data).first()
|
|
if existing and (self._user is None or existing.id != self._user.id):
|
|
raise ValidationError('Username already in use.')
|
|
|
|
def validate_email(self, field):
|
|
from app.models.user import User
|
|
existing = User.query.filter_by(email=field.data).first()
|
|
if existing and (self._user is None or existing.id != self._user.id):
|
|
raise ValidationError('Email address already in use.')
|
|
|
|
def validate_password(self, field):
|
|
"""Password is required when creating a new account."""
|
|
if self._user is None and not field.data:
|
|
raise ValidationError('Password is required for new accounts.')
|
|
|
|
class CustomerInviteForm(FlaskForm):
|
|
"""Simplified form for creating a customer account via email invitation.
|
|
|
|
Admin enters Full Name and Email only. A username is auto-generated
|
|
from the email address. The customer sets their own username and
|
|
password via the emailed link.
|
|
"""
|
|
full_name = StringField('Full Name', validators=[DataRequired(), Length(max=150)])
|
|
email = StringField('Email', validators=[DataRequired(), Email(), Length(max=255)])
|
|
|
|
def validate_email(self, field):
|
|
if User.query.filter_by(email=field.data.strip().lower()).first():
|
|
raise ValidationError('An account with this email address already exists.')
|
|
|
|
|
|
class SetPasswordForm(FlaskForm):
|
|
"""Public form for customer to choose their username and password via emailed link."""
|
|
username = StringField('Choose a Username', validators=[DataRequired(), Length(min=3, max=100)])
|
|
password = PasswordField('Password', validators=[DataRequired(), Length(min=8, max=100)])
|
|
confirm_password = PasswordField('Confirm Password',
|
|
validators=[DataRequired(),
|
|
EqualTo('password', message='Passwords must match.')])
|
|
|
|
def validate_username(self, field):
|
|
existing = User.query.filter_by(username=field.data.strip()).first()
|
|
if existing:
|
|
raise ValidationError('This username is already taken. Please choose another.') |