基于flask的网页聊天室（四）
前言
接前天的内容，今天完成了消息的处理
具体内容
上次使用了flask_login做用户登录，但是直接访问login_requare装饰的函数会报401错误，这里可以自定义login_requare拦截后的操作：
@login_manager.unauthorized_handler
def unauthorized():
return redirect(url_for('auth.login'))
这样会重定向到登录页面
然后建立了用户的消息表：
class User(UserMixin,db.Model):
id = db.Column(db.Integer,primary_key=True,autoincrement=True)
email = db.Column(db.String(256),unique=True,nullable=False)
username = db.Column(db.String(32),unique=True,nullable=False)
password_hash = db.Column(db.String(128))
avatar_url = db.Column(db.String(256))
messages = db.relationship('Message',back_populates='author',cascade='all')
def __init__(self,**kwargs):
super(User,self).__init__(**kwargs)
if self.email is not None and self.avatar_url is None:
self.avatar_url = 'https://gravatar.com/avatar/'+hashlib.md5(self.email.encode('utf-8')).hexdigest()+'?d=identicon'
class Message(db.Model):
id = db.Column(db.Integer,primary_key=True,autoincrement=True)
content = db.Column(db.Text,nullable=False)
create_time = db.Column(db.DateTime,default=datetime.utcnow,index=True)
author_id = db.Column(db.Integer,db.ForeignKey('user.id'))
author = db.relationship('User',back_populates='messages')
然后用bootstarp做了个消息的显示以及发送的页面，再自己随便改吧改吧。
接下来就是怎样做消息的发送以及接收了
这里使用websocket的方式，它可以使客户端与服务端建立起全双工的通信方式
客户端使用socket.io.js,服务端使用flask-socketio扩展
于是首先要pip安装flask-socketio
然后：
from flask_socketio import SocketIO
socketio = SocketIO(app, async_mode='eventlet')
socketio.run(app,host='0.0.0.0')
像上边这样把原来的app启动改一下，eventlet是服务器，它支持websocket，必须要先pip安装
然后具体处理消息，首先是消息的发送：
在前端
首先要引入socket.io.js，然后
function send() {
//点击发送按钮
$('#send_button').click(function () {
var content = $('#text_area').val().trim();
if (content==""){
}else {
socket.emit('new_message',content);
$('#text_area').val("")
}
})
}
通过socket.emit，把消息发送到服务端new_message就是相当于这个事件的名称，content作为参数传递
在服务端：
@socketio.on('new_message')
def new_message(content):
print(content)
message = models.Message(author=current_user._get_current_object(),content=clean_html(content))
db.session.add(message)
db.session.commit()
emit('new_message',{'message_html':render_template('message.html',message=message)},broadcast=True)
用装饰器socketio.on(‘new_message’)装饰的函数就是new_message的事件处理，这里把它存入表中，并返回一个消息的html字符串，传入message对象渲染，broadcast=True表示广播，意思是所有与服务端建立websocket链接的都能收到该消息
接下来是客户端消息的显示：
function get() {
socket.on('new_message',function (data) {
$('#message_container').append(data.message_html);
scrollToEnd();
})
}
这里同样用socket.on方法接受客户端返回的信息，并把它添加到显示区域，并把滚轮滚到底部。
除此之外，每次访问浏览器页面我默认设置的是显示历史10条消息，之后可能会添加历史消息刷新
@chat.route('/chat', methods=['GET', "POST"],endpoint='chat')
@login_required
def chatroom():
if request.method == 'GET':
message_list = db.session.query(models.Message).order_by(models.Message.id).all()
message_list.reverse()
message_list = message_list[:9]
message_list.reverse()
return render_template('chatroom.html',message_list=message_list)
到这里，基本的内容就算完成了，下面是一个效果图，是我用手机和电脑交互的：
总结
聊天室的内容基本完成，最初目标已经达到，代码放在了github上，如果之后有空还会再完善或添加功能