python 搭建 websocket server 发送 sensor 数据

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

搞了几天，顺便把代码贴这里，需要的 python 包：

gevent，gevent-websocket，bottle，wiringpi-python

简单说明：

- gevent 提供了支持 concurrent，以及 concurrent 下的 socket 操作。注意，引入 monkey 后，socket 就是 genvent 封装过的了，不再是系统原生 socket。

- 连接到 /ws 后，链接为长链接，使用 gevent.Timeout() 提供的 alarm 来周期执行 sensor 读操作，并将读到的数据发送给 client。

- 浏览器端，直接访问根目录，返回是一个 web client，提供了一个 textarea，将 server 发来的数据显示出来。

- 浏览器端，没有写发送，需要的话可以按 F12 在 console 里面，直接调用 ws.send('your_message') 来想服务器发送。

from bottle import route, run, request, abort, Bottle, static_file

from gevent import monkey; monkey.patch_all()

from gevent import sleep, Timeout

import wiringpi

import serial

app = Bottle()

host = '0.0.0.0'

port = 8080

SensorData = ''

fdI2c = wiringpi.wiringPiI2CSetup(0x40)

ser = serial.Serial("/dev/ttyUSB0", 9600, timeout = 0.1)

def readSensor():

    global SensorData

    serBuffer = ser.readline().rstrip()

    if len(serBuffer) == 0:

        serBuffer = "";

    strength = wiringpi.wiringPiI2CReadReg16(fdI2c, 0x1c);

    position = wiringpi.wiringPiI2CReadReg16(fdI2c, 0x0c);

    angle = wiringpi.wiringPiI2CReadReg16(fdI2c, 0x0e);

    SensorData = serBuffer + ":" \

                 + str(strength) + ":" \

                 + str(position) + ":" \

                 + str(angle);

@app.route('/ws')

def handle_websocket():

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

    if not wsock:

        abort(400, 'Expected WebSocket request.')

    while True:

        try:

            global SensorData

            with Timeout(0.5, False) as timeout:

                message = wsock.receive()

            if not message:

                readSensor()

                message = SensorData

            wsock.send("%r" % message)

            message = u''

        except WebSocketError:

            break

@app.route('/')

def send_index():

    return """<html>

<head>

  <script type='application/javascript' src='http://libs.baidu.com/jquery/2.0.0/jquery.min.js'></script>

  <script type="text/javascript">

    var ws = new WebSocket("ws://%(host)s/ws");

    ws.onopen = function() {

        ws.send("Hello, world");

    };

    ws.onmessage = function (evt) {

        $('#chat').val($('#chat').val() + evt.data + '\\n');

    };

  </script>

</head>

<body>

<form action='#' id='chatform' method='get'>

    <textarea id='chat' cols='100' rows='20'></textarea>

</form>

</body>

</html>"""%{'host':request.headers.get('Host')}

from gevent.pywsgi import WSGIServer

from geventwebsocket.handler import WebSocketHandler

from geventwebsocket import WebSocketError

server = WSGIServer((host, port), app, handler_class=WebSocketHandler)

print "access @ http://%s:%s" % (host, port)

server.serve_forever()