|
|
# encoding: utf-8
|
|
|
"""
|
|
|
An application for managing IPython profiles.
|
|
|
|
|
|
To be invoked as the `ipython profile` subcommand.
|
|
|
|
|
|
Authors:
|
|
|
|
|
|
* Min RK
|
|
|
|
|
|
"""
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Copyright (C) 2008-2011 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 logging
|
|
|
import os
|
|
|
|
|
|
from IPython.config.application import Application, boolean_flag
|
|
|
from IPython.core.application import (
|
|
|
BaseIPythonApplication, base_flags, base_aliases
|
|
|
)
|
|
|
from IPython.core.profiledir import ProfileDir
|
|
|
from IPython.utils.path import get_ipython_dir
|
|
|
from IPython.utils.traitlets import Unicode, Bool, Dict
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Constants
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
create_help = """Create an ipcluster profile by name
|
|
|
|
|
|
Create an ipython profile directory by its name or
|
|
|
profile directory path. Profile directories contain
|
|
|
configuration, log and security related files and are named
|
|
|
using the convention 'profile_<name>'. By default they are
|
|
|
located in your ipython directory. Once created, you will
|
|
|
can edit the configuration files in the profile
|
|
|
directory to configure IPython. Most users will create a
|
|
|
cluster directory by profile name,
|
|
|
`ipython profile create myprofile`, which will put the directory
|
|
|
in `<ipython_dir>/profile_myprofile`.
|
|
|
"""
|
|
|
list_help = """List available IPython profiles
|
|
|
|
|
|
List all available profiles, by profile location, that can
|
|
|
be found in the current working directly or in the ipython
|
|
|
directory. Profile directories are named using the convention
|
|
|
'profile_<profile>'.
|
|
|
"""
|
|
|
profile_help = """Manage IPython profiles
|
|
|
|
|
|
Profile directories contain
|
|
|
configuration, log and security related files and are named
|
|
|
using the convention 'profile_<name>'. By default they are
|
|
|
located in your ipython directory. You can create profiles
|
|
|
with `ipython profile create <name>`, or see the profiles you
|
|
|
already have with `ipython profile list`
|
|
|
|
|
|
To get started configuring IPython, simply do:
|
|
|
|
|
|
$> ipython profile create
|
|
|
|
|
|
and IPython will create the default profile in <ipython_dir>/profile_default,
|
|
|
where you can edit ipython_config.py to start configuring IPython.
|
|
|
|
|
|
"""
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Profile Application Class (for `ipython profile` subcommand)
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
class ProfileList(Application):
|
|
|
name = u'ipython-profile'
|
|
|
description = list_help
|
|
|
|
|
|
aliases = Dict(dict(
|
|
|
ipython_dir = 'ProfileList.ipython_dir',
|
|
|
log_level = 'Application.log_level',
|
|
|
))
|
|
|
flags = Dict(dict(
|
|
|
debug = ({'Application' : {'log_level' : 0}},
|
|
|
"Set log_level to 0, maximizing log output."
|
|
|
)
|
|
|
))
|
|
|
ipython_dir = Unicode(get_ipython_dir(), config=True,
|
|
|
help="""
|
|
|
The name of the IPython directory. This directory is used for logging
|
|
|
configuration (through profiles), history storage, etc. The default
|
|
|
is usually $HOME/.ipython. This options can also be specified through
|
|
|
the environment variable IPYTHON_DIR.
|
|
|
"""
|
|
|
)
|
|
|
|
|
|
def list_profile_dirs(self):
|
|
|
# Find the search paths
|
|
|
paths = [os.getcwdu(), self.ipython_dir]
|
|
|
|
|
|
self.log.warn('Searching for IPython profiles in paths: %r' % paths)
|
|
|
for path in paths:
|
|
|
files = os.listdir(path)
|
|
|
for f in files:
|
|
|
full_path = os.path.join(path, f)
|
|
|
if os.path.isdir(full_path) and f.startswith('profile_'):
|
|
|
profile = f.split('_',1)[-1]
|
|
|
start_cmd = 'ipython profile=%s' % profile
|
|
|
print start_cmd + " ==> " + full_path
|
|
|
|
|
|
def start(self):
|
|
|
self.list_profile_dirs()
|
|
|
|
|
|
|
|
|
create_flags = {}
|
|
|
create_flags.update(base_flags)
|
|
|
create_flags.update(boolean_flag('reset', 'ProfileCreate.overwrite',
|
|
|
"reset config files to defaults", "leave existing config files"))
|
|
|
create_flags.update(boolean_flag('cluster', 'ProfileCreate.cluster',
|
|
|
"Include parallel computing config files",
|
|
|
"Don't include parallel computing config files"))
|
|
|
|
|
|
class ProfileCreate(BaseIPythonApplication):
|
|
|
name = u'ipython-profile'
|
|
|
description = create_help
|
|
|
auto_create = Bool(True, config=False)
|
|
|
|
|
|
def _copy_config_files_default(self):
|
|
|
return True
|
|
|
|
|
|
cluster = Bool(False, config=True,
|
|
|
help="whether to include parallel computing config files")
|
|
|
def _cluster_changed(self, name, old, new):
|
|
|
cluster_files = [ 'ipcontroller_config.py',
|
|
|
'ipengine_config.py',
|
|
|
'ipcluster_config.py'
|
|
|
]
|
|
|
if new:
|
|
|
for cf in cluster_files:
|
|
|
self.config_files.append(cf)
|
|
|
else:
|
|
|
for cf in cluster_files:
|
|
|
if cf in self.config_files:
|
|
|
self.config_files.remove(cf)
|
|
|
|
|
|
def parse_command_line(self, argv):
|
|
|
super(ProfileCreate, self).parse_command_line(argv)
|
|
|
# accept positional arg as profile name
|
|
|
if self.extra_args:
|
|
|
self.profile = self.extra_args[0]
|
|
|
|
|
|
flags = Dict(create_flags)
|
|
|
|
|
|
aliases = Dict(dict(profile='BaseIPythonApplication.profile'))
|
|
|
|
|
|
classes = [ProfileDir]
|
|
|
|
|
|
def init_config_files(self):
|
|
|
super(ProfileCreate, self).init_config_files()
|
|
|
# use local imports, since these classes may import from here
|
|
|
from IPython.frontend.terminal.ipapp import TerminalIPythonApp
|
|
|
apps = [TerminalIPythonApp]
|
|
|
try:
|
|
|
from IPython.frontend.qt.console.qtconsoleapp import IPythonQtConsoleApp
|
|
|
except ImportError:
|
|
|
pass
|
|
|
else:
|
|
|
apps.append(IPythonQtConsoleApp)
|
|
|
if self.cluster:
|
|
|
from IPython.parallel.apps.ipcontrollerapp import IPControllerApp
|
|
|
from IPython.parallel.apps.ipengineapp import IPEngineApp
|
|
|
from IPython.parallel.apps.ipclusterapp import IPClusterStart
|
|
|
from IPython.parallel.apps.iploggerapp import IPLoggerApp
|
|
|
apps.extend([
|
|
|
IPControllerApp,
|
|
|
IPEngineApp,
|
|
|
IPClusterStart,
|
|
|
IPLoggerApp,
|
|
|
])
|
|
|
for App in apps:
|
|
|
app = App()
|
|
|
app.config.update(self.config)
|
|
|
app.log = self.log
|
|
|
app.overwrite = self.overwrite
|
|
|
app.copy_config_files=True
|
|
|
app.profile = self.profile
|
|
|
app.init_profile_dir()
|
|
|
app.init_config_files()
|
|
|
|
|
|
def stage_default_config_file(self):
|
|
|
pass
|
|
|
|
|
|
class ProfileApp(Application):
|
|
|
name = u'ipython-profile'
|
|
|
description = profile_help
|
|
|
|
|
|
subcommands = Dict(dict(
|
|
|
create = (ProfileCreate, "Create a new profile dir with default config files"),
|
|
|
list = (ProfileList, "List existing profiles")
|
|
|
))
|
|
|
|
|
|
def start(self):
|
|
|
if self.subapp is None:
|
|
|
print "No subcommand specified. Must specify one of: %s"%(self.subcommands.keys())
|
|
|
print
|
|
|
self.print_description()
|
|
|
self.print_subcommands()
|
|
|
self.exit(1)
|
|
|
else:
|
|
|
return self.subapp.start()
|
|
|
|
|
|
|
