88 lines
2.6 KiB
Python
88 lines
2.6 KiB
Python
|
import jupyter_client as client
|
||
|
|
import threading
|
||
|
|
import json
|
||
|
|
import sys
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
find_connection_file = client.find_connection_file
|
||
|
|
|
||
|
|
semaphore = threading.Semaphore(value=0)
|
||
|
|
interested_lock = threading.Lock()
|
||
|
|
interested = []
|
||
|
|
|
||
|
|
def msg_router(io, shell):
|
||
|
|
while True:
|
||
|
|
msg = io()
|
||
|
|
msg['channel'] = 'io'
|
||
|
|
msgid = msg['parent_header'].get('msg_id', None)
|
||
|
|
with interested_lock:
|
||
|
|
if msgid not in interested:
|
||
|
|
continue
|
||
|
|
print(json.dumps(msg, default=str))
|
||
|
|
if (msg.get('msg_type', '') == 'status' and
|
||
|
|
msg['content']['execution_state'] == 'idle'):
|
||
|
|
break
|
||
|
|
|
||
|
|
while True:
|
||
|
|
msg = shell()
|
||
|
|
msg['channel'] = 'shell'
|
||
|
|
msgid = msg['parent_header'].get('msg_id', None)
|
||
|
|
with interested_lock:
|
||
|
|
if msgid not in interested:
|
||
|
|
continue
|
||
|
|
print(json.dumps(msg, default=str))
|
||
|
|
if msg.get('msg_type', '') in ['execute_reply',
|
||
|
|
'inspect_reply',
|
||
|
|
'complete_reply']:
|
||
|
|
semaphore.release()
|
||
|
|
|
||
|
|
def create_client(name):
|
||
|
|
if name.endswith('.json'):
|
||
|
|
cf = find_connection_file(name)
|
||
|
|
else:
|
||
|
|
cf = find_connection_file('emacs-' + name)
|
||
|
|
c = client.BlockingKernelClient(connection_file=cf)
|
||
|
|
c.load_connection_file()
|
||
|
|
c.start_channels()
|
||
|
|
io, shell = c.get_iopub_msg, c.get_shell_msg
|
||
|
|
t = threading.Thread(target=msg_router, args=(io, shell))
|
||
|
|
t.setDaemon(True)
|
||
|
|
t.start()
|
||
|
|
return c
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('--conn-file')
|
||
|
|
parser.add_argument('--execute', action='store_true')
|
||
|
|
parser.add_argument('--inspect', action='store_true')
|
||
|
|
parser.add_argument('--complete', action='store_true')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
c = create_client(args.conn_file)
|
||
|
|
|
||
|
|
with interested_lock:
|
||
|
|
if args.execute:
|
||
|
|
msgid = c.execute(sys.stdin.read(), allow_stdin=False)
|
||
|
|
interested.append(msgid)
|
||
|
|
|
||
|
|
elif args.inspect:
|
||
|
|
req = json.loads(sys.stdin.read())
|
||
|
|
code = req['code']
|
||
|
|
msgid = c.inspect(code,
|
||
|
|
cursor_pos=req.get('pos', len(code)),
|
||
|
|
detail_level=req.get('detail', 0))
|
||
|
|
interested.append(msgid)
|
||
|
|
|
||
|
|
elif args.complete:
|
||
|
|
req = json.loads(sys.stdin.read())
|
||
|
|
code = req['code']
|
||
|
|
pos = req.get('pos', len(code))
|
||
|
|
# causes things to hang as kernel doesn't come back with a
|
||
|
|
# complete_reply
|
||
|
|
if code[pos-1] in ['\n', '\r']:
|
||
|
|
sys.exit(0)
|
||
|
|
msgid = c.complete(code,
|
||
|
|
cursor_pos=pos)
|
||
|
|
interested.append(msgid)
|
||
|
|
|
||
|
|
semaphore.acquire()
|