Ignore self node in connectionchooser.chooseConnection()

2019-07-29 14:37:56 +03:00 · 2019-07-29 14:37:56 +03:00 · bdb09c2d00
commit bdb09c2d00
parent 7215003c6f
1 changed files with 5 additions and 1 deletions
--- a/src/network/connectionchooser.py
+++ b/src/network/connectionchooser.py
@ -1,3 +1,4 @@
+# pylint: disable=too-many-branches
 import random  # nosec

 import knownnodes
@ -38,7 +39,10 @@ def chooseConnection(stream):
    for _ in range(50):
        peer = random.choice(knownnodes.knownNodes[stream].keys())
        try:
-            rating = knownnodes.knownNodes[stream][peer]['rating']
+            peer_info = knownnodes.knownNodes[stream][peer]
+            if peer_info.get('self'):
+                continue
+            rating = peer_info["rating"]
        except TypeError:
            logger.warning('Error in %s', peer)
            rating = 0