fix: handle APIException when rest_framework is not installed

Radi85 · Sep 12, 2020 · 31423c6 · 31423c6
1 parent ac5e2ef
commit 31423c6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/comment/exceptions.py b/comment/exceptions.py
@@ -1,9 +1,11 @@
-from rest_framework.exceptions import APIException
-from rest_framework.status import HTTP_400_BAD_REQUEST
+try:
+    from rest_framework.exceptions import APIException
+except ModuleNotFoundError:
+    APIException = Exception
 
 
 class CommentBadRequest(APIException):
-    status_code = HTTP_400_BAD_REQUEST
+    status_code = 400
     default_detail = 'Bad Request'
 
     def __init__(self, detail=None, status_code=None):