Shorten Bootstrapper methods:

handle_close() and set_connection_fully_established()
navjotcis/bug-fixes18
Dmitri Bogomolov 3 years ago
parent d6cab9935d
commit 3108115570
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13

@ -366,16 +366,21 @@ def bootstrap(connection_class):
"""
BMProto.bm_command_addr(self)
self._succeed = True
# pylint: disable=attribute-defined-outside-init
self.close_reason = "Thanks for bootstrapping!"
self.set_state("close")
def set_connection_fully_established(self):
"""Only send addr here"""
# pylint: disable=attribute-defined-outside-init
self.fullyEstablished = True
self.sendAddr()
def handle_close(self):
"""
After closing the connection switch knownnodes.knownNodesActual
back to False if the bootstrapper failed.
"""
self._connection_base.handle_close(self)
BMProto.handle_close(self)
if not self._succeed:
knownnodes.knownNodesActual = False

Loading…
Cancel
Save