You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
from IPython.kernel.zmq.kernelbase import Kernel
|
|
from pexpect import replwrap, EOF
|
|
|
|
import signal
|
|
from subprocess import check_output
|
|
import re
|
|
|
|
__version__ = '0.2'
|
|
|
|
version_pat = re.compile(r'version (\d+(\.\d+)+)')
|
|
|
|
class BashKernel(Kernel):
|
|
implementation = 'bash_kernel'
|
|
implementation_version = __version__
|
|
language = 'bash'
|
|
@property
|
|
def language_version(self):
|
|
m = version_pat.search(self.banner)
|
|
return m.group(1)
|
|
|
|
_banner = None
|
|
@property
|
|
def banner(self):
|
|
if self._banner is None:
|
|
self._banner = check_output(['bash', '--version']).decode('utf-8')
|
|
return self._banner
|
|
|
|
language_info = {'codemirror_mode': 'shell',
|
|
'mimetype': 'text/x-sh',
|
|
'file_extension': 'sh'
|
|
}
|
|
|
|
def __init__(self, **kwargs):
|
|
Kernel.__init__(self, **kwargs)
|
|
self._start_bash()
|
|
|
|
def _start_bash(self):
|
|
# Signal handlers are inherited by forked processes, and we can't easily
|
|
# reset it from the subprocess. Since kernelapp ignores SIGINT except in
|
|
# message handlers, we need to temporarily reset the SIGINT handler here
|
|
# so that bash and its children are interruptible.
|
|
sig = signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
try:
|
|
self.bashwrapper = replwrap.bash()
|
|
finally:
|
|
signal.signal(signal.SIGINT, sig)
|
|
|
|
def do_execute(self, code, silent, store_history=True, user_expressions=None,
|
|
allow_stdin=False):
|
|
if not code.strip():
|
|
return {'status': 'ok', 'execution_count': self.execution_count,
|
|
'payload': [], 'user_expressions': {}}
|
|
|
|
interrupted = False
|
|
try:
|
|
output = self.bashwrapper.run_command(code.rstrip(), timeout=None)
|
|
except KeyboardInterrupt:
|
|
self.bashwrapper.child.sendintr()
|
|
interrupted = True
|
|
self.bashwrapper._expect_prompt()
|
|
output = self.bashwrapper.child.before
|
|
except EOF:
|
|
output = self.bashwrapper.child.before + 'Restarting Bash'
|
|
self._start_bash()
|
|
|
|
if not silent:
|
|
stream_content = {'name': 'stdout', 'text': output}
|
|
self.send_response(self.iopub_socket, 'stream', stream_content)
|
|
|
|
if interrupted:
|
|
return {'status': 'abort', 'execution_count': self.execution_count}
|
|
|
|
try:
|
|
exitcode = int(self.bashwrapper.run_command('echo $?').rstrip())
|
|
except Exception:
|
|
exitcode = 1
|
|
|
|
if exitcode:
|
|
return {'status': 'error', 'execution_count': self.execution_count,
|
|
'ename': '', 'evalue': str(exitcode), 'traceback': []}
|
|
else:
|
|
return {'status': 'ok', 'execution_count': self.execution_count,
|
|
'payload': [], 'user_expressions': {}}
|
|
|
|
def do_complete(self, code, cursor_pos):
|
|
code = code[:cursor_pos]
|
|
default = {'matches': [], 'cursor_start': 0,
|
|
'cursor_end': cursor_pos, 'metadata': dict(),
|
|
'status': 'ok'}
|
|
|
|
if not code or code[-1] == ' ':
|
|
return default
|
|
|
|
tokens = code.replace(';', ' ').split()
|
|
if not tokens:
|
|
return default
|
|
|
|
token = tokens[-1]
|
|
start = cursor_pos - len(token)
|
|
cmd = 'compgen -cdfa %s' % token
|
|
output = self.bashwrapper.run_command(cmd).rstrip()
|
|
|
|
matches = output.split()
|
|
if not matches:
|
|
return default
|
|
matches = [m for m in matches if m.startswith(token)]
|
|
|
|
return {'matches': matches, 'cursor_start': start,
|
|
'cursor_end': cursor_pos, 'metadata': dict(),
|
|
'status': 'ok'}
|
|
|
|
if __name__ == '__main__':
|
|
from IPython.kernel.zmq.kernelapp import IPKernelApp
|
|
IPKernelApp.launch_instance(kernel_class=BashKernel)
|