IS: added connection related logger lines

This commit is contained in:
Rossen Georgiev 2015-08-10 13:06:32 +01:00
parent 7f9c7816b1
commit 18be87d460
3 changed files with 16 additions and 8 deletions

View File

@ -114,10 +114,11 @@ class IS(object):
self._connect() self._connect()
self._send_login() self._send_login()
break break
except: except (LoginError, ConnectionError):
if not blocking: if not blocking:
raise raise
self.logger.info("Retrying connection is %d seconds." % retry)
time.sleep(retry) time.sleep(retry)
def close(self): def close(self):
@ -252,12 +253,14 @@ class IS(object):
else: else:
raise ConnectionError("invalid banner from server") raise ConnectionError("invalid banner from server")
except ConnectionError: except ConnectionError as e:
self.logger.error(str(e))
self.close() self.close()
raise raise
except (socket.error, socket.timeout) as e: except (socket.error, socket.timeout) as e:
self.close() self.close()
self.logger.error("Socket error: %s" % str(e))
if str(e) == "timed out": if str(e) == "timed out":
raise ConnectionError("no banner from server") raise ConnectionError("no banner from server")
else: else:
@ -303,12 +306,14 @@ class IS(object):
else: else:
self.logger.info("Login successful") self.logger.info("Login successful")
except LoginError: except LoginError as e:
self.logger.error(str(e))
self.close() self.close()
raise raise
except: except:
self.close() self.close()
raise LoginError("failed to login") self.logger.error("Failed to login")
raise LoginError("Failed to login")
def _socket_readlines(self, blocking=False): def _socket_readlines(self, blocking=False):
""" """
@ -317,6 +322,7 @@ class IS(object):
try: try:
self.sock.setblocking(0) self.sock.setblocking(0)
except socket.error as e: except socket.error as e:
self.logger.error("socket error when setblocking(0): %s" % str(e))
raise ConnectionDrop("connection dropped") raise ConnectionDrop("connection dropped")
while True: while True:
@ -330,8 +336,10 @@ class IS(object):
# sock.recv returns empty if the connection drops # sock.recv returns empty if the connection drops
if not short_buf: if not short_buf:
self.logger.error("socket.recv(): returned empty")
raise ConnectionDrop("connection dropped") raise ConnectionDrop("connection dropped")
except socket.error as e: except socket.error as e:
self.logger.error("socket error on recv(): %s" % str(e))
if "Resource temporarily unavailable" in str(e): if "Resource temporarily unavailable" in str(e):
if not blocking: if not blocking:
if len(self.buf) == 0: if len(self.buf) == 0:

View File

@ -35,7 +35,7 @@ except ImportError:
return {'confidence': 0.0, 'encoding': 'windows-1252'} return {'confidence': 0.0, 'encoding': 'windows-1252'}
from .exceptions import (UnknownFormat, ParseError) from .exceptions import (UnknownFormat, ParseError)
from . import base91, string_type_parse, is_py3 from . import base91, string_type_parse
__all__ = ['parse'] __all__ = ['parse']
@ -98,7 +98,7 @@ def parse(packet):
""" """
if not isinstance(packet, string_type_parse): if not isinstance(packet, string_type_parse):
raise TypeError("Epected packet to be str/unicode/bytes, got %s", type(packet)) raise TypeError("Expected packet to be str/unicode/bytes, got %s", type(packet))
# attempt to detect encoding # attempt to detect encoding
if isinstance(packet, bytes): if isinstance(packet, bytes):

View File

@ -252,9 +252,9 @@ class TC_IS(unittest.TestCase):
def test_connect_raising_exceptions(self): def test_connect_raising_exceptions(self):
self.m.StubOutWithMock(self.ais, "_connect") self.m.StubOutWithMock(self.ais, "_connect")
self.m.StubOutWithMock(self.ais, "_send_login") self.m.StubOutWithMock(self.ais, "_send_login")
self.ais._connect().AndRaise(Exception("first")) self.ais._connect().AndRaise(aprslib.exceptions.ConnectionError("first"))
self.ais._connect() self.ais._connect()
self.ais._send_login().AndRaise(Exception("second")) self.ais._send_login().AndRaise(aprslib.exceptions.LoginError("second"))
self.ais._connect() self.ais._connect()
self.ais._send_login() self.ais._send_login()
self.m.ReplayAll() self.m.ReplayAll()