forked from LBRYCommunity/lbry-sdk
e9a1dbca4b
make cli easier to use on windows by converting keyword args to json
cli now accepts commands like:
lbrynet-cli resolve_name name=one
as well as:
lbrynet-cli resolve_name ‘{“name”: “one”}’
85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
import sys
|
|
import json
|
|
import argparse
|
|
|
|
from lbrynet.conf import API_CONNECTION_STRING
|
|
from jsonrpc.proxy import JSONRPCProxy
|
|
|
|
help_msg = "Usage: lbrynet-cli method json-args\n" \
|
|
+ "Examples: " \
|
|
+ "lbrynet-cli resolve_name '{\"name\": \"what\"}'\n" \
|
|
+ "lbrynet-cli get_balance\n" \
|
|
+ "lbrynet-cli help '{\"function\": \"resolve_name\"}'\n" \
|
|
+ "\n******lbrynet-cli functions******\n"
|
|
|
|
|
|
def guess_type(x):
|
|
if '.' in x:
|
|
try:
|
|
return float(x)
|
|
except ValueError:
|
|
# not a float
|
|
pass
|
|
try:
|
|
return int(x)
|
|
except ValueError:
|
|
return x
|
|
|
|
def main():
|
|
api = JSONRPCProxy.from_url(API_CONNECTION_STRING)
|
|
|
|
try:
|
|
s = api.is_running()
|
|
except:
|
|
print "lbrynet-daemon isn't running"
|
|
sys.exit(1)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('method', nargs=1, type=str)
|
|
parser.add_argument('params', nargs="+")
|
|
args = parser.parse_args()
|
|
meth = args.method[0]
|
|
params = {}
|
|
if args.params:
|
|
if len(args.params) != 1:
|
|
for i in args.params:
|
|
k, v = i.split('=')[0], i.split('=')[1:]
|
|
if isinstance(v, list):
|
|
v = ''.join(v)
|
|
params[k] = guess_type(v)
|
|
else:
|
|
try:
|
|
params = json.loads(args.params[0])
|
|
except ValueError:
|
|
for i in args.params:
|
|
k, v = i.split('=')[0], i.split('=')[1:]
|
|
if isinstance(v, list):
|
|
v = ''.join(v)
|
|
params[k] = guess_type(v)
|
|
|
|
msg = help_msg
|
|
for f in api.help():
|
|
msg += f + "\n"
|
|
|
|
if meth in ['--help', '-h', 'help']:
|
|
print msg
|
|
sys.exit(1)
|
|
|
|
if meth in api.help():
|
|
try:
|
|
if params:
|
|
r = api.call(meth, params)
|
|
else:
|
|
r = api.call(meth)
|
|
print json.dumps(r, sort_keys=True)
|
|
except:
|
|
print "Something went wrong, here's the usage for %s:" % meth
|
|
print api.help({'function': meth})
|
|
else:
|
|
print "Unknown function"
|
|
print msg
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|