From e3b5af54903ff2b1f150706068ea58d0b6077949 Mon Sep 17 00:00:00 2001
From: juanifioren <juanifioren@gmail.com>
Date: Fri, 6 Mar 2015 12:54:27 -0300
Subject: [PATCH] Add setup to extra scope claims.

---
 doc.md                      | 7 +++----
 oidc_provider/lib/claims.py | 9 ++++++---
 2 files changed, 9 insertions(+), 7 deletions(-)

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: