一、写在前面
正所谓“纸上得来终觉浅,方知此事要躬行”,在看文档和视频之余,我觉得还是要动手做点什么东西才能更好地学习吧,毕竟有些东西光看文档真的难以理解,于是就试着使用Flask框架做了一个简易留言板,实现了注册、登录和留言的功能。
二、开发环境
系统:Windows10
Python版本:Python3.7
Flask版本:Flask 1.0.2
MySQL版本:MySQL8.0
三、具体步骤
1.连接数据库
在做连接MySQL数据库的时候碰到了一堆问题,在上两篇博客中都有提到,这里就不多说了。完成数据库设置之后,我创建了两张表,一张用于储存用户信息的users表,一张用于储存留言信息的messages表。定义两张表的代码如下:
1 from flask_sqlalchemy import SQLAlchemy 2 import mysql.connector 3 4 app = Flask(__name__) 5 app.config['SECRET_KEY'] = '123456' 6 app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+mysqlconnector://root:qwer1234@localhost/flask?auth_plugin=mysql_native_password" 7 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True 8 app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True 9 10 # 获取SQLAlchemy实例对象 11 db = SQLAlchemy(app) 12 13 14 # 用户 15 class User(db.Model): 16 __tablename__ = "users" 17 id = db.Column(db.Integer, primary_key=True) 18 username = db.Column(db.String(16), unique=True) # 用户名 19 password = db.Column(db.String(16)) # 密码 20 email = db.Column(db.String(32), unique=True) # 邮箱 21 22 def __repr__(self): 23 return '<User %r>' % self.username 24 25 26 # 留言 27 class Message(db.Model): 28 __tablename__ = "messages" 29 id = db.Column(db.Integer, primary_key=True) 30 user = db.Column(db.String(16)) # 用户名 31 content = db.Column(db.String(100)) # 内容 32 date = db.Column(db.Date) # 日期 33 time = db.Column(db.Time) # 时间 34 is_anonymous = db.Column(db.Boolean) # 是否匿名 35 36 def __repr__(self): 37 return '<Message %r User %r>' % (self.id, self.user)
这里先说下SQLAlchemy常用数据类型:
1)Integer,整形,映射到数据库中是int类型。
2)String,可变字符类型,映射到数据库中是varchar类型。
3)Date,存储时间,只能存储年月日。映射到数据库中是date类型。
4)Time,存储时间,可以存储时分秒。映射到数据库中也是time类型。
5)Boolean,布尔类型,映射到数据库中的是tinyint类型。
可以看到两张表中都有一个值为整数的id字段,该字段为主键,然后就是两个类中都有__repr__()方法,该方法和__str__()类似,会返回对这个类的描述。但是这还只是完成了对这两张表的定义,还没有在数据库中完成创建。我是把表的定义写在app.py中的,然后创建了一个model.py用于实现创建数据表,其中的代码如下:
1 from app import db 2 3 4 db.create_all()
2.用户注册
注册的时候需要输入用户名、密码和邮箱,前端完成验证,后台完成注册用户的操作。在进行数据库操作的时候,可能因为某些原因导致用户注册失败,所以还是需要使用异常处理的,若注册失败则返回注册页面并显示提示信息,若注册成功则使用redirect()方法跳转到登录页面,比如跳转到登录页面的的代码就是:
注册需要GET方法和POST方法,因此在编写路由的时候需要写上methods=['GET','POST'],完整代码如下:
1 @app.route('/register', methods=['GET', 'POST']) 2 def register(): 3 if request.method == 'GET': 4 return render_template('register.html') 5 else: 6 try: 7 usr = User() 8 usr.id = User.query.count() + 1 9 usr.username = request.form["usr"] 10 usr.password = request.form["pwd"] 11 usr.email = request.form["email"] 12 db.session.add(usr) 13 db.session.commit() 14 return redirect(url_for("login")) 15 except: 16 return render_template('register.html', msg="注册失败!请重试!")
3.用户登录
用户登录的时候需要对输入的用户名和密码进行验证,验证用户名是否存在和密码是否正确。在上一篇博客中已经简单介绍过Flask中使用SQLAlchemy的一些基本操作,而在用户登录的时候需要根据用户名来查询,也就是要使用filter()方法,语句如下:
但是这样还不能得到结果,需要在后面加上一个all()才能得到查询结果。如果用户名和密码都正确,则能够登录成功,此时要使用session来记录用户名,然后跳转到留言主页。
1 @app.route('/login', methods=['GET', 'POST']) 2 def login(): 3 if request.method == "GET": 4 return render_template("login.html") 5 else: 6 usr = request.form["usr"] 7 pwd = request.form["pwd"] 8 try: 9 the_user = User.query.filter(User.username == usr).all() # 查询用户是否存在 10 if len(the_user) == 1: 11 if the_user[0].password == pwd: 12 session.clear() 13 session["user"] = usr 14 return redirect(url_for("index")) 15 else: 16 msg = "密码错误!" 17 return render_template("login.html", msg=msg) 18 else: 19 msg = "用户名不存在!" 20 return render_template("login.html", msg=msg) 21 except: 22 msg = "登录失败,请重试!" 23 return render_template("login.html", msg=msg)
4.发表留言
用户登录后可以发表留言,用户在发表的时候还可以选择是否匿名,下面是留言板块的截图:
在后台接收到前端发送的用户的留言信息后,要通过session来获取用户名,还要用Python中的datetime模块来得到此时的日期时间,除此之外,还要知道用户是否选择了匿名。如果用户选择了匿名留言,则会向后台发送数据,反之则不会,所以要用异常处理来得到该数据,再将该字段的值设置为True或者False。
和前面一样的,留言主页也需要GET方法和POST方法,然后就是需要查询留言数据并显示到页面上,代码如下:
1 @app.route('/index', methods=['GET', 'POST']) 2 def index(): 3 if request.method == 'GET': 4 result = Message.query.all() 5 return render_template("index.html", result=result) 6 else: 7 msg = Message() 8 msg.id = len(Message.query.all()) + 1 9 msg.user = session["user"] 10 msg.content = request.form["text_input"] # 留言内容 11 now = datetime.datetime.now() # 获取当前时间 12 msg.date = now.date() 13 msg.time = datetime.time(now.hour, now.minute, now.second) 14 try: 15 key = request.form["the_select"] 16 msg.is_anonymous = True 17 except: 18 msg.is_anonymous = False 19 db.session.add(msg) 20 db.session.commit() 21 result = Message.query.all() 22 return render_template("index.html", result=result)
四、运行截图
下面是一个留言主页的截图,其中有几条留言,有匿名的也有不匿名的:
五、个人总结
这个留言板还是很简单的,主要是刚开始学习Flask框架,有很多不了解的地方,因此遇到了很多问题也花了很多时间,后面我会继续学习,并且对这个留言板增加功能和进行改进的,比如回复别人的留言、删除自己的留言等等。
完整代码已上传到GitHub!