mirror of
https://github.com/ElementsProject/lightning.git
synced 2024-11-19 18:11:28 +01:00
3e3b05e1b2
`pylightning` is not much more than an alias for `pyln-client`, so this removes the need to install that as well just to run the tests.
39 lines
937 B
Python
Executable File
39 lines
937 B
Python
Executable File
#!/usr/bin/env python3
|
|
"""Simple plugin to test the connected_hook.
|
|
|
|
It can mark some node_ids as rejects and it'll check for each
|
|
connection if it should be disconnected immediately or if it can
|
|
continue.
|
|
|
|
"""
|
|
|
|
from pyln.client import Plugin
|
|
|
|
plugin = Plugin()
|
|
|
|
|
|
@plugin.hook('peer_connected')
|
|
def on_connected(peer, plugin, **kwargs):
|
|
if peer['id'] in plugin.reject_ids:
|
|
print("{} is in reject list, disconnecting".format(peer['id']))
|
|
return {'result': 'disconnect', 'error_message': 'You are in reject list'}
|
|
|
|
print("{} is allowed".format(peer['id']))
|
|
return {'result': 'continue'}
|
|
|
|
|
|
@plugin.init()
|
|
def init(configuration, options, plugin):
|
|
plugin.reject_ids = []
|
|
|
|
|
|
@plugin.method('reject')
|
|
def reject(node_id, plugin):
|
|
"""Mark a given node_id as reject for future connections.
|
|
"""
|
|
print("Rejecting connections from {}".format(node_id))
|
|
plugin.reject_ids.append(node_id)
|
|
|
|
|
|
plugin.run()
|