mirror of
https://github.com/0glabs/0g-storage-node.git
synced 2025-04-02 22:45:41 +00:00
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
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
|