def submit_new_book():
form = SubmitNewBookForm()
if form.validate_on_submit():
start_spider.queue(form.url.data)
flash('任务已经提交到队列，这本书或许需要一些时间才能被搜索到', 'is-success')
else:
flash_errors(form)
return render_template('submit_new_book.html', form=form)
def settings():
form = SettingForm()
if form.validate_on_submit():
current_user.about_me = form.about_me.data
db.session.commit()
flash('资料修改成功', 'is-success')
elif request.method == 'GET':
form.about_me.data = current_user.about_me
else:
flash_errors(form)
return render_template('settings.html', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('恭喜，您已注册成功!', 'is-success')
return redirect(url_for('auth.login'))
else:
flash_errors(form)
return render_template('auth/register.html', title='注册', form=form)
def reset_password(token):
if current_user.is_authenticated:
return redirect(url_for('main.index'))
user = User.verify_reset_password_token(token)
if not user:
return redirect(url_for('main.index'))
form = ResetPasswordForm()
if form.validate_on_submit():
user.set_password(form.password.data)
db.session.commit()
flash('密码重置成功', 'is-success')
return redirect(url_for('auth.login'))
else:
flash_errors(form)
return render_template('auth/reset_password.html', form=form)
def reset_password_request():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = ResetPasswordRequestForm()
if form.validate_on_submit():
user = User.query.filter_by(email=form.email.data).first()
if user:
send_password_reset_email.queue(user)
flash('请点击邮箱中的链接完成密码重置，如果没有收到邮件，请检查垃圾箱', 'is-success')
else:
flash('邮箱不存在', 'is-danger')
return redirect(url_for('auth.login'))
else:
flash_errors(form)
return render_template('auth/reset_password_request.html', title='重置密码', form=form)
def login():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = LoginForm()
if form.validate_on_submit():
user = User.query.filter_by(username=form.username.data).first()
if user is None or not user.check_password(form.password.data):
flash('用户名或密码不正确', 'is-danger')
return redirect(url_for('auth.login'))
login_user(user, remember=form.remember_me.data)
next_page = request.args.get('next')
if not next_page or url_parse(next_page).netloc != '':
next_page = url_for('main.index')
return redirect(next_page)
else:
flash_errors((form))
return render_template('auth/login.html', title='登入', form=form)
def comment():
form = CommentForm()
if form.validate_on_submit():
u = User.query.get(form.user_id.data)
b = Book.query.get(form.book_id.data)
if u and b:
c = Comment.query.filter_by(user=u, book=b).first()
if c:
c.body = form.body.data
c.score = form.score.data
c.timestamp = arrow.utcnow()
flash('修改成功!', 'is-success')
else:
c = Comment(user=u, book=b, body=form.body.data, score=form.score.data)
db.session.add(c)
flash('评论成功', 'is-success')
b.set_avg()
db.session.commit()
else:
flash('该用户或书籍不存在', 'is-danger')
return redirect(url_for('main.index'))
else:
flash_errors(form)
return redirect(url_for('book.index', book_id=form.book_id.data))