具体内容
上次使用了flask_login做用户登录,但是直接访问login_requare装饰的函数会报401错误,这里可以自定义login_requare拦截后的操作:
@login_manager.unauthorized_handler
def unauthorized():
return redirect(url_for(‘auth.login‘))
这样会重定向到登录页面
然后建立了用户的消息表:
‘‘‘
想要学习Python?Python学习交流群:1004391443满足你的需求,资料都已经上传群文件,可以自行下载!
‘‘‘
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(http://www.my516.com)
message_list = message_list[:9]
message_list.reverse()
return render_template(‘chatroom.html‘,message_list=message_list)
到这里,基本的内容就算完成了,下面是一个效果图,是我用手机和电脑交互的:
---------------------
作者:娇兮心有之
原文地址:https://www.cnblogs.com/hyhy904/p/10995898.html