会员模块实现
会员注册
class RegistForm(FlaskForm):
name = StringField(
label="昵称",
validators=[
DataRequired("昵称不能为空！")
],
description="昵称",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入昵称！",
}
)
email = StringField(
label="邮箱",
validators=[
DataRequired("邮箱不能为空！"),
Email("邮箱格式不正确！")
],
description="邮箱",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入邮箱！",
}
)
phone = StringField(
label="手机",
validators=[
DataRequired("手机号不能为空！"),
Regexp("1[34578]\d{9}", message="手机格式不正确！"),
Length(min=11, max=11, message="手机长度不正确！")
],
description="手机",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入手机！",
}
)
pwd = PasswordField(
label="密码",
validators=[
DataRequired("密码不能为空！")
],
description="密码",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入密码！",
}
)
repwd = PasswordField(
label="确认密码",
validators=[
DataRequired("请输入确认密码！"),
EqualTo('pwd', message="两次密码不一致！")
],
description="确认密码",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入确认密码！",
}
)
submit = SubmitField(
'注册',
render_kw={
"class": "btn btn-lg btn-success btn-block",
}
)
def validate_name(self, field):
name = field.data
user = User.query.filter_by(name=name).count()
if user == 1:
raise ValidationError("昵称已经存在！")
def validate_email(self, field):
email = field.data
user = User.query.filter_by(email=email).count()
if user == 1:
raise ValidationError("邮箱已经存在！")
def validate_phone(self, field):
phone = field.data
user = User.query.filter_by(phone=phone).count()
if user == 1:
raise ValidationError("手机号码已经存在！")
视图函数
@home.route("/regist/", methods=["GET", "POST"])
def regist():
"""
会员注册
"""
form = RegistForm()
if form.validate_on_submit():
data = form.data
user = User(
name=data["name"],
email=data["email"],
phone=data["phone"],
pwd=generate_password_hash(data["pwd"]),
uuid=uuid.uuid4().hex
)
db.session.add(user)
db.session.commit()
flash("注册成功！", "ok")
return render_template("home/regist.html", form=form)
修改对应的前端文件
会员登录
class LoginForm(FlaskForm):
name = StringField(
label="账号",
validators=[
DataRequired("账号不能为空！")
],
description="账号",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入账号！",
}
)
pwd = PasswordField(
label="密码",
validators=[
DataRequired("密码不能为空！")
],
description="密码",
render_kw={
"class": "form-control input-lg",
"placeholder": "请输入密码！",
}
)
submit = SubmitField(
'登录',
render_kw={
"class": "btn btn-lg btn-primary btn-block",
}
)
视图函数
在登录时需要校验密码,需要在User模型中添加check_pwd函数,和admin模型中的一样
@home.route("/login/", methods=["GET", "POST"])
def login():
"""
登录
"""
form = LoginForm()
if form.validate_on_submit():
data = form.data
user = User.query.filter_by(name=data["name"]).first()
if user:
if not user.check_pwd(data["pwd"]):
flash("密码错误！", "err")
return redirect(url_for("home.login"))
else:
flash("账户不存在！", "err")
return redirect(url_for("home.login"))
session["user"] = user.name
session["user_id"] = user.id
userlog = Userlog(
user_id=user.id,
ip=request.remote_addr
)
db.session.add(userlog)
db.session.commit()
return redirect(url_for("home.user"))
return render_template("home/login.html", form=form)
在登录时添加登录日志
修改对应的前端文件
修改会员资料
class UserdetailForm(FlaskForm):
name = StringField(
label="账号",
validators=[
DataRequired("账号不能为空！")
],
description="账号",
render_kw={
"class": "form-control",
"placeholder": "请输入账号！",
}
)
email = StringField(
label="邮箱",
validators=[
DataRequired("邮箱不能为空！"),
Email("邮箱格式不正确！")
],
description="邮箱",
render_kw={
"class": "form-control",
"placeholder": "请输入邮箱！",
}
)
phone = StringField(
label="手机",
validators=[
DataRequired("手机号不能为空！"),
Regexp("1[34578]\d{9}", message="手机格式不正确！"),
Length(min=11, max=11, message="手机长度不正确！")
],
description="手机",
render_kw={
"class": "form-control",
"placeholder": "请输入手机！",
}
)
face = FileField(
label="头像",
validators=[
DataRequired("请上传头像！")
],
description="头像",
)
info = TextAreaField(
label="简介",
validators=[
DataRequired("简介不能为空！")
],
description="简介",
render_kw={
"class": "form-control",
"rows": 10
}
)
submit = SubmitField(
'保存修改',
render_kw={
"class": "btn btn-success",
}
)
视图函数
修改文件名
def change_filename(filename):
"""
修改文件名称
"""
fileinfo = os.path.splitext(filename)
filename = datetime.datetime.now().strftime("%Y%m%d%H%M%S") + \
str(uuid.uuid4().hex) + fileinfo[-1]
return filename
def user_login_req(f):
"""
登录装饰器
"""
@wraps(f)
def decorated_function(*args, **kwargs):
if "user" not in session:
return redirect(url_for("home.login", next=request.url))
return f(*args, **kwargs)
return decorated_function
@home.route("/user/", methods=["GET", "POST"])
@user_login_req
def user():
form = UserdetailForm()
user = User.query.get(int(session["user_id"]))
form.face.validators = []
if request.method == "GET":
# 赋初值
form.name.data = user.name
form.email.data = user.email
form.phone.data = user.phone
form.info.data = user.info
if form.validate_on_submit():
data = form.data
if form.face.data != "":
file_face = secure_filename(form.face.data.filename)
if not os.path.exists(app.config["FC_DIR"]):
os.makedirs(app.config["FC_DIR"])
os.chmod(app.config["FC_DIR"], 6)
user.face = change_filename(file_face)
form.face.data.save(app.config["FC_DIR"] + user.face)
name_count = User.query.filter_by(name=data["name"]).count()
if data["name"] != user.name and name_count == 1:
flash("昵称已经存在!", "err")
return redirect(url_for("home.user"))
email_count = User.query.filter_by(email=data["email"]).count()
if data["email"] != user.email and email_count == 1:
flash("邮箱已经存在!", "err")
return redirect(url_for("home.user"))
phone_count = User.query.filter_by(phone=data["phone"]).count()
if data["phone"] != user.phone and phone_count == 1:
flash("手机已经存在!", "err")
return redirect(url_for("home.user"))
# 保存
user.name = data["name"]
user.email = data["email"]
user.phone = data["phone"]
user.info = data["info"]
db.session.add(user)
db.session.commit()
flash("修改成功!", "ok")
return redirect(url_for("home.user"))
return render_template("home/user.html", form=form, user=user)
在初始化文件中添加app.config["FC_DIR"] = os.path.join(os.path.abspath(os.path.dirname(__file__)), "static/uploads/users/")
修改对应的前端文件
修改密码
class PwdForm(FlaskForm):
old_pwd = PasswordField(
label="旧密码",
validators=[
DataRequired("旧密码不能为空！")
],
description="旧密码",
render_kw={
"class": "form-control",
"placeholder": "请输入旧密码！",
}
)
new_pwd = PasswordField(
label="新密码",
validators=[
DataRequired("新密码不能为空！"),
],
description="新密码",
render_kw={
"class": "form-control",
"placeholder": "请输入新密码！",
}
)
submit = SubmitField(
'修改密码',
render_kw={
"class": "btn btn-success",
}
)
视图函数
@home.route("/pwd/", methods=["GET", "POST"])
@user_login_req
def pwd():
"""
修改密码
"""
form = PwdForm()
if form.validate_on_submit():
data = form.data
user = User.query.filter_by(name=session["user"]).first()
if not user.check_pwd(data["old_pwd"]):
flash("旧密码错误！", "err")
return redirect(url_for('home.pwd'))
user.pwd = generate_password_hash(data["new_pwd"])
db.session.add(user)
db.session.commit()
flash("修改密码成功，请重新登录！", "ok")
return redirect(url_for('home.logout'))
return render_template("home/pwd.html", form=form)
修改对应的前端文件
会员登录日志
登陆日志不需要表单
@home.route("/loginlog/<int:page>/", methods=["GET"])
@user_login_req
def loginlog(page=1):
"""
会员登录日志
"""
if page <= 0:
page = 1
page_data = Userlog.query.filter_by(
user_id=int(session["user_id"])
).order_by(
Userlog.addtime.desc()
).paginate(page=page, per_page=10)
return render_template("home/loginlog.html", page_data=page_data)
新建app/templates/ui/home_page.html,用来分页
{% macro page(data,url) -%}
{% if data %}
<nav aria-label="Page navigation">
<ul class="pagination">
<li><a href="{{ url_for(url,page=1) }}">首页</a></li>
{% if data.has_prev %}
<li><a href="{{ url_for(url,page=data.prev_num) }}">上一页</a></li>
{% else %}
<li class="disabled"><a href="#">上一页</a></li>
{% endif %}
{% for v in data.iter_pages() %}
{% if v %}
{% if v == data.page %}
<li class="active"><a href="#">{{ v }}</a></li>
{% else %}
<li><a href="{{ url_for(url,page=v) }}">{{ v }}</a></li>
{% endif %}
{% endif %}
{% endfor %}
{% if data.has_next %}
<li><a href="{{ url_for(url,page=data.next_num) }}">下一页</a></li>
{% else %}
<li class="disabled"><a href="#">下一页</a></li>
{% endif %}
<li><a href="{{ url_for(url,page=data.pages) }}">尾页</a></li>
</ul>
</nav>
{% endif %}
{%- endmacro %}
修改对应的前端代码,并使用这个分页
网友评论