def __init__(self, address, port):
# call base class init
super(Shell, self).__init__()
# create router
self.router = RoutingNode(address, port)
# save shell address
self.address = '{}.1'.format(address)
# add ShellNode to node dict
self.router.nodeClasses['ShellNode'] = ShellNode
class Shell(object):
def __init__(self, address, port):
# call base class init
super(Shell, self).__init__()
# create router
self.router = RoutingNode(address, port)
# save shell address
self.address = '{}.1'.format(address)
# add ShellNode to node dict
self.router.nodeClasses['ShellNode'] = ShellNode
def start(self, script=None):
# start router
self.router.start()
# add prining node
self.router.input.put({'sender': self.address,
'receiver': self.router.address, 'request': 'new_node',
'args': [ShellNode], 'kargs': {}})
# wait a bit
time.sleep(1)
# split script
if not script is None:
scriptLines = script.split('\n')
# create requester
request = requester(self.router.input,
self.address)
# main loop
while True:
if script == None:
# get commands
userInput = raw_input('> ')
elif len(scriptLines) > 0:
userInput = scriptLines[0]
# delete first line
scriptLines.remove(scriptLines[0])
else:
script = None
continue
# check for exir
if userInput == 'quit' or userInput == 'exit':
break
# parse input
try:
eval(userInput, dict(
self.router.nodeClasses.items() + \
{'request': request}.items()))
except Exception, e:
print "invalid input: '{}'".format(userInput)
print e
continue
# wait a bit
time.sleep(0.5)
# stop router
self.router.input.put({'sender': self.address,
'receiver': self.router.address, 'request': 'stop',
'args': [], 'kargs': {}})
