From 46ff59c4528b09b2e8aaf27cf0af36d264a4f172 Mon Sep 17 00:00:00 2001 From: Bo-Chun Louis Chen <louistw@uab.edu> Date: Thu, 24 May 2018 14:56:24 -0500 Subject: [PATCH] Added EmployeeAssignForm --- app/admin/forms.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/app/admin/forms.py b/app/admin/forms.py index 3c08ced..2c1ca69 100644 --- a/app/admin/forms.py +++ b/app/admin/forms.py @@ -1,6 +1,8 @@ from flask_wtf import FlaskForm from wtforms import StringField, SubmitField from wtforms.validators import DataRequired +from wtforms.ext.sqlalchemy.fields import QuerySelectField +from ..models import Department, Role class DepartmentForm(FlaskForm): """ @@ -17,3 +19,13 @@ class RoleForm(FlaskForm): name = StringField('Name', validators=[DataRequired()]) description = StringField('Description', validators=[DataRequired()]) submit = SubmitField('Submit') + +class EmployeeAssignForm(FlaskForm): + """ + Form for admin to assign departments and roles to employees + """ + department = QuerySelectField(query_factory=lambda: Department.query.all(), + get_label="name") + role = QuerySelectField(query_factory=lambda: Role.query.all(), + get_label="name") + submit = SubmitField('Submit') -- GitLab