blob: 25da5d90e91ea0b803b80958757f63c6dc66772f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
"""The display part of a simply two process chat app."""
# This file has been placed in the public domain.
import zmq
from zmq.utils.win32 import allow_interrupt
def main(addrs):
context = zmq.Context()
control = context.socket(zmq.PUB)
control.bind('inproc://control')
updates = context.socket(zmq.SUB)
updates.setsockopt(zmq.SUBSCRIBE, "")
updates.connect('inproc://control')
for addr in addrs:
print "Connecting to: ", addr
updates.connect(addr)
def interrupt_polling():
"""Fix CTRL-C on Windows using "self pipe trick"."""
control.send_multipart(['', 'quit'])
with allow_interrupt(interrupt_polling):
message = ''
while message != 'quit':
message = updates.recv_multipart()
if len(message) < 2:
print 'Invalid message.'
continue
account = message[0]
message = ' '.join(message[1:])
if message == 'quit':
print 'Killed by "%s".' % account
break
print '%s: %s' % (account, message)
if __name__ == '__main__':
import sys
if len(sys.argv) < 2:
print "usage: display.py <address> [,<address>...]"
raise SystemExit
main(sys.argv[1:])
|