From 6b90332730fa6789785142a5ee16278892add155 Mon Sep 17 00:00:00 2001
From: Dmitri Bogomolov <4glitch@gmail.com>
Date: Wed, 3 Mar 2021 11:51:05 +0200
Subject: [PATCH] Pass config and state to network.start(),

don't import announcethread in the network init, use dotted imports,

work around too-many-locals pylint warning
---
 src/bitmessagemain.py   | 11 +-------
 src/network/__init__.py | 62 +++++++++++++++++++----------------------
 2 files changed, 30 insertions(+), 43 deletions(-)

diff --git a/src/bitmessagemain.py b/src/bitmessagemain.py
index ac4e229f..90ccf3b7 100755
--- a/src/bitmessagemain.py
+++ b/src/bitmessagemain.py
@@ -235,17 +235,8 @@ class Main(object):
         # start network components if networking is enabled
         if state.enableNetwork:
             start_proxyconfig()
-            network.start()
+            network.start(config, state)
 
-            # Optional components
-            for i in range(config.getint('threads', 'receive')):
-                receiveQueueThread = network.ReceiveQueueThread(i)
-                receiveQueueThread.daemon = True
-                receiveQueueThread.start()
-            if config.safeGetBoolean('bitmessagesettings', 'udp'):
-                state.announceThread = network.AnnounceThread()
-                state.announceThread.daemon = True
-                state.announceThread.start()
             if config.safeGetBoolean('bitmessagesettings', 'upnp'):
                 import upnp
                 upnpThread = upnp.uPnPThread()
diff --git a/src/network/__init__.py b/src/network/__init__.py
index ab1e3170..269bd298 100644
--- a/src/network/__init__.py
+++ b/src/network/__init__.py
@@ -2,46 +2,42 @@
 Network subsystem package
 """
 
-from announcethread import AnnounceThread
-from connectionpool import BMConnectionPool
-from receivequeuethread import ReceiveQueueThread
-from threads import StoppableThread
+from .connectionpool import BMConnectionPool
+from .threads import StoppableThread
 
 
-__all__ = [
-    "AnnounceThread", "BMConnectionPool",
-    "ReceiveQueueThread", "StoppableThread"
-    # "AddrThread", "AnnounceThread", "BMNetworkThread", "Dandelion",
-    # "DownloadThread", "InvThread", "UploadThread",
-]
+__all__ = ["BMConnectionPool", "StoppableThread"]
 
 
-def start():
+def start(config, state):
     """Start network threads"""
-    from addrthread import AddrThread
-    from dandelion import Dandelion
-    from downloadthread import DownloadThread
-    from invthread import InvThread
-    from networkthread import BMNetworkThread
-    from knownnodes import readKnownNodes
-    from uploadthread import UploadThread
+    from .addrthread import AddrThread
+    from .announcethread import AnnounceThread
+    from .dandelion import Dandelion
+    from .downloadthread import DownloadThread
+    from .invthread import InvThread
+    from .networkthread import BMNetworkThread
+    from .knownnodes import readKnownNodes
+    from .receivequeuethread import ReceiveQueueThread
+    from .uploadthread import UploadThread
 
     readKnownNodes()
     # init, needs to be early because other thread may access it early
     Dandelion()
     BMConnectionPool().connectToStream(1)
-    asyncoreThread = BMNetworkThread()
-    asyncoreThread.daemon = True
-    asyncoreThread.start()
-    invThread = InvThread()
-    invThread.daemon = True
-    invThread.start()
-    addrThread = AddrThread()
-    addrThread.daemon = True
-    addrThread.start()
-    downloadThread = DownloadThread()
-    downloadThread.daemon = True
-    downloadThread.start()
-    uploadThread = UploadThread()
-    uploadThread.daemon = True
-    uploadThread.start()
+    for thread in (
+        BMNetworkThread(), InvThread(), AddrThread(),
+        DownloadThread(), UploadThread()
+    ):
+        thread.daemon = True
+        thread.start()
+
+    # Optional components
+    for i in range(config.getint('threads', 'receive')):
+        thread = ReceiveQueueThread(i)
+        thread.daemon = True
+        thread.start()
+    if config.safeGetBoolean('bitmessagesettings', 'udp'):
+        state.announceThread = AnnounceThread()
+        state.announceThread.daemon = True
+        state.announceThread.start()