From 7a04d64f9cd06145225c0c28623949bf717a1455 Mon Sep 17 00:00:00 2001 From: TheAMM Date: Sat, 5 Aug 2017 22:42:21 +0300 Subject: [PATCH] Weed out last few hardcoded urls --- nyaa/__init__.py | 2 +- nyaa/views/account.py | 9 ++++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/nyaa/__init__.py b/nyaa/__init__.py index 492c29b..47ccd22 100644 --- a/nyaa/__init__.py +++ b/nyaa/__init__.py @@ -49,7 +49,7 @@ def create_app(config): app.logger.error(exception) flask.flash(flask.Markup( 'An error occurred! Debug information has been logged.'), 'danger') - return flask.redirect('/') + return flask.redirect(flask.url_for('main.home')) # Get git commit hash app.config['COMMIT_HASH'] = None diff --git a/nyaa/views/account.py b/nyaa/views/account.py index ce7922c..f3dd182 100644 --- a/nyaa/views/account.py +++ b/nyaa/views/account.py @@ -93,7 +93,8 @@ def register(): @bp.route('/profile', methods=['GET', 'POST']) def profile(): if not flask.g.user: - return flask.redirect('/') # so we dont get stuck in infinite loop when signing out + # so we dont get stuck in infinite loop when signing out + return flask.redirect(flask.url_for('main.home')) form = forms.ProfileForm(flask.request.form) @@ -130,11 +131,13 @@ def profile(): def redirect_url(): + home_url = flask.url_for('main.home') + url = flask.request.args.get('next') or \ flask.request.referrer or \ - '/' + home_url if url == flask.request.url: - return '/' + return home_url return url