diff --git a/rdmo/projects/viewsets.py b/rdmo/projects/viewsets.py index efd28c1b2..0afd3c815 100644 --- a/rdmo/projects/viewsets.py +++ b/rdmo/projects/viewsets.py @@ -334,6 +334,11 @@ def visibility(self, request, pk=None): throttle_classes=[EmailThrottle]) def contact(self, request, pk): if settings.PROJECT_CONTACT: + try: + project = self.get_object() + except Http404: + return Response(status=status.HTTP_404_NOT_FOUND) + if request.method == 'POST': subject = request.data.get('subject') message = request.data.get('message') @@ -347,11 +352,10 @@ def contact(self, request, pk): 'message': [_('This field may not be blank.')] if not message else [] }) else: - project = self.get_object() project.catalog.prefetch_elements() return Response(get_contact_message(request, project)) else: - return 404 + return Response(status=status.HTTP_404_NOT_FOUND) @action(detail=False, url_path='upload-accept', permission_classes=(IsAuthenticated, )) def upload_accept(self, request):