python flask websocket_Flask中使用websocket

因为业务需求需要使用websocket，所以在flask中使用websocket。在网上很多的资料，本文只记录我使用的方法。

首先是需要导入包

pip install gevent-websocket

导入包之后直接在项目中使用

from flask import Flask, request

from geventwebsocket.handler import WebSocketHandler #提供WS(websocket)协议处理

from geventwebsocket.server import WSGIServer #websocket服务承载

#WSGIServer导入的就是gevent.pywsgi中的类

from gevent.pywsgi import WSGIServer

from geventwebsocket.websocket import WebSocket

app = Flask(__name__)

clientList = []

@app.route("/")

def func():

client_socket = request.environ.get('wsgi.websocket')

if not client_socket:

return "error"

clientList.append(client_socket)

while 1:

for i in clientList:

try:

i.send(json.dumps(“111111”))

except Exception as e:

continue

if __name__ == '__main__':

http_server = WSGIServer(('127.0.0.1', 8888), application=app, handler_class=WebSocketHandler)

http_server.serve_forever()

其实使用比较简单，这里还是说一下我踩得几个坑吧：

1.在这个使用之后运行的时候尤其是pycharm运行的时候不要直接拿IDE的运行，而是使用命令行python app.py来运行，否则是直接运行flask项目，而不是使用下面的这种方式去启动项目。

2.因为在Gin框架中可以将get请求升级成websocket，这里也是同理。但是我之前想使用flask restful这个包，直接把get请求升级成websocket，但是那种方式是行不通的。现在还没有解决，如果有人成功了，可以在本文下留言到自己的博客地址供大家学习。非常感谢！

3.因为代码中有一个while 1，这导致在运行代码的时候会一直在这个请求里阻塞住，这个也是由于个人能力有限，只能把websocket这个接口单独拆出来做一个服务。当然这个在和很多业内的朋友讨论的时候也一致认为应该如此。但是如果有人解决，也请参考上一条，不胜感激！