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