0g-storage-node/tests/utility/simple_rpc_proxy.py
peilun-conflux d80e7e22ca
Prune no reward chunks. (#145)
* Prune no reward chunks.

* Add tests.

* Fix tests.

* Fix clippy.

* Revert test.

* Enable market in shard_sync_test.

* Add tx prune status.

* Fix tests.
2024-08-06 15:06:15 +08:00

32 lines
1008 B
Python

from jsonrpcclient import request, parse, Ok
import requests
class SimpleRpcProxy:
def __init__(self, url, timeout=60):
self.url = url
self.timeout = timeout
def __getattr__(self, name):
return RpcCaller(self.url, name, self.timeout)
class RpcCaller:
def __init__(self, url, method, timeout):
self.url = url
self.method = method
self.timeout = timeout
def __call__(self, *args, **argsn):
r = request(self.method, *args)
try:
response = requests.post(self.url, json=r, timeout=self.timeout)
parsed = parse(response.json())
if isinstance(parsed, Ok):
return parsed.result
else:
print("Failed to call RPC, method = %s(%s), error = %s" % (self.method, str(*args)[-1500:], parsed))
except Exception as ex:
print("Failed to call RPC, method = %s(%s), exception = %s" % (self.method, str(*args)[-1500:], ex))
return None