教程 Redis+ flask+vue 在线聊天

2019-12-06 来源:  发布在  https://www.cnblogs.com/pypypy/p/11997637.html

知识点

  • 基于 Server-Sent Event 工作方式,Web 即时通信
  • Redis 包 发布订阅功能的使用
  • flask 快速入门,常用对象实例方法函数
  • Vuejs 列表页面自动渲染

效果图


代码段


  • app.py 主文件
import datetime
from flask import Flask, session, redirect, Response, request, render_template, url_for, flash
from redis import StrictRedis

app = Flask(__name__)

# session cookie密钥
app.secret_key = 'pardon110'

# 连接redis数据库,默认是零号库,可随便更改
rds = StrictRedis(db=3)

# 消息生成器
def event_stream():
    # 从数据库连接上获取发布订阅管理对象实例
    pub = rds.pubsub()
    # 在管理订阅(建立通道)频道
    pub.subscribe('chat')
    # 监听频道信息
    for message in pub.listen():
        print(type(message['data']), type(message), message)
        # 只响应有消息的(字节),首次无消息返回的为int状态码对象,直接忽略
        if isinstance(message['data'], bytes):
            # 转为utf8字符串,发送 SSE(Server Send Event)协议格式的数据
            yield 'data: %s\n\n' % message['data'].decode()

# 首次访问需要登录
@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        session['username'] = request.form['user']
        # 重定向到home处理器
        flash('您已经成功登录!')
        return redirect(url_for('home'))
    return '<p><strong>登录</strong></p><form action="" method="post">用户名: <input name="user">'

# 接收js发送过来的消息
@app.route('/post', methods=['POST'])
def post():
    # 获取表单提交内容
    message = request.form['message']
    # 获取当前请求对象的session实例
    user = session.get('username', 'anonymous')
    # 返回一个指定字段的时间值
    now = datetime.datetime.now().replace(microsecond=0).time()
    # 通过频道发布消息
    rds.publish('chat', u'[%s] %s: %s' % (now.isoformat(), user, message))
    # 响应对象
    return Response(status=204)

# 事件流接口
@app.route('/stream')
def stream():
    return Response(event_stream(), mimetype="text/event-stream")

@app.route('/')
@app.route('/<name>')
def home(name=None):
    # 通过路由参数或querystring动注册为当前用户
    if name or len(request.args) > 0:
        session['username'] = name if name else request.args.get(
            'user', '')
        # 消息闪现(存储在session内,模板页用完即丢)
        flash(session['username']+'已经成功登录,加入聊天室!')
    # 否则强制用户登录
    if 'username' not in session:
        return redirect('/login')

    # 模板渲染
    data = {
        "username": session['username'],
        "tip": "正在聊天中..."
    }
    # 关键字参数解包,返回元组(框架会自动解析为一个完整的response对象)
    return render_template('home.html', **data), 200

@app.route('/logout')
def logout():
    # 清空当前session信息
    session.pop('username', None)
    return redirect(url_for('login'))

if __name__ == '__main__':
    # app.debug = True
    # falsk1.0.2默认多线程开启,默认端口是5000
    app.run(host='0.0.0.0', debug=True)
  • sse.js 文件
//  vue实例
var vm = new Vue({
    el: '#out',
    // 自定义定界符,避免vue定界符与jinja2冲突,改变vue起始定界
    delimiters: ['@{', '}}'],
    // 实例属性
    data: {
        items: []
    },
    mounted() {
        // 监听服务端消息
        sse()
    }
})

// 通过EventSource对象与后端建立连接
function sse() {
    var source = new EventSource('/stream')
    // 监听消息事件(与nodejs用data事件sock流
    source.onmessage = function (e) {
        // XSS in chat is fun
        // 动态渲染页面(代理vue实例数据操作)
        vm.$data.items.push(e.data)
    }
}

// ajax与后端交互,向服务端发送消息
$('#in').keyup(function (e) {
   // 回车keycode为13
    if (e.keyCode == 13) {
        $.post('/post', { 'message': $(this).val() })
        $(this).val('')
    }
})

// 自动消失闪现
$(document).ready(function () {
    setTimeout(() => {
        $(".flashes").fadeOut(3000);
    })
})

前置条件

  1. 启动 redis 数据库服务 启动 flask 5000 端口(需自行安装 redis 客户端)
  2. 选择 post 登录,或路由或 querystring 方式自动注册登录
  3. 在不同客户端登录,比如多个不同浏览器进入聊天室
  4. 开始输入消息,回车发送显示开始聊天
 

小结

  • sse 协议格式,python 生成器
  • vuejs 定界符变更,生命周期及 vm 实例属性的使用
  • flask 基本入门,redis 数据库发布订阅应用
  • 学习交流群887934385
  • 免费领取flask及python资料

相关文章