CHANGED: map integrity error to 400

seebass · Aug 3, 2015 · ee55f43 · ee55f43
1 parent d1720b0
commit ee55f43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/drf_tools/exceptions.py b/drf_tools/exceptions.py
@@ -1,12 +1,12 @@
 import logging
 
 from django.core.exceptions import ValidationError, ObjectDoesNotExist, PermissionDenied
+from django.db import IntegrityError
 from django.http import Http404
 from rest_framework.exceptions import APIException
 from rest_framework import status
 from rest_framework.response import Response
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -19,7 +19,7 @@ def exception_handler(exc):
             headers['X-Throttle-Wait-Seconds'] = '%d' % exc.wait
             headers['Retry-After'] = '%d' % exc.wait
         status_code = exc.status_code
-    elif isinstance(exc, (ValueError, ValidationError)):
+    elif isinstance(exc, (ValueError, ValidationError, IntegrityError)):
         status_code = status.HTTP_400_BAD_REQUEST
     elif isinstance(exc, PermissionDenied):
         status_code = status.HTTP_403_FORBIDDEN

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='drf-tools',
-    version="0.9.15",
+    version="0.9.16",
     url='https://github.com/seebass/drf-tools',
     license='MIT',
     description='Multiple extensions and test utilities for Django REST Framework 3',