Skip to content

Commit

Permalink
Added oAuth generic implementation
Browse files Browse the repository at this point in the history
  • Loading branch information
jaykay12 committed May 9, 2020
1 parent d8f7720 commit 6ccb225
Show file tree
Hide file tree
Showing 2 changed files with 77 additions and 2 deletions.
7 changes: 5 additions & 2 deletions service/api/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,16 +21,19 @@ def create_app():
accesslogger.info("Loaded: Configuration of Stage")
elif environ.get('ENV') == 'TESTING':
app.config.from_object('config.TestConfig')
app.config.from_object('config.oAuthConfig')
accesslogger.info("Loaded: Configuration of Testing")
else:
app.config.from_object('config.DevelopmentConfig')
app.config.from_object('config.oAuthConfig')
accesslogger.info("Loaded: configuration of Development")

db.init_app(app)

with app.app_context():
from . import routes # Import routes
db.create_all() # Create database tables for our data models
from . import routes_auth
from . import routes_oauth
db.create_all()

return app

Expand Down
72 changes: 72 additions & 0 deletions service/api/oauth.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
import json

from rauth import OAuth2Service
from flask import current_app, url_for, request, redirect, session

class OAuthSignIn(object):
providers = None

def __init__(self, provider_name):
self.provider_name = provider_name
credential = current_app.config['OAUTH_CREDENTIALS'][provider_name]
self.consumer_id = credential['id']
self.consumer_secret = credential['secret']

def authorize(self):
pass

def callback(self):
pass

def get_callback_url(self):
return url_for('oauth_callback', provider=self.provider_name, _external=True)

@classmethod
def load_provider(self):
if self.providers is None:
self.providers = {}
for provider_class in self.__subclasses__():
provider = provider_class()
self.providers[provider.provider_name] = provider

@classmethod
def get_provider(self, provider_name):
self.load_provider()
return self.providers[provider_name]

class FacebookSignIn(OAuthSignIn):
def __init__(self):
super(FacebookSignIn, self).__init__('facebook')
self.service = OAuth2Service(
name = 'facebook',
client_id = self.consumer_id,
client_secret = self.consumer_secret,
authorize_url = 'https://graph.facebook.com/oauth/authorize',
access_token_url = 'https://graph.facebook.com/oauth/access_token',
base_url = 'https://graph.facebook.com/'
)

def authorize(self):
return redirect(self.service.get_authorize_url(
scope = 'email',
response_type = 'code',
redirect_uri = self.get_callback_url()
))

def callback(self):
def decode_json(payload):
return json.loads(payload.decode('utf-8'))

if 'code' not in request.args:
return None,None,None
oauth_session = self.service.get_auth_session(
data = {'code': request.args['code'],
'grant_type': 'authorization_code',
'redirect_uri': self.get_callback_url()},
decoder = decode_json
)

me = oauth_session.get('me?fields=id,email').json()
print(me)

return ('facebook$' + me['id'], me.get('email').split('@')[0], me.get('email'))

0 comments on commit 6ccb225

Please sign in to comment.