diff --git a/bert.py b/bert.py index c6e3f17..52111ef 100755 --- a/bert.py +++ b/bert.py @@ -140,10 +140,14 @@ def me(chan, msg): def getData(): - raw = irc.recv(4096) + try: + raw = irc.recv(4096) + except TimeoutError: + print("Connection timed out!") + sys.exit(0) if not raw: - print("Connection lost!") - sys.exit(0) + print("Connection lost!") + sys.exit(0) try: data = raw.decode("UTF-8") except: