|
|
""" Defines helper functions for creating kernel entry points and process
|
|
|
launchers.
|
|
|
"""
|
|
|
|
|
|
# Standard library imports.
|
|
|
import logging
|
|
|
import atexit
|
|
|
import os
|
|
|
import socket
|
|
|
from subprocess import Popen, PIPE
|
|
|
import sys
|
|
|
|
|
|
# System library imports.
|
|
|
import zmq
|
|
|
from zmq.log import handlers
|
|
|
# Local imports.
|
|
|
from IPython.core.ultratb import FormattedTB
|
|
|
from IPython.external.argparse import ArgumentParser
|
|
|
from IPython.zmq.log import logger
|
|
|
|
|
|
def split_ports(s, n):
|
|
|
"""Parser helper for multiport strings"""
|
|
|
if not s:
|
|
|
return tuple([0]*n)
|
|
|
ports = map(int, s.split(','))
|
|
|
if len(ports) != n:
|
|
|
raise ValueError
|
|
|
return ports
|
|
|
|
|
|
def select_random_ports(n):
|
|
|
"""Selects and return n random ports that are open."""
|
|
|
ports = []
|
|
|
for i in xrange(n):
|
|
|
sock = socket.socket()
|
|
|
sock.bind(('', 0))
|
|
|
ports.append(sock)
|
|
|
for i, sock in enumerate(ports):
|
|
|
port = sock.getsockname()[1]
|
|
|
sock.close()
|
|
|
ports[i] = port
|
|
|
return ports
|
|
|
|
|
|
def parse_url(args):
|
|
|
if args.url:
|
|
|
iface = args.url.split('://',1)
|
|
|
if len(args) == 2:
|
|
|
args.transport,iface = iface
|
|
|
iface = iface.split(':')
|
|
|
args.ip = iface[0]
|
|
|
if iface[1]:
|
|
|
args.regport = iface[1]
|
|
|
args.url = "%s://%s:%i"%(args.transport, args.ip,args.regport)
|
|
|
|
|
|
|
|
|
|
|
|
def make_base_argument_parser():
|
|
|
""" Creates an ArgumentParser for the generic arguments supported by all
|
|
|
ipcluster entry points.
|
|
|
"""
|
|
|
parser = ArgumentParser()
|
|
|
parser.add_argument('--ip', type=str, default='127.0.0.1',
|
|
|
help='set the controller\'s IP address [default: local]')
|
|
|
parser.add_argument('--transport', type=str, default='tcp',
|
|
|
help='set the transport to use [default: tcp]')
|
|
|
parser.add_argument('--regport', type=int, metavar='PORT', default=10101,
|
|
|
help='set the XREP port for registration [default: 10101]')
|
|
|
parser.add_argument('--logport', type=int, metavar='PORT', default=20202,
|
|
|
help='set the PUB port for logging [default: 10201]')
|
|
|
parser.add_argument('--loglevel', type=int, metavar='LEVEL', default=logging.DEBUG,
|
|
|
help='set the log level [default: DEBUG]')
|
|
|
parser.add_argument('--ident', type=str,
|
|
|
help='set the ZMQ identity [default: random]')
|
|
|
parser.add_argument('--packer', type=str, default='json',
|
|
|
choices=['json','pickle'],
|
|
|
help='set the message format method [default: json]')
|
|
|
parser.add_argument('--url', type=str,
|
|
|
help='set transport,ip,regport in one arg, e.g. tcp://127.0.0.1:10101')
|
|
|
|
|
|
return parser
|
|
|
|
|
|
|
|
|
def connect_logger(context, iface, root="ip", loglevel=logging.DEBUG):
|
|
|
lsock = context.socket(zmq.PUB)
|
|
|
lsock.connect(iface)
|
|
|
handler = handlers.PUBHandler(lsock)
|
|
|
handler.setLevel(loglevel)
|
|
|
handler.root_topic = root
|
|
|
logger.addHandler(handler)
|
|
|
|
