Broadcast now uses handler instances instead of socks

This commit is contained in:
PhilipTrauner 2016-03-21 00:15:00 +01:00
parent e28e8f51fd
commit 444f73b3f8

View file

@ -8,23 +8,23 @@ class Broadcast:
def broadcast(self, data, route, channel, exclude=[]): def broadcast(self, data, route, channel, exclude=[]):
if channel in self.channels: if channel in self.channels:
for sock in self.channels[channel]: for handler in self.channels[channel]:
if not sock in exclude: if not handler in exclude:
sock.send(data, route) handler.sock.send(data, route)
else: else:
raise Broadcast.ChannelError(channel) raise Broadcast.ChannelError(channel)
def remove(self, sock, channel): def remove(self, handler, channel):
if channel in self.channels: if channel in self.channels:
if sock in self.channels[channel]: if handler in self.channels[channel]:
del self.channels[channel][self.channels[channel].index(sock)] del self.channels[channel][self.channels[channel].index(handler)]
else: else:
raise Broadcast.ChannelError(channel) raise Broadcast.ChannelError(channel)
def add(self, sock, channel): def add(self, handler, channel):
if channel in self.channels: if channel in self.channels:
if not sock in self.channels[channel]: if not handler in self.channels[channel]:
self.channels[channel].append(sock) self.channels[channel].append(handler)
else: else:
raise Broadcast.ChannelError(channel) raise Broadcast.ChannelError(channel)