diff --git a/arkindex/documents/api/elements.py b/arkindex/documents/api/elements.py index 3ba5d1954d1d02186db801b0288b8c6f84c262b5..329e450aedcb8c62f7560ff95d5f57b7ea66c09f 100644 --- a/arkindex/documents/api/elements.py +++ b/arkindex/documents/api/elements.py @@ -103,10 +103,10 @@ class ElementsList(CorpusACLMixin, ListAPIView): } }, { - 'name': 'best_classes', + 'name': 'with_best_classes', 'in': 'query', 'description': ( - 'List best classifications for each element. ' + 'Returns best classifications for each element. ' 'If not set, elements best_classes field will always be null' ), 'required': False, @@ -158,8 +158,8 @@ class ElementsList(CorpusACLMixin, ListAPIView): .exclude(state=ClassificationState.Rejected) ).distinct() - best_classes = self.request.query_params.get('best_classes') - if best_classes and best_classes.lower() not in ('false', '0'): + with_best_classes = self.request.query_params.get('with_best_classes') + if with_best_classes and with_best_classes.lower() not in ('false', '0'): filtered_queryset = filtered_queryset.prefetch_related(best_classifications_prefetch) # ID is required by postgres to order elements with common corpus, type and name @@ -360,10 +360,10 @@ class ElementParents(ListAPIView): } }, { - 'name': 'best_classes', + 'name': 'with_best_classes', 'in': 'query', 'description': ( - 'List best classifications for each element. ' + 'Returns best classifications for each element. ' 'If not set, elements best_classes field will always be null' ), 'required': False, @@ -398,8 +398,8 @@ class ElementParents(ListAPIView): prefetch_related_lookups = ('zone__image__server', 'corpus', 'type') - best_classes = self.request.query_params.get('best_classes') - if best_classes and best_classes.lower() not in ('false', '0'): + with_best_classes = self.request.query_params.get('with_best_classes') + if with_best_classes and with_best_classes.lower() not in ('false', '0'): prefetch_related_lookups += (best_classifications_prefetch, ) class_filter = self.request.query_params.get('best_class') @@ -496,10 +496,10 @@ class ElementChildren(ListAPIView): } }, { - 'name': 'best_classes', + 'name': 'with_best_classes', 'in': 'query', 'description': ( - 'List best classifications for each element. ' + 'Returns best classifications for each element. ' 'If not set, elements best_classes field will always be null' ), 'required': False, @@ -534,8 +534,8 @@ class ElementChildren(ListAPIView): prefetch_related_lookups = ('zone__image__server', 'corpus', 'type') - best_classes = self.request.query_params.get('best_classes') - if best_classes and best_classes.lower() not in ('false', '0'): + with_best_classes = self.request.query_params.get('with_best_classes') + if with_best_classes and with_best_classes.lower() not in ('false', '0'): prefetch_related_lookups += (best_classifications_prefetch, ) if recursive_param is None or recursive_param.lower() in ('false', '0'): @@ -575,10 +575,10 @@ class ManageSelection(ListAPIView): 'tags': ['elements'], 'parameters': [ { - 'name': 'best_classes', + 'name': 'with_best_classes', 'in': 'query', 'description': ( - 'List best classifications for each element. ' + 'Returns best classifications for each element. ' 'If not set, elements best_classes field will always be null' ), 'required': False, @@ -598,8 +598,8 @@ class ManageSelection(ListAPIView): elements = self.request.session.get('elements_selected') filtered_queryset = Element.objects.filter(id__in=elements).order_by('corpus', 'type', 'name') - best_classes = self.request.query_params.get('best_classes') - if best_classes and best_classes.lower() not in ('false', '0'): + with_best_classes = self.request.query_params.get('with_best_classes') + if with_best_classes and with_best_classes.lower() not in ('false', '0'): return filtered_queryset.prefetch_related(best_classifications_prefetch) return filtered_queryset diff --git a/arkindex/documents/tests/test_classes.py b/arkindex/documents/tests/test_classes.py index e4b8d3d3b22d5314b1719161e757eabcc654c681..aa5f685c636d25c1e92318647c6876669de06669 100644 --- a/arkindex/documents/tests/test_classes.py +++ b/arkindex/documents/tests/test_classes.py @@ -160,7 +160,7 @@ class TestClasses(FixtureAPITestCase): with self.assertNumQueries(6): response = self.client.get( reverse('api:elements'), - data={'type': self.classified.slug, 'best_classes': True} + data={'type': self.classified.slug, 'with_best_classes': True} ) self.assertEqual(response.status_code, status.HTTP_200_OK) data = response.json() @@ -178,7 +178,7 @@ class TestClasses(FixtureAPITestCase): reverse('api:elements'), data={ 'type': self.classified.slug, - 'best_classes': False, + 'with_best_classes': False, } ) self.assertEqual(response.status_code, status.HTTP_200_OK) @@ -190,7 +190,7 @@ class TestClasses(FixtureAPITestCase): with self.assertNumQueries(6): response = self.client.get( reverse('api:elements-parents', kwargs={'pk': str(self.common_children.id)}), - data={'type': self.classified.slug, 'best_classes': 1} + data={'type': self.classified.slug, 'with_best_classes': 1} ) self.assertEqual(response.status_code, status.HTTP_200_OK) data = response.json() @@ -206,7 +206,7 @@ class TestClasses(FixtureAPITestCase): with self.assertNumQueries(6): response = self.client.get( reverse('api:elements-children', kwargs={'pk': str(self.parent.id)}), - data={'type': self.classified.slug, 'best_classes': 'yes'} + data={'type': self.classified.slug, 'with_best_classes': 'yes'} ) self.assertEqual(response.status_code, status.HTTP_200_OK) data = response.json() @@ -227,7 +227,7 @@ class TestClasses(FixtureAPITestCase): with self.assertNumQueries(6): response = self.client.get( reverse('api:elements-children', kwargs={'pk': str(self.parent.id)}), - data={'type': self.classified.slug, 'best_classes': 'yes'} + data={'type': self.classified.slug, 'with_best_classes': 'yes'} ) self.assertEqual(response.status_code, status.HTTP_200_OK) data = response.json() @@ -247,7 +247,7 @@ class TestClasses(FixtureAPITestCase): parent.classifications.all().filter(confidence=.7).update(state=ClassificationState.Validated) response = self.client.get( reverse('api:elements-parents', kwargs={'pk': str(self.common_children.id)}), - data={'type': self.classified.slug, 'best_classes': 'yes'} + data={'type': self.classified.slug, 'with_best_classes': 'yes'} ) for elt in response.json()['results']: if elt['id'] == str(parent.id): @@ -276,7 +276,7 @@ class TestClasses(FixtureAPITestCase): ) response = self.client.get( reverse('api:elements-parents', kwargs={'pk': str(self.common_children.id)}), - data={'type': self.classified.slug, 'best_classes': 1} + data={'type': self.classified.slug, 'with_best_classes': 1} ) self.assertEqual(response.status_code, status.HTTP_200_OK) for elt in response.json()['results']: @@ -295,7 +295,7 @@ class TestClasses(FixtureAPITestCase): classif.save() response = self.client.get( reverse('api:elements-parents', kwargs={'pk': str(self.common_children.id)}), - data={'type': self.classified.slug, 'best_classes': 1} + data={'type': self.classified.slug, 'with_best_classes': 1} ) self.assertEqual(response.status_code, status.HTTP_200_OK) for elt in response.json()['results']: diff --git a/arkindex/documents/tests/test_elements_api.py b/arkindex/documents/tests/test_elements_api.py index 0758fd36b90d5c6b6ed0db27609ef414a0233734..9a1584559a4931288a8a8ccc2cd46eee24f9fa7a 100644 --- a/arkindex/documents/tests/test_elements_api.py +++ b/arkindex/documents/tests/test_elements_api.py @@ -951,7 +951,7 @@ class TestElementsAPI(FixtureAPITestCase): self.assertEqual(results[0]['id'], str(self.page.id)) self.assertEqual(results[1]['id'], str(self.vol.id)) - response = self.client.get(reverse('api:elements-selection') + '?best_classes=true') + response = self.client.get(reverse('api:elements-selection') + '?with_best_classes=true') self.assertEqual(response.status_code, status.HTTP_200_OK) results = response.json()['results'] self.assertEqual(len(results), 2)