def setting():
"""设置用户信息方法
"""
g.user = user = User.is_login()
form = SettingForm()
u = User()
if request.method == "POST":
if not u.has_username(form.username.data) or session["username"] == form.username.data:
"""如果修改后的用户名不存在数据库或者和当前登录的用户名相同则更新数据库中对应的记录"""
if form.username.data == "":
flash(u"用户名不能为空", "error")
return render_template("/user/setting.html", form=form, title=u"修改设置")
if form.password.data == "":
flash(u"密码不能为空", "error")
return render_template("/user/setting.html", form=form, title=u"修改设置")
if form.email.data == "":
flash(u"邮箱不能为空", "error")
return render_template("/user/setting.html", form=form, title=u"修改设置")
if not u.edit(user, form.username.data, form.password.data, form.email.data):
"""写入数据库"""
flash(u"修改失败", "error")
return render_template("/user/setting.html", form=form, title=u"修改设置")
user.dologin()
flash(u"用户信息修改成功")
return redirect(url_for("index"))
else:
flash(u"用户已存在", "warning")
return render_template("/user/setting.html", form=form, title=u"修改设置")
else:
form.username = user.username
form.password = user.password
form.email = user.email
return render_template("/user/setting.html", form=form, title=u"修改设置")
def login():
"""登录用户视图
"""
form = LoginForm(request.form)
if form.validate_on_submit():
u = User.has_username(form.username.data)
if not u:
flash(u"无此用户", "warning")
return render_template("/user/login.html", form=form, title=u"登录")
if not u.is_correct_password(form.password.data):
flash(u"密码错误", "error")
return render_template("/user/login.html", form=form, title=u"登录")
u.dologin()
"""如果输入的登录用户名和密码都正确则成功跳转到系统主页
"""
flash(u" %s 登录成功" % form.username.data)
return redirect(url_for("index"))
return render_template("/user/login.html", form=form, title=u"登录")
