Removed data types

This commit is contained in:
PhilipTrauner 2015-11-02 22:34:53 +01:00
parent 3b62e1aa14
commit df36769978

View file

@ -6,12 +6,11 @@ import _thread
class Server: class Server:
def __init__(self, host, port, handler, handler_args=[]): def __init__(self, host_port_pair, handler, handler_args=[]):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.bind((host, port)) self.sock.bind(host_port_pair)
self.sock.listen(2) self.sock.listen(2)
self.handler = handler self.handler = handler
self.broadcast = Server.Broadcast()
while 1: while 1:
sock, info = self.sock.accept() sock, info = self.sock.accept()
_thread.start_new_thread(self.controller, (sock, info, handler_args)) _thread.start_new_thread(self.controller, (sock, info, handler_args))
@ -19,14 +18,11 @@ class Server:
def controller(self, sock, info, handler_args): def controller(self, sock, info, handler_args):
sock = ESock(sock) sock = ESock(sock)
handler = self.handler(sock, info, self.broadcast, **handler_args) handler = self.handler(sock, info, **handler_args)
self.broadcast.add(sock)
while 1: while 1:
try: try:
data, type = handler.sock.recv() handler.handle(sock.recv())
handler.handle(data, type)
except (socket.error, OSError): except (socket.error, OSError):
self.broadcast.remove(handler.sock)
handler.finish() handler.finish()
handler.sock.close() handler.sock.close()
_thread.exit() _thread.exit()
@ -47,22 +43,4 @@ class Server:
pass pass
def finish(self): def finish(self):
pass pass
class Broadcast:
def __init__(self):
self.socks = []
def broadcast(self, data, exclude=[]):
for sock in self.socks:
if not sock in exclude:
sock.send(data)
def remove(self, sock):
if sock in self.socks:
del self.socks[self.socks.index(sock)]
def add(self, sock):
if not sock in self.socks:
self.socks.append(sock)