diff --git a/identity/django.py b/identity/django.py index 019c3b9..80f1d2b 100644 --- a/identity/django.py +++ b/identity/django.py @@ -170,11 +170,13 @@ def wrapper(request, *args, **kwargs): if context: try: return function(request, *args, context=context, **kwargs) - except TypeError: - raise RuntimeError( - "Since identity 0.6.0, the '@login_required(...)' decorated " - "view should accept a keyword argument named 'context'. " - "For example, def my_view(request, *, context): ...") from None + except TypeError as exp: + if "context" in str(exp): + raise RuntimeError( + "Since identity 0.6.0, the '@login_required(...)' decorated " + "view should accept a keyword argument named 'context'. " + "For example, def my_view(request, *, context): ...") from exp + raise # Save an http 302 by calling self.login(request) instead of redirect(self.login) return self.login( request,