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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import pygame
import nanothreads
import nanotubes
import time
import weakref
import random
class PygameListener(nanotubes.Listener):
"""
A custom listener which tracks connections in a list. Connected sockets
are available via the connections attribute, and use the MessageProtocol
to communicate.
"""
def __init__(self, (host, port)):
nanotubes.Listener.__init__(self, (host, port), nanotubes.MessageProtocol)
self.connections = []
def broadcast(self, name, **kw):
"""
Broadcast a message to all connected sockets.
"""
for conn_ref in self.connections:
conn_ref().protocol.post(name, **kw)
def cleanup(self):
"""
Remove dead connections.
"""
connections = []
for conn_ref in self.connections:
if conn_ref() is not None:
connections.append(conn_ref)
self.connections[:] = connections
def handle_new(self, conn):
self.connections.append(weakref.ref(conn))
def main():
pygame.init()
#start a server listening for connections on port 1984
server = PygameListener(('10.1.1.55',1984))
screen = pygame.display.set_mode((320,200))
while True:
time.sleep(0.01)
#the SocketHandler must be polled regularly.
nanotubes.SocketHandler.poll()
for event in pygame.event.get():
if event.type == pygame.QUIT:
raise SystemExit
elif event.type == pygame.MOUSEMOTION:
#broadcast MOUSEMOTION events to all connected clients.
server.broadcast(event.type, **event.dict)
for conn_ref in server.connections:
#get a reference to the connection
conn = conn_ref()
#if the reference is dead, ignore it
if conn is None: continue
#recieved events are stored in .protocol.messages
for event_type, args in conn.protocol.messages:
#if recieved event is pygame.MOUSEMOTION, draw a pixel
if event_type == pygame.MOUSEMOTION:
screen.set_at(args['pos'], (255,255,255))
#remove messages from the connections queue.
conn.protocol.messages[:] = []
pygame.display.flip()
#cleanup and remove dead connections.
server.cleanup()
yield None
nanothreads.install(main())
nanothreads.loop()
|
