diff --git a/wii-pair/daemon.py b/wii-pair/daemon.py index f538089..44b9f7e 100755 --- a/wii-pair/daemon.py +++ b/wii-pair/daemon.py @@ -173,26 +173,29 @@ try: conn, addr = s.accept() while True: - m = conn.recv(1) + try: + m = conn.recv(1) - if len(m) == 0: - conn.close() - break + if len(m) == 0: + conn.close() + break - r = b'' - for w in wiis: - if w.pos_x is None: - v = -128 - else: - v = int(round(w.pos_x*100)) - r += struct.pack('b', v) - if w.pos_y is None: - v = -128 - else: - v = int(round(w.pos_y*100)) - r += struct.pack('b', v) + r = b'' + for w in wiis: + if w.pos_x is None: + v = -128 + else: + v = int(round(w.pos_x*100)) + r += struct.pack('b', v) + if w.pos_y is None: + v = -128 + else: + v = int(round(w.pos_y*100)) + r += struct.pack('b', v) - conn.send(r) + conn.send(r) + except ConnectionResetError: + continue except (KeyboardInterrupt, SystemExit): t1.stop = True t2.stop = True