Fix destination after OAuth.

This commit is contained in:
Edward Betts 2019-09-27 20:15:24 +01:00
parent 7b1c14b1ce
commit d982f975cf

8
app.py
View file

@ -193,6 +193,11 @@ def random_painting():
@app.route('/oauth/start') @app.route('/oauth/start')
def start_oauth(): def start_oauth():
next_page = request.args.get('next')
if next_page:
session['after_login'] = next_page
client_key = app.config['CLIENT_KEY'] client_key = app.config['CLIENT_KEY']
client_secret = app.config['CLIENT_SECRET'] client_secret = app.config['CLIENT_SECRET']
base_url = 'https://www.wikidata.org/w/index.php' base_url = 'https://www.wikidata.org/w/index.php'
@ -235,7 +240,8 @@ def oauth_callback():
session['owner_key'] = oauth_tokens.get('oauth_token') session['owner_key'] = oauth_tokens.get('oauth_token')
session['owner_secret'] = oauth_tokens.get('oauth_token_secret') session['owner_secret'] = oauth_tokens.get('oauth_token_secret')
return redirect(url_for('browse_page')) next_page = session.get('next_page')
return redirect(next_page) if next_page else random_painting()
def get_username(): def get_username():
if 'owner_key' not in session: if 'owner_key' not in session: