Init singleWorker with queue and inventory

Author: g1itch

src/class_singleWorker.py

@@ -12,6 +12,8 @@
 from struct import pack
 from subprocess import call  # nosec
 
+from six.moves import configparser, queue
+
 import defaults
 import helper_inbox
 import helper_msgcoding
@@ -30,9 +32,7 @@
 )
 from bmconfigparser import BMConfigParser
 from helper_sql import sqlExecute, sqlQuery
-from inventory import Inventory
 from network import knownnodes, StoppableThread
-from six.moves import configparser, queue
 
 
 def sizeof_fmt(num, suffix='h/s'):
@@ -48,15 +48,17 @@ def sizeof_fmt(num, suffix='h/s'):
 class singleWorker(StoppableThread):
     """Thread for performing PoW"""
 
-    def __init__(self):
+    def __init__(self, queue, inventory):
         super(singleWorker, self).__init__(name="singleWorker")
+        self.inventory = inventory
+        self.queue = queue
         proofofwork.init()
 
     def stopThread(self):
         """Signal through the queue that the thread should be stopped"""
 
         try:
-            queues.workerQueue.put(("stopThread", "data"))
+            self.queue.put(("stopThread", "data"))
         except queue.Full:
             self.logger.error('workerQueue is Full')
         super(singleWorker, self).stopThread()
@@ -119,7 +121,8 @@ def run(self):
         # For the case if user deleted knownnodes
         # but is still having onionpeer objects in inventory
         if not knownnodes.knownNodesActual:
-            for item in Inventory().by_type_and_tag(protocol.OBJECT_ONIONPEER):
+            for item in self.inventory.by_type_and_tag(
+                    protocol.OBJECT_ONIONPEER):
                 queues.objectProcessorQueue.put((
                     protocol.OBJECT_ONIONPEER, item.payload
                 ))
@@ -134,17 +137,17 @@ def run(self):
 
         # just in case there are any pending tasks for msg
         # messages that have yet to be sent.
-        queues.workerQueue.put(('sendmessage', ''))
+        self.queue.put(('sendmessage', ''))
         # just in case there are any tasks for Broadcasts
         # that have yet to be sent.
-        queues.workerQueue.put(('sendbroadcast', ''))
+        self.queue.put(('sendbroadcast', ''))
 
         # send onionpeer object
-        queues.workerQueue.put(('sendOnionPeerObj', ''))
+        self.queue.put(('sendOnionPeerObj', ''))
 
         while state.shutdown == 0:
             self.busy = 0
-            command, data = queues.workerQueue.get()
+            command, data = self.queue.get()
             self.busy = 1
             if command == 'sendmessage':
                 try:
@@ -191,7 +194,7 @@ def run(self):
                     command
                 )
 
-            queues.workerQueue.task_done()
+            self.queue.task_done()
         self.logger.info("Quitting...")
 
     def _getKeysForAddress(self, address):
@@ -290,13 +293,12 @@ def doPOWForMyV2Pubkey(self, adressHash):
 
         inventoryHash = calculateInventoryHash(payload)
         objectType = 1
-        Inventory()[inventoryHash] = (
-            objectType, streamNumber, payload, embeddedTime, '')
+        self.inventory.put(
+            inventoryHash, objectType, streamNumber, payload, embeddedTime)
 
         self.logger.info(
             'broadcasting inv with hash: %s', hexlify(inventoryHash))
 
-        queues.invQueue.put((streamNumber, inventoryHash))
         queues.UISignalQueue.put(('updateStatusBar', ''))
         try:
             BMConfigParser().set(
@@ -378,13 +380,11 @@ def sendOutOrStoreMyV3Pubkey(self, adressHash):
 
         inventoryHash = calculateInventoryHash(payload)
         objectType = 1
-        Inventory()[inventoryHash] = (
-            objectType, streamNumber, payload, embeddedTime, '')
+        self.inventory.put(objectType, streamNumber, payload, embeddedTime)
 
         self.logger.info(
             'broadcasting inv with hash: %s', hexlify(inventoryHash))
 
-        queues.invQueue.put((streamNumber, inventoryHash))
         queues.UISignalQueue.put(('updateStatusBar', ''))
         try:
             BMConfigParser().set(
@@ -471,15 +471,13 @@ def sendOutOrStoreMyV4Pubkey(self, myAddress):
 
         inventoryHash = calculateInventoryHash(payload)
         objectType = 1
-        Inventory()[inventoryHash] = (
+        self.inventory.put(
             objectType, streamNumber, payload, embeddedTime,
-            doubleHashOfAddressData[32:]
-        )
+            doubleHashOfAddressData[32:])
 
         self.logger.info(
             'broadcasting inv with hash: %s', hexlify(inventoryHash))
 
-        queues.invQueue.put((streamNumber, inventoryHash))
         queues.UISignalQueue.put(('updateStatusBar', ''))
         try:
             BMConfigParser().set(
@@ -507,7 +505,7 @@ def sendOnionPeerObj(self, peer=None):
         objectPayload = encodeVarint(peer.port) + protocol.encodeHost(peer.host)
         tag = calculateInventoryHash(objectPayload)
 
-        if Inventory().by_type_and_tag(objectType, tag):
+        if self.inventory.by_type_and_tag(objectType, tag):
             return  # not expired
 
         payload = pack('>Q', embeddedTime)
@@ -520,14 +518,14 @@ def sendOnionPeerObj(self, peer=None):
             payload, TTL, log_prefix='(For onionpeer object)')
 
         inventoryHash = calculateInventoryHash(payload)
-        Inventory()[inventoryHash] = (
+        self.inventory.put(
             objectType, streamNumber, buffer(payload),
-            embeddedTime, buffer(tag)
-        )
+            embeddedTime, buffer(tag))
+
         self.logger.info(
             'sending inv (within sendOnionPeerObj function) for object: %s',
             hexlify(inventoryHash))
-        queues.invQueue.put((streamNumber, inventoryHash))
+        queues.UISignalQueue.put(('updateStatusBar', ''))
 
     def sendBroadcast(self):
         """Send a broadcast-type object (assemble the object, perform PoW and put it to the inv announcement queue)"""
@@ -688,14 +686,14 @@ def sendBroadcast(self):
 
             inventoryHash = calculateInventoryHash(payload)
             objectType = 3
-            Inventory()[inventoryHash] = (
+            self.inventory.put(
                 objectType, streamNumber, payload, embeddedTime, tag)
+
             self.logger.info(
                 'sending inv (within sendBroadcast function)'
                 ' for object: %s',
                 hexlify(inventoryHash)
             )
-            queues.invQueue.put((streamNumber, inventoryHash))
 
             queues.UISignalQueue.put((
                 'updateSentItemStatusByAckdata', (
@@ -847,7 +845,8 @@ def sendMsg(self):
                                     hexlify(privEncryptionKey))
                             )
 
-                            for value in Inventory().by_type_and_tag(1, toTag):
+                            for value in self.inventory.by_type_and_tag(
+                                    1, toTag):
                                 # if valid, this function also puts it
                                 # in the pubkeys table.
                                 if protocol.decryptAndCheckPubkeyPayload(
@@ -1303,8 +1302,9 @@ def sendMsg(self):
 
             inventoryHash = calculateInventoryHash(encryptedPayload)
             objectType = 2
-            Inventory()[inventoryHash] = (
-                objectType, toStreamNumber, encryptedPayload, embeddedTime, '')
+            self.inventory.put(
+                objectType, toStreamNumber, encryptedPayload, embeddedTime)
+
             if BMConfigParser().has_section(toaddress) or \
                not protocol.checkBitfield(behaviorBitfield, protocol.BITFIELD_DOESACK):
                 queues.UISignalQueue.put((
@@ -1329,7 +1329,6 @@ def sendMsg(self):
                 'Broadcasting inv for my msg(within sendmsg function): %s',
                 hexlify(inventoryHash)
             )
-            queues.invQueue.put((toStreamNumber, inventoryHash))
 
             # Update the sent message in the sent table with the
             # necessary information.
@@ -1461,10 +1460,10 @@ def requestPubKey(self, toAddress):
 
         inventoryHash = calculateInventoryHash(payload)
         objectType = 1
-        Inventory()[inventoryHash] = (
-            objectType, streamNumber, payload, embeddedTime, '')
+        self.inventory.put(
+            objectType, streamNumber, payload, embeddedTime)
+
         self.logger.info('sending inv (for the getpubkey message)')
-        queues.invQueue.put((streamNumber, inventoryHash))
 
         # wait 10% past expiration
         sleeptill = int(time.time() + TTL * 1.1)

src/inventory.py

@@ -4,6 +4,8 @@
 import storage.filesystem
 import storage.sqlite
 from bmconfigparser import BMConfigParser
+# TODO: init with queue
+from queues import invQueue
 from singleton import Singleton
 
 
@@ -39,3 +41,7 @@ def __getattr__(self, attr):
     # hint for pylint: this is dictionary like object
     def __getitem__(self, key):
         return self._realInventory[key]
+
+    def put(self, invhash, obj_type, stream, payload, embedded_time, tag=''):
+        self[invhash] = (obj_type, stream, payload, embedded_time, tag)
+        invQueue.put((stream, invhash))