Personal/realms-wiki
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

define tokegetter

Browse source
This commit is contained in:
Matthew Scragg 2015-11-13 16:55:42 -06:00
parent 6edaec3876
commit 7178a6190e
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
realms/modules/auth/oauth/models.py
View file

@ -1,3 +1,4 @@
from flask import session
from flask_login import login_user from flask_login import login_user
from flask_oauthlib.client import OAuth from flask_oauthlib.client import OAuth
@ -81,6 +82,9 @@ providers = {
} }
} }
@oauth.tokengetter
def get_token(provider):
return session.get(provider + "_token")
class User(BaseUser): class User(BaseUser):
type = 'oauth' type = 'oauth'
@ -109,6 +113,7 @@ class User(BaseUser):
@staticmethod @staticmethod
def auth(provider, data, resp): def auth(provider, data, resp):
oauth_token = resp.get(User.get_provider_value(provider, 'token_name')) oauth_token = resp.get(User.get_provider_value(provider, 'token_name'))
session[provider + "_token"] = (oauth_token, '')
field_map = providers.get(provider).get('field_map') field_map = providers.get(provider).get('field_map')
if not field_map: if not field_map:
raise NotImplementedError raise NotImplementedError