Client¶
ws4py comes with various client implementations and they roughly share the same interface.
Built-in¶
The built-in client relies only on modules provided by the Python stdlib. The client’s inner loop runs within a thread and therefore holds the thread alive until the websocket is closed.
1from ws4py.client.threadedclient import WebSocketClient
2
3class DummyClient(WebSocketClient):
4 def opened(self):
5 def data_provider():
6 for i in range(1, 200, 25):
7 yield "#" * i
8
9 self.send(data_provider())
10
11 for i in range(0, 200, 25):
12 print i
13 self.send("*" * i)
14
15 def closed(self, code, reason=None):
16 print "Closed down", code, reason
17
18 def received_message(self, m):
19 print m
20 if len(m) == 175:
21 self.close(reason='Bye bye')
22
23if __name__ == '__main__':
24 try:
25 ws = DummyClient('ws://localhost:9000/', protocols=['http-only', 'chat'])
26 ws.connect()
27 ws.run_forever()
28 except KeyboardInterrupt:
29 ws.close()
In this snippet, when the handshake is successful, the opened()
method is called and within this method we immediately send a bunch of messages to the server. First we demonstrate how you can use a generator to do so, then we simply send strings.
Assuming the server echoes messages as they arrive, the received_message(message)
method will print out the messages returned by the server and simply close the connection once it receives the last sent messages, which length is 175.
Finally the closed(code, reason=None)
method is called with the code and reason given by the server.
See also
Tornado¶
If you are using a Tornado backend you may use the Tornado client that ws4py provides as follow:
from ws4py.client.tornadoclient import TornadoWebSocketClient
from tornado import ioloop
class MyClient(TornadoWebSocketClient):
def opened(self):
for i in range(0, 200, 25):
self.send("*" * i)
def received_message(self, m):
print m
if len(m) == 175:
self.close(reason='Bye bye')
def closed(self, code, reason=None):
ioloop.IOLoop.instance().stop()
ws = MyClient('ws://localhost:9000/echo', protocols=['http-only', 'chat'])
ws.connect()
ioloop.IOLoop.instance().start()
gevent¶
If you are using a gevent backend you may use the gevent client that ws4py provides as follow:
from ws4py.client.geventclient import WebSocketClient
This client can benefit from gevent’s concepts as demonstrated below:
ws = WebSocketClient('ws://localhost:9000/echo', protocols=['http-only', 'chat'])
ws.connect()
def incoming():
"""
Greenlet waiting for incoming messages
until ``None`` is received, indicating we can
leave the loop.
"""
while True:
m = ws.receive()
if m is not None:
print str(m)
else:
break
def send_a_bunch():
for i in range(0, 40, 5):
ws.send("*" * i)
greenlets = [
gevent.spawn(incoming),
gevent.spawn(send_a_bunch),
]
gevent.joinall(greenlets)