73 lines
1.8 KiB
Text
73 lines
1.8 KiB
Text
|
#!/usr/bin/env python3
|
||
|
|
'Folding@home v8 Client command line control'
|
||
|
|
|
||
|
|
# /// script
|
||
|
|
# requires-python = ">=3.6"
|
||
|
|
# dependencies = ["websocket-client"]
|
||
|
|
# ///
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
from urllib.parse import urlparse
|
||
|
|
|
||
|
|
try:
|
||
|
|
from websocket import create_connection
|
||
|
|
except Exception as e:
|
||
|
|
print('Missing python3-websocket library', file = sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
# Parse args
|
||
|
|
epilog = '''\
|
||
|
|
Examples:
|
||
|
|
fahctl fold
|
||
|
|
fahctl finish
|
||
|
|
fahctl pause
|
||
|
|
'''
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description = __doc__,
|
||
|
|
formatter_class = argparse.RawDescriptionHelpFormatter, epilog = epilog)
|
||
|
|
|
||
|
|
parser.add_argument('command',
|
||
|
|
choices = ['fold', 'pause', 'finish', 'state', 'groups'],
|
||
|
|
help = 'The command to send to the client')
|
||
|
|
parser.add_argument('group', nargs = '?', metavar='GROUP',
|
||
|
|
help = 'Optional target resource group name')
|
||
|
|
parser.add_argument('-a', '--address', default = '127.0.0.1:7396',
|
||
|
|
help = 'Client address (default: %(default)s)')
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
u = urlparse('ws://' + args.address)
|
||
|
|
host = u.hostname or '127.0.0.1'
|
||
|
|
port = u.port or 7396
|
||
|
|
url = f'ws://{host}:{port}/api/websocket'
|
||
|
|
|
||
|
|
# Connect
|
||
|
|
try:
|
||
|
|
ws = create_connection(url)
|
||
|
|
except Exception as e:
|
||
|
|
print(f'{e}: {url}', file = sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
data_str = ws.recv()
|
||
|
|
data = json.loads(data_str)
|
||
|
|
groups = list(data.get("groups", {}).keys())
|
||
|
|
|
||
|
|
if args.group is not None and args.group not in groups:
|
||
|
|
print(f'Group "{args.group}" does not exist', file = sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
# Send command
|
||
|
|
if args.command == 'state': print(data_str)
|
||
|
|
elif args.command == 'groups': print(json.dumps(groups))
|
||
|
|
else:
|
||
|
|
msg = dict(cmd = 'state', state = args.command)
|
||
|
|
if args.group is not None: msg['group'] = args.group
|
||
|
|
ws.send(json.dumps(msg))
|
||
|
|
|
||
|
|
|
||
|
|
# Close
|
||
|
|
ws.close()
|