Fix /uprn lookup broken by previous commit

mysociety · Apr 16, 2024 · 8774d9e · 8774d9e
1 parent cf62a83
commit 8774d9e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/mapit_labour/tests/test_views.py b/mapit_labour/tests/test_views.py
@@ -14,6 +14,8 @@
 
 
 class LoginRequiredTestCase(LoadTestData, TestCase):
+    fixtures = ["uk", "test_areas"]
+
     def test_redirect_to_login(self):
         self.client.logout()
 

diff --git a/mapit_labour/views.py b/mapit_labour/views.py
@@ -62,11 +62,8 @@
 def uprn(request, uprn, format="json"):
     uprn = get_object_or_404(UPRN, format=format, uprn=uprn)
 
-    try:
-        generation = int(request.GET["generation"])
-    except:
-        generation = Generation.objects.current()
-    areas = list(add_codes(Area.objects.by_location(uprn.location, generation)))
+    query = Generation.objects.query_args(request, format)
+    areas = list(add_codes(Area.objects.by_location(uprn.location, query)))
 
     shortcuts = {}
     for area in areas: