Move /user/activate/<payload> route into users

Move supporting functions as well:
  - get_serializer
  - get_activation_link
This commit is contained in:
Kfir Hadas 2017-07-10 23:13:38 +03:00
parent 0887dde6fc
commit 50529920bd
3 changed files with 38 additions and 37 deletions

View File

@ -11,7 +11,6 @@ from flask_paginate import Pagination
from werkzeug import url_encode
from werkzeug.datastructures import CombinedMultiDict
from itsdangerous import BadSignature, URLSafeSerializer
from sqlalchemy.orm import joinedload
from nyaa import api_handler, app, backend, db, forms, models, torrents, views
@ -286,27 +285,6 @@ def render_rss(label, query, use_elastic, magnet_links=False):
return response
@app.route('/user/activate/<payload>')
def activate_user(payload):
s = get_serializer()
try:
user_id = s.loads(payload)
except BadSignature:
flask.abort(404)
user = models.User.by_id(user_id)
if not user:
flask.abort(404)
user.status = models.UserStatusType.ACTIVE
db.session.add(user)
db.session.commit()
return flask.redirect('/login')
@cached_function
def _create_upload_category_choices():
''' Turns categories in the database into a list of (id, name)s '''
@ -566,18 +544,6 @@ def _get_cached_torrent_file(torrent):
return open(cached_torrent, 'rb'), os.path.getsize(cached_torrent)
def get_serializer(secret_key=None):
if secret_key is None:
secret_key = app.secret_key
return URLSafeSerializer(secret_key)
def get_activation_link(user):
s = get_serializer()
payload = s.dumps(user.id)
return flask.url_for('activate_user', payload=payload, _external=True)
@app.template_filter()
def timesince(dt, default='just now'):
"""

View File

@ -6,8 +6,8 @@ from ipaddress import ip_address
import flask
# Importing nyaa.routes for get_activation_link
from nyaa import app, db, forms, models, routes
from nyaa import app, db, forms, models
from nyaa.views.users import get_activation_link
bp = flask.Blueprint('account', __name__)
@ -72,7 +72,7 @@ def register():
db.session.commit()
if app.config['USE_EMAIL_VERIFICATION']: # force verification, enable email
activ_link = routes.get_activation_link(user)
activ_link = get_activation_link(user)
send_verification_email(user.email, activ_link)
return flask.render_template('waiting.html')
else: # disable verification, set user as active and auto log in

View File

@ -3,6 +3,8 @@ import math
import flask
from flask_paginate import Pagination
from itsdangerous import BadSignature, URLSafeSerializer
from nyaa import app, db, forms, models
from nyaa.search import (DEFAULT_MAX_SEARCH_RESULT, DEFAULT_PER_PAGE, SERACH_PAGINATE_DISPLAY_MSG,
_generate_query_string, search_db, search_elastic)
@ -135,6 +137,27 @@ def view_user(user_name):
admin_form=admin_form)
@bp.route('/user/activate/<payload>')
def activate_user(payload):
s = get_serializer()
try:
user_id = s.loads(payload)
except BadSignature:
flask.abort(404)
user = models.User.by_id(user_id)
if not user:
flask.abort(404)
user.status = models.UserStatusType.ACTIVE
db.session.add(user)
db.session.commit()
return flask.redirect(flask.url_for('account.login'))
def _create_user_class_choices(user):
choices = [('regular', 'Regular')]
default = 'regular'
@ -154,3 +177,15 @@ def _create_user_class_choices(user):
default = 'banned'
return default, choices
def get_serializer(secret_key=None):
if secret_key is None:
secret_key = app.secret_key
return URLSafeSerializer(secret_key)
def get_activation_link(user):
s = get_serializer()
payload = s.dumps(user.id)
return flask.url_for('users.activate_user', payload=payload, _external=True)