def user_edit(id):
user = User.query.get(id)
form = UserForm(next=request.args.get('next', None), obj=user)
form.employee_id.choices = [(0, "")]
form.employee_id.choices.extend([(g.id, g.dept_name) for g in
Employee.query.filter_by(active=True).order_by('emp_name')])
if form.validate_on_submit():
form.populate_obj(user)
db.session.add(user)
db.session.commit()
next_url = form.next.data
if not next_url or next_url == request.path:
next_url = url_for('account.main')
return redirect(url_for('account.user'))
return render_template("account/user.html", form=form)
def user_create():
form = UserForm(next=request.args.get('next', None))
# form.employee.choices = [(1, "1"),]
# form.employee.choices.extend([(g.id, g.dept_name) for g in
# Employee.query.filter_by(active=True).order_by('emp_name')])
if form.validate_on_submit():
user = User()
form.populate_obj(user)
db.session.add(user)
db.session.commit()
flash(_("Welcome, %(name)s", name=user.username), "success")
next_url = form.next.data
if not next_url or next_url == request.path:
next_url = url_for('account.main', username=user.username)
return redirect(next_url)
return render_template("account/user.html", form=form)
