Handle case of no in progress records

NASA-IMPACT · Oct 12, 2023 · 81e86ec · 81e86ec
1 parent 0953f8b
commit 81e86ec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/admin_ui/views/v2.py b/app/admin_ui/views/v2.py
@@ -225,12 +225,17 @@ def get_object(self, queryset=None):
             queryset = self.queryset
 
         canonical_uuid = self.kwargs["canonical_uuid"]
-        return (
+        obj = (
             queryset.filter(Q(uuid=canonical_uuid) | Q(model_instance_uuid=canonical_uuid))
             .order_by('status')
             .first()
         )
 
+        if not obj:
+            raise Http404(f'No in progress draft with Canonical UUID {canonical_uuid!r}')
+
+        return obj
+
     def get_success_url(self, **kwargs):
         url = reverse(
             "canonical-redirect",