handle remote exception on routing table ping

lbryio · Dec 1, 2022 · 3669beb · 3669beb
1 parent 001819d
commit 3669beb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lbry/dht/protocol/routing_table.py b/lbry/dht/protocol/routing_table.py
@@ -8,6 +8,7 @@
 
 from lbry import utils
 from lbry.dht import constants
+from lbry.dht.error import RemoteException
 from lbry.dht.protocol.distance import Distance
 if typing.TYPE_CHECKING:
     from lbry.dht.peer import KademliaPeer, PeerManager
@@ -395,7 +396,7 @@ async def add_peer(self, peer: 'KademliaPeer', probe: typing.Callable[['Kademlia
             try:
                 await probe(to_replace)
                 return False
-            except asyncio.TimeoutError:
+            except (asyncio.TimeoutError, RemoteException):
                 log.debug("Replacing dead contact in bucket %i: %s:%i with %s:%i ", bucket_index,
                           to_replace.address, to_replace.udp_port, peer.address, peer.udp_port)
                 if to_replace in self.buckets[bucket_index]: