From 51594687e2c474beb6dd06f93fe3d2f336937694 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Fri, 9 Mar 2018 18:49:49 +1100
Subject: [PATCH] Use blueprints for apps

---
 eosweb/core/main.py       |  3 +--
 eosweb/redditauth/main.py | 57 +++++++++++++++++++++++----------------
 2 files changed, 35 insertions(+), 25 deletions(-)

diff --git a/eosweb/core/main.py b/eosweb/core/main.py
index 48f76fe..fd5e261 100644
--- a/eosweb/core/main.py
+++ b/eosweb/core/main.py
@@ -402,7 +402,6 @@ def logout():
 
 @app.route('/auth/login_callback')
 def login_callback():
-	print(flask.session)
 	if 'login_next' in flask.session and flask.session['login_next']:
 		return flask.redirect(flask.session['login_next'])
 	else:
@@ -443,7 +442,7 @@ def email_authenticate():
 
 for app_name in app.config['APPS']:
 	app_main = importlib.import_module(app_name + '.main')
-	app_main.main(app)
+	app.register_blueprint(app_main.blueprint)
 
 # === Model-Views ===
 
diff --git a/eosweb/redditauth/main.py b/eosweb/redditauth/main.py
index a2ec860..0d282f5 100644
--- a/eosweb/redditauth/main.py
+++ b/eosweb/redditauth/main.py
@@ -1,5 +1,5 @@
 #   Eos - Verifiable elections
-#   Copyright © 2017  RunasSudo (Yingtong Li)
+#   Copyright © 2017-18  RunasSudo (Yingtong Li)
 #
 #   This program is free software: you can redistribute it and/or modify
 #   it under the terms of the GNU Affero General Public License as published by
@@ -23,7 +23,18 @@ from eos.redditauth.election import *
 import base64
 import uuid
 
-def main(app):
+blueprint = flask.Blueprint('eosweb.redditauth', __name__)
+
+app = None
+oauth = None
+reddit = None
+
+@blueprint.record
+def reddit_register(setup_state):
+	global app, oauth, reddit
+	
+	app = setup_state.app
+	
 	oauth = OAuth()
 	reddit = oauth.remote_app('Reddit',
 		request_token_url=None,
@@ -39,28 +50,28 @@ def main(app):
 		consumer_secret=app.config['REDDIT_OAUTH_CLIENT_SECRET']
 	)
 	
-	@app.route('/auth/reddit/login')
-	def reddit_login():
-		return reddit.authorize(callback=app.config['BASE_URI'] + flask.url_for('reddit_oauth_authorized'), state=uuid.uuid4())
-	
 	@reddit.tokengetter
 	def get_reddit_oauth_token():
 		return (flask.session.get('user').oauth_token, '')
+
+@blueprint.route('/auth/reddit/login')
+def reddit_login():
+	return reddit.authorize(callback=app.config['BASE_URI'] + flask.url_for('eosweb.redditauth.reddit_oauth_authorized'), state=uuid.uuid4())
+
+@blueprint.route('/auth/reddit/oauth_callback')
+def reddit_oauth_authorized():
+	resp = reddit.authorized_response()
+	if resp is None:
+		# Request denied
+		return flask.redirect(flask.url_for('login_cancelled'))
 	
-	@app.route('/auth/reddit/oauth_callback')
-	def reddit_oauth_authorized():
-		resp = reddit.authorized_response()
-		if resp is None:
-			# Request denied
-			return flask.redirect(flask.url_for('login_cancelled'))
-		
-		user = RedditUser()
-		user.oauth_token = resp['access_token']
-		flask.session['user'] = user
-		
-		me = reddit.get('https://oauth.reddit.com/api/v1/me', headers={
-			'User-Agent': app.config['REDDIT_USER_AGENT']
-		})
-		user.username = me.data['name']
-		
-		return flask.redirect(flask.url_for('login_complete'))
+	user = RedditUser()
+	user.oauth_token = resp['access_token']
+	flask.session['user'] = user
+	
+	me = reddit.get('https://oauth.reddit.com/api/v1/me', headers={
+		'User-Agent': app.config['REDDIT_USER_AGENT']
+	})
+	user.username = me.data['name']
+	
+	return flask.redirect(flask.url_for('login_complete'))