First take at cleaning up extensions.

IPython/extensions/__init__.py

@@ -1,13 +1,2 b''
 # -*- coding: utf-8 -*-
-"""This directory is meant for special-purpose extensions to IPython.
-
-This can include things which alter the syntax processing stage (see
-PhysicalQ_Input for an example of how to do this).
-
-Any file located here can be called with an 'execfile =' option as
-
-  execfile = extensions/filename.py
-
-since the IPython directory itself is already part of the search path for
-files listed as 'execfile ='.
-"""
+"""This directory is meant for IPython extensions."""

IPython/extensions/parallelmagic.py

@@ -1,209 +1,201 b''
 #!/usr/bin/env python
 # encoding: utf-8
 
 """Magic command interface for interactive parallel work."""
 
 #-----------------------------------------------------------------------------
 #  Copyright (C) 2008-2009  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 new
 
-from IPython.core.component import Component
-from IPython.utils.traitlets import Bool, Any
+from IPython.config.configurable import Configurable
+from IPython.utils.traitlets import Bool, Any, Instance
 from IPython.utils.autoattr import auto_attr
 from IPython.testing import decorators as testdec
 
 #-----------------------------------------------------------------------------
 # Definitions of magic functions for use with IPython
 #-----------------------------------------------------------------------------
 
 
 NO_ACTIVE_MULTIENGINE_CLIENT = """
 Use activate() on a MultiEngineClient object to activate it for magics.
 """
 
 
-class ParalleMagicComponent(Component):
+class ParalleMagicComponent(Configurable):
     """A component to manage the %result, %px and %autopx magics."""
 
     active_multiengine_client = Any()
     verbose = Bool(False, config=True)
+    shell = Instance('IPython.core.iplib.InteractiveShell')
 
-    def __init__(self, parent, name=None, config=None):
-        super(ParalleMagicComponent, self).__init__(parent, name=name, config=config)
+    def __init__(self, shell, config=None):
+        super(ParalleMagicComponent, self).__init__(config=config)
+        self.shell = shell
         self._define_magics()
         # A flag showing if autopx is activated or not
         self.autopx = False
 
-    # Access other components like this rather than by a regular attribute.
-    # This won't lookup the InteractiveShell object until it is used and
-    # then it is cached.  This is both efficient and couples this class 
-    # more loosely to InteractiveShell.
-    @auto_attr
-    def shell(self):
-        return Component.get_instances(
-            root=self.root,
-            klass='IPython.core.iplib.InteractiveShell')[0]
-
     def _define_magics(self):
         """Define the magic functions."""
         self.shell.define_magic('result', self.magic_result)
         self.shell.define_magic('px', self.magic_px)
         self.shell.define_magic('autopx', self.magic_autopx)
 
     @testdec.skip_doctest
     def magic_result(self, ipself, parameter_s=''):
         """Print the result of command i on all engines..
 
         To use this a :class:`MultiEngineClient` instance must be created 
         and then activated by calling its :meth:`activate` method.
 
         Then you can do the following::
 
             In [23]: %result
             Out[23]: 
             <Results List>
             [0] In [6]: a = 10
             [1] In [6]: a = 10
     
             In [22]: %result 6
             Out[22]: 
             <Results List>
             [0] In [6]: a = 10
             [1] In [6]: a = 10
         """
         if self.active_multiengine_client is None:
             print NO_ACTIVE_MULTIENGINE_CLIENT
             return
 
         try:
             index = int(parameter_s)
         except:
             index = None
         result = self.active_multiengine_client.get_result(index)
         return result
 
     @testdec.skip_doctest
     def magic_px(self, ipself, parameter_s=''):
         """Executes the given python command in parallel.
 
         To use this a :class:`MultiEngineClient` instance must be created 
         and then activated by calling its :meth:`activate` method.
         
         Then you can do the following::
 
             In [24]: %px a = 5
             Parallel execution on engines: all
             Out[24]: 
             <Results List>
             [0] In [7]: a = 5
             [1] In [7]: a = 5
         """
 
         if self.active_multiengine_client is None:
             print NO_ACTIVE_MULTIENGINE_CLIENT
             return
         print "Parallel execution on engines: %s" % self.active_multiengine_client.targets
         result = self.active_multiengine_client.execute(parameter_s)
         return result
 
     @testdec.skip_doctest
     def magic_autopx(self, ipself, parameter_s=''):
         """Toggles auto parallel mode.
 
         To use this a :class:`MultiEngineClient` instance must be created 
         and then activated by calling its :meth:`activate` method. Once this
         is called, all commands typed at the command line are send to
         the engines to be executed in parallel. To control which engine
         are used, set the ``targets`` attributed of the multiengine client
         before entering ``%autopx`` mode.
 
         Then you can do the following::
 
             In [25]: %autopx
             %autopx to enabled
 
             In [26]: a = 10
             <Results List>
             [0] In [8]: a = 10
             [1] In [8]: a = 10
 
 
             In [27]: %autopx
             %autopx disabled
         """
         if self.autopx:
             self._disable_autopx()
         else:
             self._enable_autopx()
 
     def _enable_autopx(self):
         """Enable %autopx mode by saving the original runsource and installing 
         pxrunsource.
         """
         if self.active_multiengine_client is None:
             print NO_ACTIVE_MULTIENGINE_CLIENT
             return
 
         self._original_runsource = self.shell.runsource
         self.shell.runsource = new.instancemethod(
             self.pxrunsource, self.shell, self.shell.__class__
         )
         self.autopx = True
         print "%autopx enabled"
     
     def _disable_autopx(self):
         """Disable %autopx by restoring the original InteractiveShell.runsource."""
         if self.autopx:
             self.shell.runsource = self._original_runsource
             self.autopx = False
             print "%autopx disabled"
 
     def pxrunsource(self, ipself, source, filename="<input>", symbol="single"):
         """A parallel replacement for InteractiveShell.runsource."""
 
         try:
             code = ipself.compile(source, filename, symbol)
         except (OverflowError, SyntaxError, ValueError):
             # Case 1
             ipself.showsyntaxerror(filename)
             return None
 
         if code is None:
             # Case 2
             return True
 
         # Case 3
         # Because autopx is enabled, we now call executeAll or disable autopx if
         # %autopx or autopx has been called
         if 'get_ipython().magic("%autopx' in source or 'get_ipython().magic("autopx' in source:
             self._disable_autopx()
             return False
         else:
             try:
                 result = self.active_multiengine_client.execute(source)
             except:
                 ipself.showtraceback()
             else:
                 print result.__repr__()
             return False
 
 
 _loaded = False
 
 
 def load_ipython_extension(ip):
     """Load the extension in IPython."""
     global _loaded
     if not _loaded:
-        prd = ParalleMagicComponent(ip, name='parallel_magic')
+        prd = ParalleMagicComponent(ip, config=ip.config)
         _loaded = True
 

IPython/extensions/pretty.py

@@ -1,166 +1,158 b''
 """Use pretty.py for configurable pretty-printing.
 
 To enable this extension in your configuration
 file, add the following to :file:`ipython_config.py`::
 
     c.Global.extensions = ['IPython.extensions.pretty']
     def dict_pprinter(obj, p, cycle):
         return p.text("<dict>")
     c.PrettyResultDisplay.verbose = True
     c.PrettyResultDisplay.defaults_for_type = [
         (dict, dict_pprinter)
     ]
     c.PrettyResultDisplay.defaults_for_type_by_name = [
         ('numpy', 'dtype', 'IPython.extensions.pretty.dtype_pprinter')
     ]
 
 This extension can also be loaded by using the ``%load_ext`` magic::
 
     %load_ext IPython.extensions.pretty
 
 If this extension is enabled, you can always add additional pretty printers
 by doing::
 
     ip = get_ipython()
     prd = ip.get_component('pretty_result_display')
     import numpy
     from IPython.extensions.pretty import dtype_pprinter
     prd.for_type(numpy.dtype, dtype_pprinter)
 
     # If you don't want to have numpy imported until it needs to be:
     prd.for_type_by_name('numpy', 'dtype', dtype_pprinter)
 """
 
 #-----------------------------------------------------------------------------
 # Imports
 #-----------------------------------------------------------------------------
 
 from IPython.core.error import TryNext
 from IPython.external import pretty
-from IPython.core.component import Component
-from IPython.utils.traitlets import Bool, List
+from IPython.config.configurable import Configurable
+from IPython.utils.traitlets import Bool, List, Instance
 from IPython.utils.io import Term
 from IPython.utils.autoattr import auto_attr
 from IPython.utils.importstring import import_item
 
 #-----------------------------------------------------------------------------
 # Code
 #-----------------------------------------------------------------------------
 
 
 _loaded = False
 
 
-class PrettyResultDisplay(Component):
+class PrettyResultDisplay(Configurable):
     """A component for pretty printing on steroids."""
 
     verbose = Bool(False, config=True)
+    shell = Instance('IPython.core.iplib.InteractiveShell')
 
     # A list of (type, func_name), like
     # [(dict, 'my_dict_printer')]
     # The final argument can also be a callable
     defaults_for_type = List(default_value=[], config=True)
 
     # A list of (module_name, type_name, func_name), like
     # [('numpy', 'dtype', 'IPython.extensions.pretty.dtype_pprinter')]
     # The final argument can also be a callable
     defaults_for_type_by_name = List(default_value=[], config=True)
 
-    def __init__(self, parent, name=None, config=None):
-        super(PrettyResultDisplay, self).__init__(parent, name=name, config=config)
+    def __init__(self, shell, config=None):
+        super(PrettyResultDisplay, self).__init__(config=config)
+        self.shell = shell
         self._setup_defaults()
 
     def _setup_defaults(self):
         """Initialize the default pretty printers."""
         for typ, func_name in self.defaults_for_type:
             func = self._resolve_func_name(func_name)
             self.for_type(typ, func)
         for type_module, type_name, func_name in self.defaults_for_type_by_name:
             func = self._resolve_func_name(func_name)
             self.for_type_by_name(type_module, type_name, func)
 
     def _resolve_func_name(self, func_name):
         if callable(func_name):
             return func_name
         elif isinstance(func_name, basestring):
             return import_item(func_name)
         else:
             raise TypeError('func_name must be a str or callable, got: %r' % func_name)
 
-    # Access other components like this rather than by a regular attribute.
-    # This won't lookup the InteractiveShell object until it is used and
-    # then it is cached.  This is both efficient and couples this class 
-    # more loosely to InteractiveShell.
-    @auto_attr
-    def shell(self):
-        return Component.get_instances(
-            root=self.root,
-            klass='IPython.core.iplib.InteractiveShell')[0]
-
     def __call__(self, otherself, arg):
         """Uber-pretty-printing display hook.
 
         Called for displaying the result to the user.
         """
         
         if self.shell.pprint:
             out = pretty.pretty(arg, verbose=self.verbose)
             if '\n' in out:
                 # So that multi-line strings line up with the left column of
                 # the screen, instead of having the output prompt mess up
                 # their first line.                
                 Term.cout.write('\n')
             print >>Term.cout, out
         else:
             raise TryNext
 
     def for_type(self, typ, func):
         """Add a pretty printer for a type."""
         return pretty.for_type(typ, func)
 
     def for_type_by_name(self, type_module, type_name, func):
         """Add a pretty printer for a type by its name and module name."""
         return pretty.for_type_by_name(type_module, type_name, func)
 
 
 #-----------------------------------------------------------------------------
 # Initialization code for the extension
 #-----------------------------------------------------------------------------
 
 
 def load_ipython_extension(ip):
     """Load the extension in IPython as a hook."""
     global _loaded
     if not _loaded:
-        prd = PrettyResultDisplay(ip, name='pretty_result_display')
+        prd = PrettyResultDisplay(ip, config=ip.config)
         ip.set_hook('result_display', prd, priority=99)
         _loaded = True
         return prd
 
 def unload_ipython_extension(ip):
     """Unload the extension."""
     # The hook system does not have a way to remove a hook so this is a pass
     pass
 
 
 #-----------------------------------------------------------------------------
 # Example pretty printers
 #-----------------------------------------------------------------------------
 
 
 def dtype_pprinter(obj, p, cycle):
     """ A pretty-printer for numpy dtype objects.
     """
     if cycle:
         return p.text('dtype(...)')
     if hasattr(obj, 'fields'):
         if obj.fields is None:
             p.text(repr(obj))
         else:
             p.begin_group(7, 'dtype([')
             for i, field in enumerate(obj.descr):
                 if i > 0:
                     p.text(',')
                     p.breakable()
                 p.pretty(field)
             p.end_group(7, '])')