Merge pull request #470 from kids-first/referral-token-exceptions

🐛 Fix referral token exceptions

creator/referral_tokens/mutations.py

@@ -1,8 +1,10 @@
 import graphene
+import uuid
 from datetime import timedelta
 from django.conf import settings
 from django.contrib.auth.models import Group
 from django.core.mail import send_mail
+from django.core.exceptions import ValidationError
 from django.template.loader import render_to_string
 from django.utils import timezone
 from graphql import GraphQLError
@@ -172,7 +174,7 @@ def mutate(self, info, token):
         try:
             _, uuid = from_global_id(token)
             referral_token = ReferralToken.objects.get(uuid=uuid)
-        except ReferralToken.DoesNotExist:
+        except (ReferralToken.DoesNotExist, ValidationError):
             raise GraphQLError("Referral token does not exist.")
 
         if not referral_token.is_valid:

creator/urls.py

@@ -23,8 +23,10 @@ def format_error(error):
             FileUploadGraphQLView, FileUploadGraphQLView
         ).format_error(error)
 
-        if hasattr(error, "original_error") and isinstance(
-            error.original_error, ValidationError
+        if (
+            hasattr(error, "original_error")
+            and isinstance(error.original_error, ValidationError)
+            and hasattr(error.original_error, "error_dict")
         ):
             error_dict = error.original_error.error_dict
             if "__all__" in error_dict: