from flask_wtf import FlaskForm
from wtforms import StringField, TextAreaField, DateTimeField, SubmitField
from wtforms.validators import DataRequired, Email

class AppointmentForm(FlaskForm):
    customer_email = StringField('Customer Email', validators=[DataRequired(), Email()])
    appointment_date = DateTimeField('Appointment Date', format='%Y-%m-%d %H:%M:%S', validators=[DataRequired()])
    notes = TextAreaField('Notes', validators=[DataRequired()])
    submit = SubmitField('Schedule Appointment')

class RepairNoteForm(FlaskForm):
    device_id = StringField('Device ID', validators=[DataRequired()])
    repair_notes = TextAreaField('Repair Notes', validators=[DataRequired()])
    submit = SubmitField('Submit Repair Notes')

class UserForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired()])
    email = StringField('Email', validators=[DataRequired(), Email()])
    submit = SubmitField('Create User')

class RepairBayForm(FlaskForm):
    name = StringField('Repair Bay Name', validators=[DataRequired()])
    location = StringField('Location', validators=[DataRequired()])
    submit = SubmitField('Create Repair Bay')