def test_users_are_not_extractors(self):
department = DepartmentFactory()
department.save()
user = User(username='******', email='*****@*****.**')
user.departments.append(department)
user.save()
assert Extractor.get_by_id(user.id) == None
assert Extractor.query.filter_by(username=user.username).first() == None
def test_users_are_not_extractors(self):
department = DepartmentFactory()
department.save()
user = User(username='******', email='*****@*****.**')
user.departments.append(department)
user.save()
assert Extractor.get_by_id(user.id) == None
assert Extractor.query.filter_by(
username=user.username).first() == None
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_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)
