From 02da3fe94750cd3ca4afc313482d0676b5bc2c2e Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Sun, 18 Feb 2018 20:33:33 +0100 Subject: [PATCH] Handle TCP server errors --- pyrogram/session/session.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py index b86fc992..8978b4de 100644 --- a/pyrogram/session/session.py +++ b/pyrogram/session/session.py @@ -337,7 +337,10 @@ class Session: while True: packet = self.connection.recv() - if packet is None or (len(packet) == 4 and Int.read(BytesIO(packet)) == -404): + if packet is None or len(packet) == 4: + if packet: + log.warning("Server sent \"{}\"".format(Int.read(BytesIO(packet)))) + if self.is_connected.is_set(): Thread(target=self.restart, name="RestartThread").start() break