diff options
-rw-r--r-- | nbxmpp/tcp.py | 2 | ||||
-rw-r--r-- | nbxmpp/websocket.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/nbxmpp/tcp.py b/nbxmpp/tcp.py index 7c6b6b7..e661a8c 100644 --- a/nbxmpp/tcp.py +++ b/nbxmpp/tcp.py @@ -365,6 +365,8 @@ class TCPConnection(Connection): self._con.get_socket().shutdown(True, True) except GLib.Error as error: self._log.info(error) + self._input_closed = True + self._output_closed = True self.state = TCPState.DISCONNECTED self.notify(signal_name) self.destroy() diff --git a/nbxmpp/websocket.py b/nbxmpp/websocket.py index 82b72eb..9459e12 100644 --- a/nbxmpp/websocket.py +++ b/nbxmpp/websocket.py @@ -169,6 +169,8 @@ class WebsocketConnection(Connection): self._output_closed = True def _finalize(self, signal_name): + self._input_closed = True + self._output_closed = True self.state = TCPState.DISCONNECTED self.notify(signal_name) self.destroy() |