src/api.py

@@ -73,13 +73,13 @@ from six.moves import configparser, http_client, xmlrpc_server
 
 import helper_inbox
 import helper_sent
-import protocol
 import proofofwork
 import queues
 import shared
 
 import shutdown
 import state
+import network.protocol as protocol
 from addresses import (
     addBMIfNotPresent,
     decodeAddress,

src/class_objectProcessor.py

@@ -19,10 +19,10 @@ import helper_msgcoding
 import helper_sent
 import highlevelcrypto
 import l10n
-import protocol
 import queues
 import shared
 import state
+import network.protocol as protocol
 from addresses import (
     decodeAddress, decodeVarint,
     encodeAddress, encodeVarint, varintDecodeError

src/class_singleWorker.py

@@ -20,11 +20,11 @@ import helper_sql
 import highlevelcrypto
 import l10n
 import proofofwork
-import protocol
 import queues
 import shared
 import state
 import tr
+import network.protocol as protocol
 from addresses import decodeAddress, decodeVarint, encodeVarint
 from bmconfigparser import config
 from helper_sql import sqlExecute, sqlQuery

src/network/bmobject.py

@@ -4,8 +4,8 @@ BMObject and it's exceptions.
 import logging
 import time
 
-import protocol
 import state
+import protocol
 import connectionpool
 from network import dandelion_ins
 from highlevelcrypto import calculateInventoryHash

src/network/bmproto.py

@@ -13,9 +13,9 @@ import time
 # magic imports!
 import addresses
 import knownnodes
-import protocol
 import state
 import connectionpool
+import protocol
 from bmconfigparser import config
 from queues import invQueue, objectProcessorQueue, portCheckerQueue
 from randomtrackingdict import RandomTrackingDict

src/network/connectionchooser.py

@@ -6,8 +6,8 @@ import logging
 import random
 
 import knownnodes
-import protocol
 import state
+import protocol
 from bmconfigparser import config
 from queues import queue, portCheckerQueue
 

src/network/protocol.py

@@ -13,16 +13,26 @@ import time
 from binascii import hexlify
 from struct import Struct, pack, unpack
 
-import defaults
-import highlevelcrypto
-import state
-from addresses import (
+try:
+    import defaults
+    import highlevelcrypto
+    import state
+    from addresses import (
         encodeVarint, decodeVarint, decodeAddress, varintDecodeError)
-from bmconfigparser import config
-from debug import logger
-from helper_sql import sqlExecute
-from network.node import Peer
-from version import softwareVersion
+    from bmconfigparser import config
+    from debug import logger
+    from helper_sql import sqlExecute
+    from version import softwareVersion
+except ImportError:
+    from pybitmessage import defaults, highlevelcrypto, state
+    from pybitmessage.addresses import (
+        encodeVarint, decodeVarint, decodeAddress, varintDecodeError)
+    from pybitmessage.bmconfigparser import config
+    from pybitmessage.debug import logger
+    from pybitmessage.helper_sql import sqlExecute
+    from pybitmessage.version import softwareVersion
+
+from .node import Peer
 
 # Network constants
 magic = 0xE9BEB4D9

src/tests/core.py

@@ -16,10 +16,10 @@ import threading
 import time
 import unittest
 
-import protocol
 import state
 import helper_sent
 import helper_addressbook
+import network.protocol as protocol
 
 from bmconfigparser import config
 from helper_msgcoding import MsgEncode, MsgDecode

src/tests/test_api_thread.py

@@ -7,7 +7,7 @@ from struct import pack
 
 from six.moves import queue, xmlrpc_client
 
-from pybitmessage import protocol
+from pybitmessage.network import protocol
 from pybitmessage.highlevelcrypto import calculateInventoryHash
 
 from .partial import TestPartialRun

src/tests/test_packets.py

@@ -3,7 +3,8 @@
 from binascii import unhexlify
 from struct import pack
 
-from pybitmessage import addresses, protocol
+from pybitmessage import addresses
+from pybitmessage.network import protocol
 
 from .samples import (
     sample_addr_data, sample_object_data, sample_object_expires)

src/tests/test_protocol.py

@@ -5,7 +5,8 @@ Tests for common protocol functions
 import sys
 import unittest
 
-from pybitmessage import protocol, state
+from pybitmessage import state
+from pybitmessage.network import protocol
 from pybitmessage.helper_startup import fixSocket