Clarify that page is an iterable

archesproject · Sep 5, 2024 · 89ef29f · 89ef29f
1 parent 49866ab
commit 89ef29f
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/arches_lingo/views/api/concepts.py b/arches_lingo/views/api/concepts.py
@@ -58,17 +58,13 @@ def get(self, request):
         concept_ids = concept_query.values_list("concept_id", flat=True).distinct()
 
         paginator = Paginator(concept_ids, items_per_page)
-        page = paginator.get_page(page_number)
-
-        data = []
-        if not page:
-            # No results: don't bother building the concept tree.
-            return JSONResponse(data)
+        if not paginator.count:
+            return JSONResponse([])
 
         builder = ConceptBuilder()
         data = [
             builder.serialize_concept(str(concept_uuid), parents=True, children=False)
-            for concept_uuid in page
+            for concept_uuid in paginator.get_page(page_number)
         ]
 
         return JSONResponse(data)