diff --git a/openid_provider/lib/endpoints/token.py b/openid_provider/lib/endpoints/token.py index 4718514..6f8fbfd 100644 --- a/openid_provider/lib/endpoints/token.py +++ b/openid_provider/lib/endpoints/token.py @@ -84,7 +84,7 @@ class TokenEndpoint(object): return dic @classmethod - def response(self, dic, status=200): + def response(cls, dic, status=200): """ Create and return a response object. """ diff --git a/openid_provider/lib/endpoints/userinfo.py b/openid_provider/lib/endpoints/userinfo.py index 84b56ab..a4b379e 100644 --- a/openid_provider/lib/endpoints/userinfo.py +++ b/openid_provider/lib/endpoints/userinfo.py @@ -64,7 +64,7 @@ class UserInfoEndpoint(object): return dic @classmethod - def response(self, dic): + def response(cls, dic): response = JsonResponse(dic, status=200) response['Cache-Control'] = 'no-store' @@ -73,7 +73,7 @@ class UserInfoEndpoint(object): return response @classmethod - def error_response(self, code, description, status): + def error_response(cls, code, description, status): response = HttpResponse(status=status) response['WWW-Authenticate'] = 'error="{0}", error_description="{1}"'.format(code, description)