如何使用websocket将消息发送给带有Flask的特定用户

Question

在我用Flask构建的网站中，人们可以互相发送PM。当用户收到pm时，我现在想实现一个类似于StackOverflow的通知。由于SO使用websockets实现了这一点，我开始使用这个关于实现Flask socketIO（与Socket.io一起工作）的教程来实现websockets。

我下载了教程中的示例，我理解其中的代码。然而，我不理解的是：

1. 我如何知道登录的人是否打开了网站并连接到了websocket
2. 如何向特定用户发送消息

因此，假设我有一个简单的路线，人们可以在其中向另一个用户发布PM：

@app.route('/admin/pm', methods=['GET', 'POST'])
@login_required
def pms():
    if request.method == 'POST':
        savePM(g.user.id, request.form['toUserId'], request.form['text'])
        # How do I emit a message here to the user to whom this message is sent?
    return render_template('sendPM.html')

我的评论已经说明了：我如何向收到该消息的用户发送消息？欢迎所有提示！

[编辑]根据Miguel的提示，我想创建一个名为user.id的房间，所以我现在创建了以下连接和断开连接事件：

@socketio.on('connect', namespace='/test')
@login_required
def websocketConnect():
    join_room(g.user.id)
    emit('my response', {'data': 'Connected'}, room=g.user.id)
@socketio.on('disconnect', namespace='/test')
@login_required
def websocketDisconnect():
    leave_room(g.user.id)
    print('Client disconnected')

但在连接时，我得到了下面的堆栈。g对象不是用socketio路由创建的吗？

Traceback (most recent call last):
  File "/Library/Python/2.7/site-packages/gevent/greenlet.py", line 327, in run
    result = self._run(*self.args, **self.kwargs)
  File "/Library/Python/2.7/site-packages/socketio/virtsocket.py", line 403, in _receiver_loop
    retval = pkt_ns.process_packet(pkt)
  File "/Library/Python/2.7/site-packages/socketio/namespace.py", line 164, in process_packet
    return self.call_method_with_acl('recv_connect', packet)
  File "/Library/Python/2.7/site-packages/socketio/namespace.py", line 240, in call_method_with_acl
    return self.call_method(method_name, packet, *args)
  File "/Library/Python/2.7/site-packages/socketio/namespace.py", line 282, in call_method
    return method(*args)
  File "/Library/Python/2.7/site-packages/flask_socketio/__init__.py", line 79, in recv_connect
    self.socketio._dispatch_message(app, self, 'connect')
  File "/Library/Python/2.7/site-packages/flask_socketio/__init__.py", line 137, in _dispatch_message
    ret = self.messages[namespace.ns_name][message](*args)
  File "/Library/Python/2.7/site-packages/flask_login.py", line 758, in decorated_view
    return func(*args, **kwargs)
  File "/Users/kramer65/repos/vg/app/views/webviews.py", line 418, in websocketConnect
    join_room(g.user.id)
  File "/Library/Python/2.7/site-packages/werkzeug/local.py", line 338, in __getattr__
    return getattr(self._get_current_object(), name)
AttributeError: '_AppCtxGlobals' object has no attribute 'user'

Answer 1

我如何知道登录的人是否打开了网站并连接到了websocket？

这取决于你来跟踪。当用户连接到套接字端点时，您将获得connect事件，您可以在那里将该用户注册为已连接。连接事件处理程序可以访问session，因此，如果登录用户的id或昵称已写入会话，则可以在套接字连接处理程序中访问该id或昵称。

此外，当用户离开时，您将获得一个disconnect事件。

如何向特定用户发送消息？

识别用户的最简单方法是将他们放在房间中，然后可以向这些房间发送消息。如果您需要单独称呼用户，那么只需将每个用户放在不同的房间中，以用户的昵称或id命名即可。管理用户房间的最佳位置再次是connect和disconnect事件处理程序。

不完全是你想要的，但我写了一个稍微复杂一点的使用房间的例子：

https://github.com/miguelgrinberg/Flask-SocketIO-Chat