realms-wiki/realms/modules/auth/oauth/models.py

126 lines
4.1 KiB
Python
Raw Normal View History

2015-10-16 01:36:47 +03:00
from flask_login import login_user
2015-10-14 06:52:30 +03:00
from flask_oauthlib.client import OAuth
2015-10-16 01:36:47 +03:00
2015-10-14 06:52:30 +03:00
from realms import config
from ..models import BaseUser
oauth = OAuth()
2015-10-15 07:08:56 +03:00
users = {}
2015-10-14 06:52:30 +03:00
2015-10-15 07:08:56 +03:00
providers = {
'twitter': {
'oauth': dict(
2015-10-16 01:36:47 +03:00
base_url='https://api.twitter.com/1.1/',
2015-10-15 01:36:22 +03:00
request_token_url='https://api.twitter.com/oauth/request_token',
access_token_url='https://api.twitter.com/oauth/access_token',
2015-10-16 01:36:47 +03:00
authorize_url='https://api.twitter.com/oauth/authenticate',
access_token_method='GET'),
'button': '<a href="/login/oauth/twitter" class="btn btn-default"><i class="fa fa-twitter"></i> Twitter</a>',
'field_map': {
'username': 'screen_name'
}
},
'github': {
'oauth': dict(
request_token_params={'scope': 'user:email'},
base_url='https://api.github.com/',
request_token_url=None,
access_token_method='POST',
access_token_url='https://github.com/login/oauth/access_token',
authorize_url='https://github.com/login/oauth/authorize'),
'button': '<a href="/login/oauth/github" class="btn btn-default"><i class="fa fa-github"></i> Github</a>'
},
'facebook': {
'oauth': dict(
request_token_params={'scope': 'email'},
base_url='https://graph.facebook.com',
request_token_url=None,
access_token_url='/oauth/access_token',
access_token_method='GET',
authorize_url='https://www.facebook.com/dialog/oauth'
),
'button': '<a href="/login/oauth/github" class="btn btn-default"><i class="fa fa-faceboook"></i> Facebook</a>'
},
'google': {
'oauth': dict(
request_token_params={
'scope': 'https://www.googleapis.com/auth/userinfo.email'
},
base_url='https://www.googleapis.com/oauth2/v1/',
request_token_url=None,
access_token_method='POST',
access_token_url='https://accounts.google.com/o/oauth2/token',
authorize_url='https://accounts.google.com/o/oauth2/auth',
),
'button': '<a href="/login/oauth/google" class="btn btn-default"><i class="fa fa-google"></i> Google</a>'
2015-10-15 07:08:56 +03:00
}
}
2015-10-15 01:36:22 +03:00
2015-10-15 07:08:56 +03:00
class User(BaseUser):
2015-10-16 01:36:47 +03:00
type = 'oauth'
provider = None
def __init__(self, provider, username, token):
self.provider = provider
self.username = username
self.id = username
self.token = token
self.auth_id = "%s-%s" % (provider, username)
2015-10-16 01:36:47 +03:00
@property
def auth_token_id(self):
return self.token
@staticmethod
def load_user(*args, **kwargs):
return User.get_by_id(args[0])
@staticmethod
def get_by_id(user_id):
return users.get(user_id)
@staticmethod
def auth(provider, resp):
field_map = providers.get(provider).get('field_map')
if not field_map:
raise NotImplementedError
fields = {}
for k, v in field_map.items():
fields[k] = resp[v]
user = User(provider, fields['username'], User.hash_password(resp['oauth_token']))
users[user.auth_id] = user
2015-10-16 01:36:47 +03:00
if user:
login_user(user, remember=True)
return True
else:
return False
2015-10-15 07:08:56 +03:00
@classmethod
def get_app(cls, provider):
2015-10-16 01:36:47 +03:00
if oauth.remote_apps.get(provider):
return oauth.remote_apps.get(provider)
return oauth.remote_app(
provider,
consumer_key=config.OAUTH.get(provider, {}).get('key'),
consumer_secret=config.OAUTH.get(provider, {}).get(
'secret'),
**providers[provider]['oauth'])
2015-10-16 01:36:47 +03:00
def get_id(self):
return unicode("%s/%s" % (self.type, self.auth_id))
2015-10-14 06:52:30 +03:00
@staticmethod
def login_form():
buttons = []
for name, val in providers.items():
if not config.OAUTH.get(name, {}).get('key') or not config.OAUTH.get(name, {}).get('secret'):
continue
buttons.append(val.get('button'))
2015-10-16 01:36:47 +03:00
return "<h4>Social Login</h4>" + " ".join(buttons)