37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
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()]
|