def source_new(request):
"""
Page with the form to create a new Source.
"""
# We can get here one of two ways: either we just got to the form
# page, or we just submitted the form. If POST, we submitted; if
# GET, we just got here.
if request.method == 'POST':
# A form bound to the POST data
sourceForm = ImageSourceForm(request.POST)
pointGenForm = PointGenForm(request.POST)
annotationAreaForm = AnnotationAreaPercentsForm(request.POST)
# is_valid() calls our ModelForm's clean() and checks validity
source_form_is_valid = sourceForm.is_valid()
point_gen_form_is_valid = pointGenForm.is_valid()
annotation_area_form_is_valid = annotationAreaForm.is_valid()
if source_form_is_valid and point_gen_form_is_valid and annotation_area_form_is_valid:
# After calling a ModelForm's is_valid(), an instance is created.
# We can get this instance and add a bit more to it before saving to the DB.
newSource = sourceForm.instance
newSource.default_point_generation_method = PointGen.args_to_db_format(**pointGenForm.cleaned_data)
newSource.image_annotation_area = AnnotationAreaUtils.percentages_to_db_format(**annotationAreaForm.cleaned_data)
newSource.labelset = LabelSet.getEmptyLabelset()
newSource.save()
# Make the user a source admin
newSource.assign_role(request.user, Source.PermTypes.ADMIN.code)
# Add a success message
messages.success(request, 'Source successfully created.')
# Redirect to the source's main page
return HttpResponseRedirect(reverse('source_main', args=[newSource.id]))
else:
# Show the form again, with error message
messages.error(request, 'Please correct the errors below.')
else:
# An unbound form (empty form)
sourceForm = ImageSourceForm()
pointGenForm = PointGenForm()
annotationAreaForm = AnnotationAreaPercentsForm()
# RequestContext needed for CSRF verification of POST form,
# and to correctly get the path of the CSS file being used.
return render_to_response('images/source_new.html', {
'sourceForm': sourceForm,
'pointGenForm': pointGenForm,
'annotationAreaForm': annotationAreaForm,
},
context_instance=RequestContext(request)
)
def labelset_list(request):
"""
Page with a list of all the labelsets
Not sure where to put a link to this page. It's a little less
useful when each source has its own labelset, but this view still
might be useful if someone wants to browse through labelsets that
they could base their labelset off of.
"""
publicSources = Source.objects.filter(visibility=Source.VisibilityTypes.PUBLIC)
publicSourcesWithLabelsets = publicSources.exclude(labelset=LabelSet.getEmptyLabelset())
return render_to_response('annotations/labelset_list.html', {
'publicSourcesWithLabelsets': publicSourcesWithLabelsets,
},
context_instance=RequestContext(request)
)
def test_post_success(self):
"""
Successful creation of a new source.
"""
datetime_before_creation = datetime.datetime.now().replace(microsecond=0)
response = self.client.post(reverse('source_new'), self.source_args)
source_id = Source.objects.latest('create_date').pk
self.assertRedirects(response, reverse('source_main',
kwargs={
'source_id': source_id,
}
))
new_source = Source.objects.get(pk=source_id)
self.assertEqual(new_source.name, self.source_args['name'])
self.assertEqual(new_source.visibility, self.source_args['visibility'])
self.assertEqual(new_source.labelset, LabelSet.getEmptyLabelset())
self.assertEqual(new_source.key1, self.source_args['key1'])
self.assertEqual(new_source.key2, '')
self.assertEqual(new_source.key3, '')
self.assertEqual(new_source.key4, '')
self.assertEqual(new_source.key5, '')
self.assertEqual(new_source.default_point_generation_method, PointGen.args_to_db_format(
point_generation_type=self.source_args['point_generation_type'],
simple_number_of_points=self.source_args['simple_number_of_points'],
))
self.assertEqual(new_source.image_height_in_cm, self.source_args['image_height_in_cm'])
self.assertEqual(new_source.image_annotation_area, AnnotationAreaUtils.percentages_to_db_format(
min_x=self.source_args['min_x'], max_x=self.source_args['max_x'],
min_y=self.source_args['min_y'], max_y=self.source_args['max_y'],
))
self.assertEqual(new_source.longitude, self.source_args['longitude'])
self.assertEqual(new_source.latitude, self.source_args['latitude'])
self.assertEqual(new_source.enable_robot_classifier, True)
# This check is of limited use since database datetimes (in
# MySQL 5.1 at least) get truncated to whole seconds. But it still
# doesn't hurt to check.
self.assertTrue(datetime_before_creation <= new_source.create_date)
self.assertTrue(new_source.create_date <= datetime.datetime.now().replace(microsecond=0))
def test_post_success(self):
"""
Successful edit of an existing source.
"""
original_source = Source.objects.get(pk=self.source_id)
original_create_date = original_source.create_date
original_enable_robot = original_source.enable_robot_classifier
response = self.client.post(reverse('source_edit', kwargs={'source_id': self.source_id}),
self.source_args,
)
self.assertRedirects(response, reverse('source_main',
kwargs={'source_id': self.source_id}
))
edited_source = Source.objects.get(pk=self.source_id)
self.assertEqual(edited_source.name, self.source_args['name'])
self.assertEqual(edited_source.visibility, self.source_args['visibility'])
self.assertEqual(edited_source.create_date, original_create_date)
self.assertEqual(edited_source.labelset, LabelSet.getEmptyLabelset())
self.assertEqual(edited_source.key1, self.source_args['key1'])
self.assertEqual(edited_source.key2, '')
self.assertEqual(edited_source.key3, '')
self.assertEqual(edited_source.key4, '')
self.assertEqual(edited_source.key5, '')
self.assertEqual(edited_source.default_point_generation_method, PointGen.args_to_db_format(
point_generation_type=self.source_args['point_generation_type'],
simple_number_of_points=self.source_args['simple_number_of_points'],
))
self.assertEqual(edited_source.image_height_in_cm, self.source_args['image_height_in_cm'])
self.assertEqual(edited_source.image_annotation_area, AnnotationAreaUtils.percentages_to_db_format(
min_x=self.source_args['min_x'], max_x=self.source_args['max_x'],
min_y=self.source_args['min_y'], max_y=self.source_args['max_y'],
))
self.assertEqual(edited_source.longitude, self.source_args['longitude'])
self.assertEqual(edited_source.latitude, self.source_args['latitude'])
self.assertEqual(edited_source.enable_robot_classifier, original_enable_robot)
def source_new(request):
"""
Page with the form to create a new Source.
"""
# We can get here one of two ways: either we just got to the form
# page, or we just submitted the form. If POST, we submitted; if
# GET, we just got here.
if request.method == 'POST':
# Bind the forms to the submitted POST data.
sourceForm = ImageSourceForm(request.POST)
location_key_form = LocationKeyForm(request.POST)
pointGenForm = PointGenForm(request.POST)
annotationAreaForm = AnnotationAreaPercentsForm(request.POST)
# <form>.is_valid() calls <form>.clean() and checks field validity.
# Make sure is_valid() is called for all forms, so all forms are checked and
# all relevant error messages appear.
source_form_is_valid = sourceForm.is_valid()
location_key_form_is_valid = location_key_form.is_valid()
point_gen_form_is_valid = pointGenForm.is_valid()
annotation_area_form_is_valid = annotationAreaForm.is_valid()
if source_form_is_valid and location_key_form_is_valid \
and point_gen_form_is_valid and annotation_area_form_is_valid:
# Since sourceForm is a ModelForm, after calling sourceForm's
# is_valid(), a Source instance is created. We retrieve this
# instance and add the other values to it before saving to the DB.
newSource = sourceForm.instance
for key_field in ['key1', 'key2', 'key3', 'key4', 'key5']:
if key_field in location_key_form.cleaned_data:
setattr(newSource, key_field, location_key_form.cleaned_data[key_field])
newSource.default_point_generation_method = PointGen.args_to_db_format(**pointGenForm.cleaned_data)
newSource.image_annotation_area = AnnotationAreaUtils.percentages_to_db_format(**annotationAreaForm.cleaned_data)
newSource.labelset = LabelSet.getEmptyLabelset()
newSource.save()
# Make the current user an admin of the new source
newSource.assign_role(request.user, Source.PermTypes.ADMIN.code)
# Add a success message
messages.success(request, 'Source successfully created.')
# Redirect to the source's main page
return HttpResponseRedirect(reverse('source_main', args=[newSource.id]))
else:
# Show the form again, with error message
messages.error(request, 'Please correct the errors below.')
else:
# Unbound (empty) forms
sourceForm = ImageSourceForm()
location_key_form = LocationKeyForm()
pointGenForm = PointGenForm()
annotationAreaForm = AnnotationAreaPercentsForm()
# RequestContext is needed for CSRF verification of the POST form,
# and to correctly get the path of the CSS file being used.
return render_to_response('images/source_new.html', {
'sourceForm': sourceForm,
'location_key_form': location_key_form,
'pointGenForm': pointGenForm,
'annotationAreaForm': annotationAreaForm,
},
context_instance=RequestContext(request)
)
