如何在websocket端点外发出websocket消息?

12

我正在使用Flask构建一个网站,其中还使用了Flask-SocketIO来使用Websockets,但有一件事我不明白。

我创建了一个聊天功能。当一个用户发送一条消息时,我使用websockets将该消息发送到服务器,之后我在同一次调用中向其他用户发出该消息:

@socketio.on('newPM', namespace='/test')
@login_required_with_global
def io_newMessage(theJson):
    emit('message', {'message': theJson['message']}, room=str(theJson['toUserId']))

但是假设我想在文件保存时向用户发出一条消息。这意味着我需要在文件被POST的视图内部发出一条消息。因此,根据 flask_socketio 文档,我可以在 emit 中添加一个名称空间。所以我写了这个:

@app.route('/doc', methods=['POST'])
@login_required
def postDoc():
    saveDocument(request.files['file'], g.user.id)
    emit('my response', {'data': 'A NEW FILE WAS POSTED'}, room=current_user.id, namespace='/test')
    return jsonify({'id': str(doc.id)})

但是从下面的堆栈跟踪来看,命名空间仍然存在问题; werkzeug出现了AttributeError: 'Request' object has no attribute 'namespace'

有人知道我在这里做错了什么吗?还是这是flask_socketio中的一个错误?欢迎所有提示!

Traceback (most recent call last):
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1820, in wsgi_app
    response = self.make_response(self.handle_exception(e))
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1403, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1817, in wsgi_app
    response = self.full_dispatch_request()
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1477, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1381, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1475, in full_dispatch_request
    rv = self.dispatch_request()
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1461, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "/usr/local/lib/python2.7/dist-packages/flask_login.py", line 758, in decorated_view
    return func(*args, **kwargs)
  File "/home/vg/app/views.py", line 768, in emitNotificationCount
    emit('menuInit', emitJson, room=current_user.id, namespace='/test')
  File "/usr/local/lib/python2.7/dist-packages/flask_socketio/__init__.py", line 444, in emit
    return request.namespace.emit(event, *args, **kwargs)
  File "/usr/local/lib/python2.7/dist-packages/werkzeug/local.py", line 338, in __getattr__
    return getattr(self._get_current_object(), name)
AttributeError: 'Request' object has no attribute 'namespace'

@Shashank - 太棒了,就是这个,谢谢!如果你把你的评论作为答案添加进来,我可以接受它。 - kramer65
1个回答

33

引用Miguel Grinberg在Flask-SocketIO GitHub上的开放问题页面上的回答:

当您想要从常规路由发出时,您必须使用socketio.emit(),只有socket处理程序才具有调用普通emit()所需的socketio上下文。

因此,以一个例子来说明:

from flask_socketio import SocketIO

app = Flask(__name__)
app.config.from_object('config')
socketio = SocketIO(app)

@app.route('/doc', methods=['POST'])
def postDoc():
    saveDocument(request.files['file'], g.user.id)
    socketio.emit('my response', {'data': 'A NEW FILE WAS POSTED'}, room=current_user.id)
    return jsonify({'id': str(doc.id)})

如果您正在使用namespace,请不要忘记在同一个namespace中发布。 - JRichardsz

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接