Merge pull request #2194 from IFRCGo/feature/fix-sentry-issue-2005

Fix ValueError issue for Country key
IFRCGo · Jul 3, 2024 · b0f3beb · b0f3beb
2 parents 1d43e98 + d2b3b22
commit b0f3beb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/api/drf_views.py b/api/drf_views.py
@@ -223,8 +223,11 @@ def get_object(self):
         # NOTE: kwargs is accepting pk for now
         # TODO: Can kwargs be other than pk??
         pk = self.kwargs["pk"]
-        country = get_object_or_404(Country, pk=int(pk))
-        return self.get_queryset().filter(id=country.id).first()
+        try:
+            country = get_object_or_404(Country, pk=int(pk))
+            return self.get_queryset().filter(id=country.id).first()
+        except ValueError:
+            raise Exception("An error occured", "Country key is unusable", pk)
 
     def get_serializer_class(self):
         if self.request.GET.get("mini", "false").lower() == "true":