fix: Health provider not accessible by name in API

Flagsmith · Feb 4, 2025 · f9dc15f · f9dc15f
1 parent d4c4aaa
commit f9dc15f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/api/features/feature_health/views.py b/api/features/feature_health/views.py
@@ -57,6 +57,7 @@ class FeatureHealthProviderViewSet(
     serializer_class = FeatureHealthProviderSerializer
     pagination_class = None  # set here to ensure documentation is correct
     model_class = FeatureHealthProvider
+    lookup_field = "name"
 
     def get_permissions(self) -> list[BasePermission]:
         return [NestedProjectPermissions()]
@@ -68,6 +69,13 @@ def get_queryset(self) -> QuerySet[FeatureHealthProvider]:
         project = get_object_or_404(Project, pk=self.kwargs["project_pk"])
         return self.model_class.objects.filter(project=project)
 
+    def get_object(self) -> FeatureHealthProvider:
+        return get_object_or_404(
+            self.model_class.objects,
+            project_id=self.kwargs["project_pk"],
+            name__iexact=self.kwargs["provider_name"],
+        )
+
     @swagger_auto_schema(
         request_body=CreateFeatureHealthProviderSerializer,
         responses={status.HTTP_201_CREATED: FeatureHealthProviderSerializer()},