From 439ff65f14e5eacc13327e06c437a3c00d412e54 Mon Sep 17 00:00:00 2001 From: Christian Decker Date: Thu, 22 Aug 2019 16:27:01 +0200 Subject: [PATCH] probe: Allow probing for a specific node Signed-off-by: Christian Decker --- probe/probe.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/probe/probe.py b/probe/probe.py index c7a54dd..0142e44 100755 --- a/probe/probe.py +++ b/probe/probe.py @@ -86,17 +86,18 @@ def start_probe(plugin): @plugin.method('probe') -def probe(plugin): +def probe(request, plugin, node_id=None): res = None - nodes = plugin.rpc.listnodes()['nodes'] - dst = choice(nodes) + if node_id is None: + nodes = plugin.rpc.listnodes()['nodes'] + node_id = choice(nodes)['nodeid'] s = plugin.Session() - p = Probe(destination=dst['nodeid'], started_at=datetime.now()) + p = Probe(destination=node_id, started_at=datetime.now()) s.add(p) try: route = plugin.rpc.getroute( - dst['nodeid'], + node_id, msatoshi=10000, riskfactor=1, exclude=exclusions + list(temporary_exclusions.keys()) @@ -126,7 +127,7 @@ def probe(plugin): ) exclusions.append(exclusion) - if p.failcode == 4103: + if p.failcode in [21, 4103]: exclusion = "{erring_channel}/{erring_direction}".format(**error) print('Adding temporary exclusion for channel {} ({} total))'.format( exclusion, len(temporary_exclusions))