1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
from ipykernel.kernelbase import Kernel
from ipython_genutils.py3compat import safe_unicode
import json, socket, sys, traceback
__version__ = '0.1'
class BALLViewKernel(Kernel):
"""PyBALL kernel for Jupyter"""
implementation = 'pyball_kernel'
implementation_version = __version__
banner = 'PyBALL'
language_info = {
'mimetype': 'text/x-python',
'name': 'python',
'file_extension': '.py',
}
pyserver_host = '127.0.0.1'
pyserver_port = 8897
def do_execute(self, code, silent, store_history=True, user_expressions=None, allow_stdin=False):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
s.connect((self.pyserver_host, self.pyserver_port))
except socket.error:
exc_type, exc_value, exc_traceback = sys.exc_info()
content = {
'name': 'stderr',
'text': 'ERROR: Cannot connect to BALL PyServer. Please make sure PyServer is running!'
}
self.send_response(self.iopub_socket, 'stream', content)
content.update({
'status': 'error',
'ename': type(exc_type).__name__,
'evalue': safe_unicode(traceback.format_exception_only(exc_type, exc_value)),
'traceback': traceback.format_tb(exc_traceback),
'execution_count': self.execution_count - 1,
'payload': [],
'user_expression': {}
})
return content
code = json.dumps({
'msg_type': 'execute_request',
'content': code
})
s.sendall(code.encode('utf-8'))
data = self.recvall(s)
s.close()
content = {}
if data is None:
content.update({
'name': 'stderr',
'text': 'WARNING: Response from BALL PyServer is incomplete!\n'
'Your code was evaluated by the BALL PyServer but the output cannot be shown!'
})
else:
content.update({
'name': 'stdout' if data['msg_type'] == 'execute_result' else 'stderr',
'text': data['content']
})
self.send_response(self.iopub_socket, 'stream', content)
content.update({
'status': 'ok',
'execution_count': self.execution_count,
'payload': [],
'user_expressions': {},
})
return content
def recvall(self, sock):
"""Reads all JSON data from the given socket and returns the dict representation of the data. Returns None
if the data is not in JSON format."""
nbytes = 4096
dat = []
while True:
dat.append(sock.recv(nbytes).decode('utf-8'))
if(len(dat[-1])) < nbytes:
break
try:
return json.loads(''.join(dat))
except ValueError:
return None
if __name__ == '__main__':
from ipykernel.kernelapp import IPKernelApp
IPKernelApp.launch_instance(kernel_class=BALLViewKernel)
|
