import asyncio
import websockets
class transport:
    async def crec(websocket_s,websocket_c):
        while True:
            message = await websocket_c.recv()
            print("Прием У.С. < "+message)
            #await counter.mess_count(websocket,message)
            await transport.ssend(websocket_s,message)
    async def srec(websocket_s,websocket_c):
        while True:
            message = await websocket_s.recv()
            print("Прием Браузер < "+message)
            #await counter.mess_count(websocket_c,message)
            await transport.csend(websocket_c,message)
                    
    async def ssend(websocket_s,message):
        d=0
        await websocket_s.send(message+str(d))
        d=d+1
        print("Отправка браузер > "+message)
                        
    async def csend(websocket_c,message):
        mess = counter.mess_count(message)
        await websocket_c.send(mess)
        print("Отправка У.С. > "+message)
class counter:
    mess_id = 0
    def mess_count(message):
        global mess_id
        message = "id >"+ str(mess_id) +"-"+ message
        mess_id = mess_id + 1
        return message
async def server(websocket_s, path):
    async with websockets.connect('ws://localhost:8000') as websocket_c:
        print("Соединение")
        
        crec_task = asyncio.ensure_future(transport.crec(websocket_s,websocket_c))
        srec_task = asyncio.ensure_future(transport.srec(websocket_s,websocket_c))
        print("Ghbdtn")
        done, pending = await asyncio.wait([crec_task, srec_task],
                                           return_when=asyncio.FIRST_COMPLETED,)
        for task in pending:
            task.cancel()
        print("HI")
        
start_server = websockets.serve(server, 'localhost', 8001)
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()