##// END OF EJS Templates
bug fixed prompt count using differents clients
bug fixed prompt count using differents clients

File last commit:

r5588:a7c91ab3
r5588:a7c91ab3
Show More
frontend.py
280 lines | 10.5 KiB | text/x-python | PythonLexer
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 # -*- coding: utf-8 -*-
"""Frontend of ipython working with python-zmq
Ipython's frontend, is a ipython interface that send request to kernel and proccess the kernel's outputs.
For more details, see the ipython-zmq design
"""
#-----------------------------------------------------------------------------
# Copyright (C) 2010 The IPython Development Team
#
# Distributed under the terms of the BSD License. The full license is in
# the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
import __builtin__
from contextlib import nested
import time
import sys
import os
import signal
import uuid
import cPickle as pickle
import code
import zmq
Omar Andres Zapata Mesa
raw_input captured and working fine
r5583 import readline
Omar Andres Zapata Mesa
tab completion is not working yet, unknow error
r5585 import rlcompleter
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 import time
#-----------------------------------------------------------------------------
# Imports from ipython
#-----------------------------------------------------------------------------
from IPython.external.argparse import ArgumentParser
from IPython.utils.traitlets import (
Int, Str, CBool, CaselessStrEnum, Enum, List, Unicode
)
from IPython.core.interactiveshell import get_default_colors
from IPython.core.excolors import exception_colors
from IPython.utils import PyColorize
from IPython.core.inputsplitter import InputSplitter
from IPython.frontend.terminal.kernelmanager import KernelManager2p as KernelManager
from IPython.zmq.session import Session
Omar Andres Zapata Mesa
bug fixed prompt count using differents clients
r5588 from IPython.frontend.terminal.completer import ClientCompleter2p
Omar Andres Zapata Mesa
tab completion is not working yet, unknow error
r5585
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 #-----------------------------------------------------------------------------
# Network Constants
#-----------------------------------------------------------------------------
from IPython.utils.localinterfaces import LOCALHOST, LOCAL_IPS
class Frontend(object):
""" this class is a simple frontend to ipython-zmq
NOTE: this class use kernelmanager to manipulate sockets
Parameters:
-----------
kernelmanager : object
instantiated object from class KernelManager in module kernelmanager
"""
def __init__(self,kernelmanager):
self.km = kernelmanager
self.session = kernelmanager.session
self.request_socket = self.km.xreq_channel.socket
self.sub_socket = self.km.sub_channel.socket
self.reply_socket = self.km.rep_channel.socket
Omar Andres Zapata Mesa
tab completion is not working yet, unknow error
r5585 self.msg_header = self.km.session.msg_header()
self.completer = ClientCompleter2p(self,self.km)
Omar Andres Zapata Mesa
raw_input captured and working fine
r5583 readline.parse_and_bind("tab: complete")
readline.parse_and_bind('set show-all-if-ambiguous on')
Omar Andres Zapata Mesa
tab completion is not working yet, unknow error
r5585 readline.set_completer(self.completer.complete)
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 history_path = os.path.expanduser('~/.ipython/history')
if os.path.isfile(history_path):
rlcompleter.readline.read_history_file(history_path)
else:
print("history file can not be readed.")
self.messages = {}
Omar Andres Zapata Mesa
working in handlers
r5579
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 self._splitter = InputSplitter()
Omar Andres Zapata Mesa
working in tab completion
r5580 self.code = ""
Omar Andres Zapata Mesa
bug fixed prompt count using differents clients
r5588
self.prompt_count = 0
self._get_initail_promt()
def _get_initail_promt(self):
self._execute('', hidden=True)
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 def interact(self):
""" let you get input from console using inputsplitter, then
while you enter code it can indent and set index id to any input
"""
Omar Andres Zapata Mesa
working in tab completion
r5580
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 try:
Omar Andres Zapata Mesa
working in tab completion
r5580 self._splitter.push(raw_input('In[%i]:'%self.prompt_count+self.code))
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 while self._splitter.push_accepts_more():
Omar Andres Zapata Mesa
working in tab completion
r5580 self.code = raw_input('.....:'+' '*self._splitter.indent_spaces)
self._splitter.push(' '*self._splitter.indent_spaces+self.code)
self._execute(self._splitter.source,False)
self._splitter.reset()
Omar Andres Zapata Mesa
working in handlers
r5579 except KeyboardInterrupt:
print('\nKeyboardInterrupt\n')
pass
Omar Andres Zapata Mesa
working in tab completion
r5580
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578
def start(self):
""" init a bucle that call interact method to get code.
"""
while True:
try:
Omar Andres Zapata Mesa
working in handlers
r5579 self.interact()
except KeyboardInterrupt:
print('\nKeyboardInterrupt\n')
pass
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 except EOFError:
answer = ''
while True:
answer = raw_input('\nDo you really want to exit ([y]/n)?')
if answer == 'y' or answer == '' :
self.km.shutdown_kernel()
sys.exit()
elif answer == 'n':
break
Omar Andres Zapata Mesa
tab completion is not working yet, unknow error
r5585
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 def _execute(self, source, hidden = True):
""" Execute 'source'. If 'hidden', do not show any output.
See parent class :meth:`execute` docstring for full details.
"""
Omar Andres Zapata Mesa
completer not working fine
r5581 self.km.xreq_channel.execute(source, hidden)
self.handle_xreq_channel()
Omar Andres Zapata Mesa
raw_input captured and working fine
r5583 self.handle_rep_channel()
Omar Andres Zapata Mesa
completer not working fine
r5581
def handle_xreq_channel(self):
# Give the kernel up to 0.5s to respond
for i in range(5):
if self.km.xreq_channel.was_called():
self.msg_xreq = self.km.xreq_channel.get_msg()
if self.msg_header["session"] == self.msg_xreq["parent_header"]["session"] :
if self.msg_xreq["content"]["status"] == 'ok' :
if self.msg_xreq["msg_type"] == "execute_reply" :
self.handle_sub_channel()
Omar Andres Zapata Mesa
bug fixed prompt count using differents clients
r5588 self.prompt_count = self.msg_xreq["content"]["execution_count"]+1
Omar Andres Zapata Mesa
working in tab completion
r5580
Omar Andres Zapata Mesa
completer not working fine
r5581 else:
print >> sys.stderr, "Error executing: ",self.msg_xreq ##traceback no implemented yet here
print >> sys.stderr, "Status in the kernel: ", self.msg_xreq["content"]["status"]
break
time.sleep(0.1)
Omar Andres Zapata Mesa
working in handlers
r5579
def handle_sub_channel(self):
""" Method to procces subscribe channel's messages
this method read a message and procces the content
in differents outputs like stdout, stderr, pyout
and status
Arguments:
sub_msg: message receive from kernel in the sub socket channel
capture by kernel manager.
"""
while self.km.sub_channel.was_called():
sub_msg = self.km.sub_channel.get_msg()
Omar Andres Zapata Mesa
working in tab completion
r5580 if self.msg_header["username"] == sub_msg['parent_header']['username'] and self.km.session.session == sub_msg['parent_header']['session']:
Omar Andres Zapata Mesa
working in handlers
r5579 if sub_msg['msg_type'] == 'status' :
if sub_msg["content"]["execution_state"] == "busy" :
pass
if sub_msg['msg_type'] == 'stream' :
if sub_msg["content"]["name"] == "stdout":
print >> sys.stdout,sub_msg["content"]["data"]
sys.stdout.flush()
Omar Andres Zapata Mesa
working in tab completion
r5580 if sub_msg["content"]["name"] == "stderr" :
Omar Andres Zapata Mesa
working in handlers
r5579 print >> sys.stderr,sub_msg["content"]["data"]
sys.stderr.flush()
if sub_msg['msg_type'] == 'pyout' :
Omar Andres Zapata Mesa
little bug fixed in pyout message print.
r5587 print >> sys.stdout,"Out[%i]:"%sub_msg["content"]["execution_count"], sub_msg["content"]["data"]["text/plain"]
Omar Andres Zapata Mesa
working in handlers
r5579 sys.stdout.flush()
Omar Andres Zapata Mesa
raw_input captured and working fine
r5583 def handle_rep_channel(self):
""" Method to capture raw_input
"""
if self.km.rep_channel.was_called() :
self.msg_rep = self.km.rep_channel.get_msg()
if self.msg_header["session"] == self.msg_rep["parent_header"]["session"] :
raw_data = raw_input(self.msg_rep["content"]["prompt"])
self.km.rep_channel.input(raw_data)
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578
def start_frontend():
""" Entry point for application.
Omar Andres Zapata Mesa
working in handlers
r5579
Omar Andres Zapata Mesa
basic kernelmanager and frontend wrote, support indentation but it dont run code yet
r5578 """
# Parse command line arguments.
parser = ArgumentParser()
kgroup = parser.add_argument_group('kernel options')
kgroup.add_argument('-e', '--existing', action='store_true',
help='connect to an existing kernel')
kgroup.add_argument('--ip', type=str, default=LOCALHOST,
help=\
"set the kernel\'s IP address [default localhost].\
If the IP address is something other than localhost, then \
Consoles on other machines will be able to connect\
to the Kernel, so be careful!")
kgroup.add_argument('--xreq', type=int, metavar='PORT', default=0,
help='set the XREQ channel port [default random]')
kgroup.add_argument('--sub', type=int, metavar='PORT', default=0,
help='set the SUB channel port [default random]')
kgroup.add_argument('--rep', type=int, metavar='PORT', default=0,
help='set the REP channel port [default random]')
kgroup.add_argument('--hb', type=int, metavar='PORT', default=0,
help='set the heartbeat port [default random]')
egroup = kgroup.add_mutually_exclusive_group()
egroup.add_argument('--pure', action='store_true', help = \
'use a pure Python kernel instead of an IPython kernel')
egroup.add_argument('--pylab', type=str, metavar='GUI', nargs='?',
const='auto', help = \
"Pre-load matplotlib and numpy for interactive use. If GUI is not \
given, the GUI backend is matplotlib's, otherwise use one of: \
['tk', 'gtk', 'qt', 'wx', 'inline'].")
egroup.add_argument('--colors', type=str,
help="Set the color scheme (LightBG,Linux,NoColor). This is guessed\
based on the pygments style if not set.")
args = parser.parse_args()
# parse the colors arg down to current known labels
if args.colors:
colors=args.colors.lower()
if colors in ('lightbg', 'light'):
colors='lightbg'
elif colors in ('dark', 'linux'):
colors='linux'
else:
colors='nocolor'
else:
colors=None
# Create a KernelManager and start a kernel.
kernel_manager = KernelManager(xreq_address=(args.ip, args.xreq),
sub_address=(args.ip, args.sub),
rep_address=(args.ip, args.rep),
hb_address=(args.ip, args.hb))
if not args.existing:
# if not args.ip in LOCAL_IPS+ALL_ALIAS:
# raise ValueError("Must bind a local ip, such as: %s"%LOCAL_IPS)
kwargs = dict(ip=args.ip)
if args.pure:
kwargs['ipython']=False
else:
kwargs['colors']=colors
if args.pylab:
kwargs['pylab']=args.pylab
kernel_manager.start_kernel(**kwargs)
kernel_manager.start_channels()
time.sleep(4)
frontend=Frontend(kernel_manager)
return frontend
if __name__ == "__main__" :
frontend=start_frontend()
frontend.start()