码迷,mamicode.com
首页 > Web开发 > 详细

基于flask的网页聊天室(四)

时间:2019-06-10 09:29:31      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:send   rem   enc   ima   trim   信息   广播   room   学习   

具体内容
上次使用了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)
到这里,基本的内容就算完成了,下面是一个效果图,是我用手机和电脑交互的:
---------------------
作者:娇兮心有之 

基于flask的网页聊天室(四)

标签:send   rem   enc   ima   trim   信息   广播   room   学习   

原文地址:https://www.cnblogs.com/hyhy904/p/10995898.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!