Как написать простой клиент Python Websockets

Question

Как написать простой клиент веб-сокетов Python3?

С учетом https://github.com/Pithikos/python-websocket-server (реализация с несколькими клиентами), который был протестирован с предоставленным авторами веб-клиентом (т. Е. Javascript), яЯ хотел бы написать простой клиент Python3, используя стандартный модуль Python Websockets. Я ожидаю, что клиент отправит текстовое сообщение и получит ответ (вроде как эхо-сервер). Ничего особенного на клиенте, никакого параллелизма, просто подключите, отправьте, восстановите и закройте (или выйдите).

#!/usr/bin/python3

import sys
import websockets

req = 'joe'
if sys.argv[1:]:
    req = sys.argv[1]

with websockets.connect("ws://localhost:9001") as ws:
    print("req: %s" % (req) )
    ws.send(req)
    res = ws.recv()
    print("res: %s" % (res) )

И когда я его запустил ...

medi@medi:~/proto/python/d4> ./client 'hello Joe'
Traceback (most recent call last):
  File "./client", line 10, in <module>
    with websockets.connect("ws://localhost:9001") as ws:
AttributeError: __enter__

Я быложидая возможности send () и recv (). У меня проблемы с поиском примера кода на основе Python (я видел много клиентов JS).