diff --git a/app/admin/forms.py b/app/admin/forms.py index 3c08ced9dda1c58ca2ff528569d16502f7355305..2c1ca69868b3afeff9a9215f0f7a3f5d52f7a653 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')