def edit_extractor(extractor_id):
extractor = Extractor.get_by_id(extractor_id)
if not extractor:
abort(404)
form = EditExtractorForm(request.form, departments=[d.id for d in extractor.departments])
form.departments.choices = [(d.id, d.name) for d in Department.query.order_by('name')]
if request.method == 'POST':
extractor.departments = [Department.get_by_id(int(d)) for d in form.departments.data]
extractor.save()
flash('Extractor updated.', 'info')
return redirect(url_for('admin.admin_dashboard'))
return render_template("admin/editExtractor.html", form=form, extractor=extractor)
def edit_user(user_id):
user = User.get_by_id(user_id)
if not user:
abort(404)
form = EditUserForm(request.form, departments=[d.id for d in user.departments])
form.departments.choices = [(d.id, d.name) for d in Department.query.order_by('name')]
if request.method == 'POST':
user.departments = [Department.get_by_id(int(d)) for d in form.departments.data]
user.save()
flash('User updated.', 'info')
return redirect(url_for('admin.admin_dashboard'))
return render_template("admin/editUser.html", form=form, user=user)
def edit_user(user_id):
user = User.get_by_id(user_id)
if not user:
abort(404)
form = EditUserForm(request.form,
departments=[d.id for d in user.departments])
form.departments.choices = [(d.id, d.name)
for d in Department.query.order_by('name')]
if request.method == 'POST':
user.departments = [
Department.get_by_id(int(d)) for d in form.departments.data
]
user.save()
flash('User updated.', 'info')
return redirect(url_for('admin.admin_dashboard'))
return render_template("admin/editUser.html", form=form, user=user)
def register():
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
invite_code = Invite_Code.query.filter_by(
code=form.invite_code.data).first()
invite_code.used = True
invite_code.save()
new_user = User.create(username=form.username.data,
email=form.email.data,
password=form.password.data,
active=True)
new_user.departments.append(
Department.get_by_id(invite_code.department_id))
flash("Thank you for registering. You can now log in.", 'success')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
def edit_extractor(extractor_id):
extractor = Extractor.get_by_id(extractor_id)
if not extractor:
abort(404)
form = EditExtractorForm(request.form,
departments=[d.id for d in extractor.departments])
form.departments.choices = [(d.id, d.name)
for d in Department.query.order_by('name')]
if request.method == 'POST':
extractor.departments = [
Department.get_by_id(int(d)) for d in form.departments.data
]
extractor.save()
flash('Extractor updated.', 'info')
return redirect(url_for('admin.admin_dashboard'))
return render_template("admin/editExtractor.html",
form=form,
extractor=extractor)
def register():
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
invite_code = Invite_Code.query.filter_by(code=form.invite_code.data).first()
invite_code.used = True
invite_code.save()
new_user = User.create(
username=form.username.data,
email=form.email.data,
password=form.password.data,
active=True
)
new_user.departments.append(Department.get_by_id(invite_code.department_id))
flash("Thank you for registering. You can now log in.", 'success')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
