当前位置: 移动技术网 > IT编程>脚本编程>Python > 基于flask的网页聊天室(四)

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

2019年05月14日  | 移动技术网IT编程  | 我要评论

南宁百货网上商城,湖北天门实验高中,打车软件取消加价

基于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上,如果之后有空还会再完善或添加功能

如对本文有疑问,请在下面进行留言讨论,广大热心网友会与你互动!! 点击进行留言回复

相关文章:

验证码:
移动技术网