forked from M-Labs/artiq
46 lines
1.1 KiB
Plaintext
46 lines
1.1 KiB
Plaintext
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import socket
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
def _get_args():
|
||
|
|
parser = argparse.ArgumentParser(description="ARTIQ client")
|
||
|
|
parser.add_argument(
|
||
|
|
"-o", "--run-once", default=[], nargs=3,
|
||
|
|
action="append",
|
||
|
|
help="run experiment once. arguments: <path> <name> <timeout>")
|
||
|
|
parser.add_argument(
|
||
|
|
"-q", "--quit-master", default=False,
|
||
|
|
action="store_true",
|
||
|
|
help="causes the master to quit")
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def _send_obj(sock, obj):
|
||
|
|
line = json.dumps(obj) + "\n"
|
||
|
|
sock.sendall(line.encode())
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args = _get_args()
|
||
|
|
|
||
|
|
with socket.create_connection(("::1", 8888)) as sock:
|
||
|
|
for path, name, timeout in args.run_once:
|
||
|
|
obj = {
|
||
|
|
"action": "run_once",
|
||
|
|
"run_params": {
|
||
|
|
"path": path,
|
||
|
|
"name": name
|
||
|
|
},
|
||
|
|
"timeout": timeout
|
||
|
|
}
|
||
|
|
_send_obj(sock, obj)
|
||
|
|
if args.quit_master:
|
||
|
|
obj = {"action": "quit"}
|
||
|
|
_send_obj(sock, obj)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|