From 25c1c96ac741c54f8ff91b43e717eecc21253e88 Mon Sep 17 00:00:00 2001 From: Dmitry Moiseev Date: Mon, 18 Nov 2013 11:26:21 +0600 Subject: fixed exception on connection lost --- ParrotZik.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/ParrotZik.py b/ParrotZik.py index b7f5dd7..f86ca97 100644 --- a/ParrotZik.py +++ b/ParrotZik.py @@ -13,7 +13,8 @@ class ParrotZik(object): if len(service_matches) == 0: print "Couldn't find Parrot Zik" - return False + self.sock ="" + return first_match = service_matches[0] port = first_match["port"] @@ -87,7 +88,10 @@ class ParrotZik(object): return self.sendMessage(message) def sendMessage(self,message): - self.sock.send(str(message)) + try: + self.sock.send(str(message)) + except: + self.sock ="" data = self.sock.recv(7) len = struct.unpack('B', data[1])[0] data = self.sock.recv(1024) -- cgit v1.2.1