v7.py

#!/usr/bin/env python

import asyncio
import serial.aio
import websockets
import random
import time
import functools

wsdelay = 1
MSGMAX = 100
MSGLONGMAX = 100000
MSGLONGINC = 10
LOOKBACK_INTERVAL = 10

class serPort(asyncio.Protocol):
    msgpart = ""
    msgs = []
    msgslong = []
    msginc = 0
    def connection_made(self, transport):
        self.transport = transport
        transport.serial.flush()
        print('port opened', transport)
        transport.serial.rts = False
        #transport.write(b'hello world\n')
    def data_received(self, data):
        datastr = data.decode('utf-8')
        #print("msg is currently: {}".format(self.msg))
        self.msgpart += datastr
        mparse = self.msgpart.split('\n')
        if len(mparse) > 1:
            if len(mparse) == 2:
                m = mparse[0]
                if len(m.split(',')) == 5:
                    self.msgs.append(m)
                    if (self.msginc >= MSGLONGINC):
                        self.msgslong.append(m)
                        self.msginc = 0
                    self.msginc += 1
                    print("{}".format(m))
            self.msgpart = ""
        if len(self.msgs) > MSGMAX:
            self.msgs.pop(0)
        if len(self.msgslong) > MSGLONGMAX:
            self.msgslong.pop(0)
        #self.transport.close()
    def connection_lost(self, exc):
        print('port closed')
        asyncio.get_event_loop().stop()

async def listener(websocket, path):
    name = await websocket.recv()
    return name

async def consumer(websocket, path, message):
    print("< {}".format(message))
    greeting = "Hello {}!".format(message)
    await websocket.send(greeting)
    print("> {}".format(greeting))

async def producer(websocket, ser):
    while True:
        await asyncio.sleep(wsdelay)
        #mstime = int(time.time()*1000)
        mstime = int(ser.msgs[-1].split(',')[0])
        val = ser.msgs[-1].split(',')[1:]
        message = "{:d},{!s}".format(mstime, ','.join(val).strip())
        return message
    
async def lookback(websocket, ser):
    print(len(ser.msgslong))
    for msg in (ser.msgslong[i] for i in range(0,len(ser.msgslong),LOOKBACK_INTERVAL)):
        try:
            mstime = int(msg.split(',')[0])
            val = msg.split(',')[1:]
            message = "{:d},{!s}".format(mstime, ','.join(val).strip())
            await websocket.send(message)
            print("lookback: {}".format(msg))
        except:
            pass

async def handler(websocket,path,mytasks=[],mycallbacks=[],ser=''):
    await lookback(websocket, ser)
    
    while True:
        listener_task = asyncio.ensure_future(listener(websocket, path))
        producer_task = asyncio.ensure_future(producer(websocket, ser))
        #print("my callbacks: {}".format(mycallbacks))
        
        #tasks = mytasks.append(listener_task) 
        tasks = [listener_task, producer_task]
        #print("mytasks: {}".format(mytasks))
        #print("tasks: {}".format(tasks))
        done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
        
        if listener_task in done:
            ## new data has arrived from the websocket
            print("listener_task triggered!")
            message = listener_task.result()
            await consumer(websocket, path, message)
            mycallbacks[0](message.encode('utf-8'))
            print(message.encode('utf-8'))
        else:
            listener_task.cancel()
        
        if producer_task in done:
            message = producer_task.result()
            await websocket.send(message)
            print("{} sent".format(message))


def main():
    loop = asyncio.get_event_loop()
    coro = serial.aio.create_serial_connection(loop, serPort, url='/dev/arduino', baudrate=38400)
    ser_task = asyncio.ensure_future(coro)
    loop.run_until_complete(coro)
    ###print("ser_task result: {}".format(ser_task.result()))
    ser_transport = ser_task.result()[0]
    ser_protocol = ser_task.result()[1]
    ###writer = asyncio.StreamWriter(ser_transport, ser_protocol, reader, loop)
    ###print(ser_transport)

    myhandler = functools.partial(handler,mytasks=[],mycallbacks=[ser_transport.write],ser=ser_protocol)
    #myhandler = functools.partial(handler,mytasks=[ser_task.result()[1].data_received])
    # replacing '0.0.0.0' restricts access to 'localhost'
    start_server = websockets.serve(myhandler, '0.0.0.0', 8765)
    ws_task = asyncio.ensure_future(start_server)
    asyncio.get_event_loop().run_until_complete(start_server)
    print("ws_task results: {}".format(ws_task.result()))
    ws_serv = ws_task.result()

    #loop = asyncio.get_event_loop()
    #coro = serial.aio.create_serial_connection(loop, serPort, '/dev/arduino',baudrate=38400,timeout=0.2)
    #loop.run_until_complete(coro)
    asyncio.get_event_loop().run_forever()

if __name__ == "__main__":
    main()