2a461bd98c
Add new RPC command requests. Adjust sessions RPC command to show pending requests. Only keep first 15 chars in version string. Set socket timeout Try more forceful closing of socket Fixes #67
85 lines
2.6 KiB
Python
Executable File
85 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (c) 2016, Neil Booth
|
|
#
|
|
# All rights reserved.
|
|
#
|
|
# See the file "LICENCE" for information about the copyright
|
|
# and warranty status of this software.
|
|
|
|
'''Script to send RPC commands to a running ElectrumX server.'''
|
|
|
|
|
|
import argparse
|
|
import asyncio
|
|
import json
|
|
from functools import partial
|
|
from os import environ
|
|
|
|
from lib.jsonrpc import JSONRPC
|
|
from server.protocol import ServerManager
|
|
|
|
|
|
class RPCClient(JSONRPC):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.queue = asyncio.Queue()
|
|
|
|
def enqueue_request(self, request):
|
|
self.queue.put_nowait(request)
|
|
|
|
async def send_and_wait(self, method, params, timeout=None):
|
|
# Raise incoming buffer size - presumably connection is trusted
|
|
self.max_buffer_size = 5000000
|
|
payload = self.request_payload(method, id_=method, params=params)
|
|
self.encode_and_send_payload(payload)
|
|
|
|
future = asyncio.ensure_future(self.queue.get())
|
|
for f in asyncio.as_completed([future], timeout=timeout):
|
|
try:
|
|
request = await f
|
|
except asyncio.TimeoutError:
|
|
future.cancel()
|
|
print('request timed out after {}s'.format(timeout))
|
|
else:
|
|
await request.process(1)
|
|
|
|
async def handle_response(self, result, error, method):
|
|
if result and method == 'sessions':
|
|
for line in ServerManager.sessions_text_lines(result):
|
|
print(line)
|
|
else:
|
|
value = {'error': error} if error else result
|
|
print(json.dumps(value, indent=4, sort_keys=True))
|
|
|
|
|
|
def main():
|
|
'''Send the RPC command to the server and print the result.'''
|
|
parser = argparse.ArgumentParser('Send electrumx an RPC command' )
|
|
parser.add_argument('-p', '--port', metavar='port_num', type=int,
|
|
help='RPC port number')
|
|
parser.add_argument('command', nargs=1, default=[],
|
|
help='command to send')
|
|
parser.add_argument('param', nargs='*', default=[],
|
|
help='params to send')
|
|
args = parser.parse_args()
|
|
|
|
if args.port is None:
|
|
args.port = int(environ.get('RPC_PORT', 8000))
|
|
|
|
loop = asyncio.get_event_loop()
|
|
coro = loop.create_connection(RPCClient, 'localhost', args.port)
|
|
try:
|
|
transport, protocol = loop.run_until_complete(coro)
|
|
coro = protocol.send_and_wait(args.command[0], args.param, timeout=15)
|
|
loop.run_until_complete(coro)
|
|
except OSError:
|
|
print('error connecting - is ElectrumX catching up or not running?')
|
|
finally:
|
|
loop.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|