forked from LBRYCommunity/lbry-sdk
122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
import sys
|
|
import json
|
|
import asyncio
|
|
import aiohttp
|
|
from docopt import docopt
|
|
from textwrap import dedent
|
|
|
|
from lbrynet.core.system_info import get_platform
|
|
from lbrynet.daemon.Daemon import Daemon
|
|
from lbrynet.daemon.DaemonControl import start
|
|
|
|
|
|
async def execute_command(command, args):
|
|
message = {'method': command, 'params': args}
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get('http://localhost:5279/lbryapi', json=message) as resp:
|
|
print(json.dumps(await resp.json(), indent=4))
|
|
|
|
|
|
def print_help():
|
|
print(dedent("""
|
|
NAME
|
|
lbry - LBRY command line client.
|
|
|
|
USAGE
|
|
lbry [--conf <config file>] <command> [<args>]
|
|
|
|
EXAMPLES
|
|
lbry commands # list available commands
|
|
lbry status # get daemon status
|
|
lbry --conf ~/l1.conf status # like above but using ~/l1.conf as config file
|
|
lbry resolve_name what # resolve a name
|
|
lbry help resolve_name # get help for a command
|
|
"""))
|
|
|
|
|
|
def print_help_for_command(command):
|
|
fn = Daemon.callable_methods.get(command)
|
|
if fn:
|
|
print(dedent(fn.__doc__))
|
|
else:
|
|
print("Invalid command name")
|
|
|
|
|
|
def guess_type(x, key=None):
|
|
if not isinstance(x, str):
|
|
return x
|
|
if key in ('uri', 'channel_name', 'name', 'file_name', 'download_directory'):
|
|
return x
|
|
if x in ('true', 'True', 'TRUE'):
|
|
return True
|
|
if x in ('false', 'False', 'FALSE'):
|
|
return False
|
|
if '.' in x:
|
|
try:
|
|
return float(x)
|
|
except ValueError:
|
|
# not a float
|
|
pass
|
|
try:
|
|
return int(x)
|
|
except ValueError:
|
|
return x
|
|
|
|
|
|
def remove_brackets(key):
|
|
if key.startswith("<") and key.endswith(">"):
|
|
return str(key[1:-1])
|
|
return key
|
|
|
|
|
|
def set_kwargs(parsed_args):
|
|
kwargs = {}
|
|
for key, arg in parsed_args.items():
|
|
k = None
|
|
if arg is None:
|
|
continue
|
|
elif key.startswith("--") and remove_brackets(key[2:]) not in kwargs:
|
|
k = remove_brackets(key[2:])
|
|
elif remove_brackets(key) not in kwargs:
|
|
k = remove_brackets(key)
|
|
kwargs[k] = guess_type(arg, k)
|
|
return kwargs
|
|
|
|
|
|
def main(argv=None):
|
|
argv = argv or sys.argv[1:]
|
|
if not argv:
|
|
print_help()
|
|
return 1
|
|
|
|
method, args = argv[0], argv[1:]
|
|
|
|
if method in ['help', '--help', '-h']:
|
|
if len(args) == 1:
|
|
print_help_for_command(args[0])
|
|
else:
|
|
print_help()
|
|
|
|
elif method in ['version', '--version', '-v']:
|
|
print(json.dumps(get_platform(get_ip=False), sort_keys=True, indent=4, separators=(',', ': ')))
|
|
|
|
|
|
elif method == 'start':
|
|
start(args)
|
|
|
|
elif method not in Daemon.callable_methods:
|
|
print('"{}" is not a valid command.'.format(method))
|
|
return 1
|
|
|
|
else:
|
|
fn = Daemon.callable_methods[method]
|
|
parsed = docopt(fn.__doc__, args)
|
|
kwargs = set_kwargs(parsed)
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(execute_command(method, kwargs))
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|