Shorten Bootstrapper methods:

handle_close() and set_connection_fully_established()
This commit is contained in:
Dmitri Bogomolov 2020-11-08 16:51:18 +02:00 committed by Muzahid
parent 1b198e9a4a
commit 3feea1ee3f
Signed by untrusted user: cis-muzahid
GPG Key ID: 1DC85E7D3AB613EA

View File

@ -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