项目知了问答的app.py文件 import random import string from flask import Flask, session, g, render_template, request, redirect, url_for, jsonify, flash from flask_mail import Message from werkzeug.security import check_password_hash, generate_password_hash import config from blueprints.forms import LoginForm, RegisterForm from exts import db, mail from models import UserModel, EmailCaptchaModel from blueprints.qa import bp as qa_bp from blueprints.auth import bp as auth_bp from flask_migrate import Migrate app = Flask(__name__) # 绑定配置文件 app.config.from_object(config) db.init_app(app) mail.init_app(app) migrate = Migrate(app, db) app.register_blueprint(qa_bp) app.register_blueprint(auth_bp) # blueprint：用来做模块化的 # 电影、读书、音乐、xxx # flask db init：只需要执行一次 # flask db migrate：将orm模型生成迁移脚本 # flask db upgrade：将迁移脚本映射到数据库中 # before_request/ before_first_request/ after_request # hook @app.before_request def my_before_request(): user_id = session.get("user_id") if user_id: user = UserModel.query.get(user_id) setattr(g, "user", user) else: setattr(g, "user", None) @app.context_processor def my_context_processor(): return {"user": g.user} @app.route("/login", methods=['GET', 'POST']) def login(): if request.method == 'GET': return render_template("login.html") else: form = LoginForm(request.form) if form.validate(): username = form.username.data password = form.password.data print(username, password) user = UserModel.query.filter_by(username=username).first() if user and check_password_hash(user.password, password): session['user_id'] = user.id return redirect('/') else: flash("用户名或密码错误", category='error') # if username in users and users[username] == password: # session['username'] = username # return redirect('/') return render_template('login.html', form=form) # if form.validate(): # username = form.username.data # password = form.password.data # user = UserModel.query.filter_by(username=username).first() # if not user: # flash("用户在数据库中不存在！", category='error') # return redirect(url_for("login")) # if check_password_hash(user.password, password): # # cookie： # # cookie中不适合存储太多的数据，只适合存储少量的数据 # # cookie一般用来存放登录授权的东西 # # flask中的session，是经过加密后存储在cookie中的 # session['user_id'] = user.id # return redirect("/") # else: # flash("用户名或密码错误", category='error') # return redirect(url_for("login")) # else: # return redirect(url_for("login")) # GET：从服务器上获取数据 # POST：将客户端的数据提交给服务器 @app.route("/register", methods=['GET', 'POST']) def register(): if request.method == 'GET': return render_template("register.html") else: # 验证用户提交的邮箱和验证码是否对应且正确 # 表单验证：flask-wtf: wtforms form = RegisterForm(request.form) if form.validate(): email = form.email.data username = form.username.data password = form.password.data user = UserModel(email=email, username=username, password=generate_password_hash(password)) db.session.add(user) db.session.commit() return redirect(url_for("login")) else: print(form.errors) return redirect(url_for("register")) @app.route("/captcha/email") def get_email_captcha(): # /captcha/email/<email> # /captcha/email?email=xxx@qq.com email = request.args.get("email") # 4/6：随机数组、字母、数组和字母的组合 source = string.digits*4 captcha = random.sample(source, 4) captcha = "".join(captcha) # I/O：Input/Output message = Message(subject="知了传课注册验证码", recipients=[email], body=f"您的验证码是:{captcha}") mail.send(message) # memcached/redis # 用数据库表的方式存储 email_captcha = EmailCaptchaModel(email=email,captcha=captcha) db.session.add(email_captcha) db.session.commit() # RESTful API # {code: 200/400/500, message: "", data: {}} return jsonify({"code": 200, "message": "", "data": None}) if __name__ == '__main__': app.run(debug=True)写出该代码的所有功能模块并对该模块进行概述，按照以下格式：会员注册模块主要用于实现新用户注册成为网站会员的功能。在会员注册页面中，用户需要填写会员信息，然后单击“同意协议并注册”按钮，程序将自动验证输入的账户是否唯一，如果唯一，就把填写的会员信息保存到数据库中，否则给出提示，需要修改成唯一账户后，方可完成注册。另外，程序还将验证输入的信息是否合法，例如，不能输入中文的账户名称等。会员注册流程如下图所示。