Update to use class comparison instead of string comparison

fecgov · Mar 6, 2024 · 03eb3b4 · 03eb3b4
1 parent d4f10e6
commit 03eb3b4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/django-backend/fecfiler/utils.py b/django-backend/fecfiler/utils.py
@@ -1,3 +1,4 @@
+from rest_framework.exceptions import ValidationError
 from django.http import HttpResponseServerError
 from fecfiler.authentication.views import delete_user_logged_in_cookies
 from rest_framework.views import exception_handler
@@ -25,8 +26,8 @@ def custom_exception_handler(exc, context):
     # Do not allow an error response body unless validation
     data = getattr(response, 'data')
     exception_type = type(exc)
-    logger.error("Error: {}".format(data))
-    if data and "ValidationError" not in str(exception_type):
+    logger.error(f"Error: {data}")
+    if data and exception_type is not ValidationError:
         response.data = None
 
     return response