Refactor error handling in AIPromptViewSet to use http_status for consistency

This commit is contained in:
IGNY8 VPS (Salman)
2025-11-10 12:04:40 +00:00
parent 389ac24230
commit 26d3d0daec

View File

@@ -56,9 +56,9 @@ class AIPromptViewSet(AccountModelViewSet):
prompt_value = request.data.get('prompt_value')
if not prompt_type:
return Response({'error': 'prompt_type is required'}, status=status.HTTP_400_BAD_REQUEST)
return Response({'error': 'prompt_type is required'}, status=http_status.HTTP_400_BAD_REQUEST)
if prompt_value is None:
return Response({'error': 'prompt_value is required'}, status=status.HTTP_400_BAD_REQUEST)
return Response({'error': 'prompt_value is required'}, status=http_status.HTTP_400_BAD_REQUEST)
# Get account - try multiple methods
account = getattr(request, 'account', None)
@@ -78,7 +78,7 @@ class AIPromptViewSet(AccountModelViewSet):
pass
if not account:
return Response({'error': 'Account not found. Please ensure you are logged in.'}, status=status.HTTP_400_BAD_REQUEST)
return Response({'error': 'Account not found. Please ensure you are logged in.'}, status=http_status.HTTP_400_BAD_REQUEST)
# Get default prompt value if creating new
from .utils import get_default_prompt
@@ -112,7 +112,7 @@ class AIPromptViewSet(AccountModelViewSet):
prompt_type = request.data.get('prompt_type')
if not prompt_type:
return Response({'error': 'prompt_type is required'}, status=status.HTTP_400_BAD_REQUEST)
return Response({'error': 'prompt_type is required'}, status=http_status.HTTP_400_BAD_REQUEST)
# Get account - try multiple methods (same as integration_views)
account = getattr(request, 'account', None)
@@ -132,7 +132,7 @@ class AIPromptViewSet(AccountModelViewSet):
pass
if not account:
return Response({'error': 'Account not found. Please ensure you are logged in.'}, status=status.HTTP_400_BAD_REQUEST)
return Response({'error': 'Account not found. Please ensure you are logged in.'}, status=http_status.HTTP_400_BAD_REQUEST)
# Get default prompt
from .utils import get_default_prompt