cis-kuldeep/PyBitmessage-2021-04-27
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Shorten Bootstrapper methods:

Browse Source 
handle_close() and set_connection_fully_established()
This commit is contained in:
Dmitri Bogomolov 2020-11-08 16:51:18 +02:00
parent d6cab9935d
commit 3108115570
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/network/tcp.py
View File

@ -366,16 +366,21 @@ def bootstrap(connection_class):
""" """
BMProto.bm_command_addr(self) BMProto.bm_command_addr(self)
self._succeed = True self._succeed = True
# pylint: disable=attribute-defined-outside-init
self.close_reason = "Thanks for bootstrapping!" self.close_reason = "Thanks for bootstrapping!"
self.set_state("close") 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): def handle_close(self):
""" """
After closing the connection switch knownnodes.knownNodesActual After closing the connection switch knownnodes.knownNodesActual
back to False if the bootstrapper failed. back to False if the bootstrapper failed.
""" """
self._connection_base.handle_close(self) BMProto.handle_close(self)
if not self._succeed: if not self._succeed:
knownnodes.knownNodesActual = False knownnodes.knownNodesActual = False