visual limit for 400 page results

PabloCastellano · PabloCastellano · commit 0ad1001ceaec · 2018-05-06T23:17:04.000+02:00
diff --git a/borme/views.py b/borme/views.py
@@ -185,7 +185,12 @@ def get_context_data(self, **kwargs):
                     context['page_companies'] = pg_companies
                     context['results_companies'] = list(map(lambda x: x['_source'], pg_companies))
                     context['count_companies'] = q_companies.count()
-                    pagerange = list(companies.page_range[:3]) + list(companies.page_range[-3:])
+                    pagerange = list(companies.page_range[:3])
+                    last_page = min(companies.page_range.stop - 1, 400)
+                    if last_page == 400:
+                        pagerange += [388, 389, 400]
+                    else:
+                        pagerange += list(companies.page_range[-3:])
                     pagerange.append(pg_companies.number)
                     pagerange = list(set(pagerange))
                     pagerange.sort()
@@ -203,7 +208,12 @@ def get_context_data(self, **kwargs):
                     context['page_persons'] = pg_persons
                     context['results_persons'] = list(map(lambda x: x['_source'], pg_persons))
                     context['count_persons'] = q_persons.count()
-                    pagerange = list(persons.page_range[:3]) + list(persons.page_range[-3:])
+                    pagerange = list(persons.page_range[:3])
+                    last_page = min(persons.page_range.stop - 1, 400)
+                    if last_page == 400:
+                        pagerange += [388, 389, 400]
+                    else:
+                        pagerange += list(persons.page_range[-3:])
                     pagerange.append(pg_persons.number)
                     pagerange = list(set(pagerange))
                     pagerange.sort()
