diff --git a/doc.md b/doc.md index d837e61..795082f 100644 --- a/doc.md +++ b/doc.md @@ -99,12 +99,11 @@ from oidc_provider.lib.claims import AbstractScopeClaims class MyAppScopeClaims(AbstractScopeClaims): - def __init__(self, user, scopes): - # Don't forget this. - super(StandardScopeClaims, self).__init__(user, scopes) - + def setup(self): # Here you can load models that will be used # in more than one scope for example. + # print self.user + # print self.scopes try: self.some_model = SomeModel.objects.get(user=self.user) except UserInfo.DoesNotExist: diff --git a/oidc_provider/lib/claims.py b/oidc_provider/lib/claims.py index 3c59bdc..e9d855a 100644 --- a/oidc_provider/lib/claims.py +++ b/oidc_provider/lib/claims.py @@ -8,6 +8,11 @@ class AbstractScopeClaims(object): self.user = user self.scopes = scopes + self.setup() + + def setup(self): + pass + def create_response_dic(self): """ Generate the dic that will be jsonify. Checking scopes given vs @@ -61,9 +66,7 @@ class StandardScopeClaims(AbstractScopeClaims): See: http://openid.net/specs/openid-connect-core-1_0.html#StandardClaims """ - def __init__(self, user, scopes): - super(StandardScopeClaims, self).__init__(user, scopes) - + def setup(self): try: self.userinfo = UserInfo.objects.get(user=self.user) except UserInfo.DoesNotExist: