from flask_wtf import FlaskForm
from wtforms import SelectField, StringField, TextAreaField, SubmitField
from wtforms.validators import DataRequired
from models import Company

class TicketForm(FlaskForm):
    company = SelectField('Company', coerce=int, validators=[DataRequired()])
    name = StringField('Name', validators=[DataRequired()])
    subject = StringField('Subject', validators=[DataRequired()])
    ticket_number = StringField('Ticket Number', validators=[DataRequired()])
    status = SelectField('Status', choices=[('Open', 'Open'), ('Closed', 'Closed')], validators=[DataRequired()])
    submit = SubmitField('Submit')

class NoteForm(FlaskForm):
    note = TextAreaField('Note', validators=[DataRequired()])
    submit = SubmitField('Submit')

class CompanyForm(FlaskForm):
    name = StringField('Company Name', validators=[DataRequired()])
    submit = SubmitField('Add Company')

class OnsiteForm(FlaskForm):
    company = SelectField('Company', coerce=int, validators=[DataRequired()])
    day_of_week = SelectField('Day of the Week', choices=[
        ('Monday', 'Monday'),
        ('Tuesday', 'Tuesday'),
        ('Wednesday', 'Wednesday'),
        ('Thursday', 'Thursday'),
        ('Friday', 'Friday')
    ], validators=[DataRequired()])
    note = TextAreaField('Note', validators=[DataRequired()])
    submit = SubmitField('Submit')

    def __init__(self, *args, **kwargs):
        super(OnsiteForm, self).__init__(*args, **kwargs)
        self.company.choices = [(company.id, company.name) for company in Company.query.all()]