Use new traitlets 4.1+ APIs

Author: SylvainCorlay

ipykernel/comm/comm.py

@@ -12,7 +12,7 @@
 from ipykernel.kernelbase import Kernel
 
 from ipykernel.jsonutil import json_clean
-from traitlets import Instance, Unicode, Bytes, Bool, Dict, Any
+from traitlets import Instance, Unicode, Bytes, Bool, Dict, Any, default
 
 
 class Comm(LoggingConfigurable):
@@ -21,15 +21,22 @@ class Comm(LoggingConfigurable):
     shell = Instance('IPython.core.interactiveshell.InteractiveShellABC',
                      allow_none=True)
     kernel = Instance('ipykernel.kernelbase.Kernel')
-    def _kernel_default(self):
+
+    @default('kernel')
+    def _default_kernel(self):
         if Kernel.initialized():
             return Kernel.instance()
 
     iopub_socket = Any()
-    def _iopub_socket_default(self):
+
+    @default('iopub_socket')
+    def _default_iopub_socket(self):
         return self.kernel.iopub_socket
+
     session = Instance('jupyter_client.session.Session')
-    def _session_default(self):
+
+    @default('session')
+    def _default_session(self):
         if self.kernel is not None:
             return self.kernel.session
 
@@ -38,7 +45,9 @@ def _session_default(self):
         which to load comm target.""")
 
     topic = Bytes()
-    def _topic_default(self):
+
+    @default('topic')
+    def _default_topic(self):
         return ('comm-%s' % self.comm_id).encode('ascii')
 
     _open_data = Dict(help="data dict, if any, to be included in comm_open")
@@ -49,7 +58,9 @@ def _topic_default(self):
 
     _closed = Bool(True)
     comm_id = Unicode()
-    def _comm_id_default(self):
+
+    @default('comm_id')
+    def _default_comm_id(self):
         return uuid.uuid4().hex
 
     primary = Bool(True, help="Am I the primary or secondary Comm?")

ipykernel/comm/manager.py

@@ -11,7 +11,7 @@
 
 from ipython_genutils.importstring import import_item
 from ipython_genutils.py3compat import string_types
-from traitlets import Instance, Unicode, Dict, Any
+from traitlets import Instance, Unicode, Dict, Any, default
 
 from .comm import Comm
 
@@ -34,10 +34,15 @@ class CommManager(LoggingConfigurable):
     kernel = Instance('ipykernel.kernelbase.Kernel')
 
     iopub_socket = Any()
-    def _iopub_socket_default(self):
+
+    @default('iopub_socket')
+    def _default_iopub_socket(self):
         return self.kernel.iopub_socket
+
     session = Instance('jupyter_client.session.Session')
-    def _session_default(self):
+
+    @default('session')
+    def _default_session(self):
         return self.kernel.session
 
     comms = Dict()

ipykernel/inprocess/client.py

@@ -13,15 +13,14 @@
 
 # IPython imports
 from ipykernel.inprocess.socket import DummySocket
-from traitlets import Type, Instance
+from traitlets import Type, Instance, default
 from jupyter_client.clientabc import KernelClientABC
 from jupyter_client.client import KernelClient
 
 # Local imports
 from .channels import (
     InProcessChannel,
     InProcessHBChannel,
-
 )
 
 #-----------------------------------------------------------------------------
@@ -50,11 +49,12 @@ class InProcessKernelClient(KernelClient):
     #--------------------------------------------------------------------------
     # Channel management methods
     #--------------------------------------------------------------------------
-    
-    def _blocking_class_default(self):
+
+    @default('blocking_class')
+    def _default_blocking_class(self):
         from .blocking import BlockingInProcessKernelClient
         return BlockingInProcessKernelClient
-    
+
     def get_connection_info(self):
         d = super(InProcessKernelClient, self).get_connection_info()
         d['kernel'] = self.kernel

ipykernel/inprocess/ipkernel.py

@@ -9,7 +9,7 @@
 
 from IPython.core.interactiveshell import InteractiveShellABC
 from ipykernel.jsonutil import json_clean
-from traitlets import Any, Enum, Instance, List, Type
+from traitlets import Any, Enum, Instance, List, Type, default
 from ipykernel.ipkernel import IPythonKernel
 from ipykernel.zmqshell import ZMQInteractiveShell
 
@@ -52,21 +52,25 @@ class InProcessKernel(IPythonKernel):
     control_stream = Any()
     _underlying_iopub_socket = Instance(DummySocket, ())
     iopub_thread = Instance(IOPubThread)
-    def _iopub_thread_default(self):
+
+    @default('iopub_thread')
+    def _default_iopub_thread(self):
         thread = IOPubThread(self._underlying_iopub_socket)
         thread.start()
         return thread
-    
+
     iopub_socket = Instance(BackgroundSocket)
-    def _iopub_socket_default(self):
+
+    @default('iopub_socket')
+    def _default_iopub_socket(self):
         return self.iopub_thread.background_socket
-    
+
     stdin_socket = Instance(DummySocket, ())
 
     def __init__(self, **traits):
         super(InProcessKernel, self).__init__(**traits)
 
-        self._underlying_iopub_socket.on_trait_change(self._io_dispatch, 'message_sent')
+        self._underlying_iopub_socket.observe(self._io_dispatch, names=['message_sent'])
         self.shell.kernel = self
 
     def execute_request(self, stream, ident, parent):
@@ -119,7 +123,7 @@ def _redirected_io(self):
 
     #------ Trait change handlers --------------------------------------------
 
-    def _io_dispatch(self):
+    def _io_dispatch(self, change):
         """ Called when a message is sent to the IO socket.
         """
         ident, msg = self.session.recv(self.iopub_socket, copy=False)
@@ -128,20 +132,25 @@ def _io_dispatch(self):
 
     #------ Trait initializers -----------------------------------------------
 
-    def _log_default(self):
+    @default('log')
+    def _default_log(self):
         return logging.getLogger(__name__)
 
-    def _session_default(self):
+    @default('session')
+    def _default_session(self):
         from jupyter_client.session import Session
         return Session(parent=self, key=b'')
 
-    def _shell_class_default(self):
+    @default('shell_class')
+    def _default_shell_class(self):
         return InProcessInteractiveShell
 
-    def _stdout_default(self):
+    @default('stdout')
+    def _default_stdout(self):
         return OutStream(self.session, self.iopub_thread, u'stdout')
 
-    def _stderr_default(self):
+    @default('stderr')
+    def _default_stderr(self):
         return OutStream(self.session, self.iopub_thread, u'stderr')
 
 #-----------------------------------------------------------------------------

ipykernel/inprocess/manager.py

@@ -3,7 +3,7 @@
 # Copyright (c) IPython Development Team.
 # Distributed under the terms of the Modified BSD License.
 
-from traitlets import Instance, DottedObjectName
+from traitlets import Instance, DottedObjectName, default
 from jupyter_client.managerabc import KernelManagerABC
 from jupyter_client.manager import KernelManager
 from jupyter_client.session import Session
@@ -24,11 +24,14 @@ class InProcessKernelManager(KernelManager):
                       allow_none=True)
     # the client class for KM.client() shortcut
     client_class = DottedObjectName('ipykernel.inprocess.BlockingInProcessKernelClient')
-    def _blocking_class_default(self):
+
+    @default('blocking_class')
+    def _default_blocking_class(self):
         from .blocking import BlockingInProcessKernelClient
         return BlockingInProcessKernelClient
 
-    def _session_default(self):
+    @default('session')
+    def _default_session(self):
         # don't sign in-process messages
         return Session(key=b'', parent=self)
 

ipykernel/ipkernel.py

@@ -91,21 +91,24 @@ def __init__(self, **kwargs):
             'text': "pandas",
             'url': "http://pandas.pydata.org/pandas-docs/stable/",
         },
-    ], config=True)
+    ]).tag(config=True)
 
     # Kernel info fields
     implementation = 'ipython'
     implementation_version = release.version
     language_info = {
-                     'name': 'python',
-                     'version': sys.version.split()[0],
-                     'mimetype': 'text/x-python',
-                     'codemirror_mode': {'name': 'ipython',
-                                         'version': sys.version_info[0]},
-                     'pygments_lexer': 'ipython%d' % (3 if PY3 else 2),
-                     'nbconvert_exporter': 'python',
-                     'file_extension': '.py'
-                    }
+        'name': 'python',
+        'version': sys.version.split()[0],
+        'mimetype': 'text/x-python',
+        'codemirror_mode': {
+            'name': 'ipython',
+            'version': sys.version_info[0]
+        },
+        'pygments_lexer': 'ipython%d' % (3 if PY3 else 2),
+        'nbconvert_exporter': 'python',
+        'file_extension': '.py'
+    }
+
     @property
     def banner(self):
         return self.shell.banner

ipykernel/kernelapp.py

@@ -16,17 +16,16 @@
 from zmq.eventloop.zmqstream import ZMQStream
 
 from IPython.core.application import (
-    BaseIPythonApplication, base_flags, base_aliases, catch_config_error
+        BaseIPythonApplication, base_flags, base_aliases, catch_config_error
 )
 from IPython.core.profiledir import ProfileDir
 from IPython.core.shellapp import (
-    InteractiveShellApp, shell_flags, shell_aliases
+        InteractiveShellApp, shell_flags, shell_aliases
 )
 from IPython.utils import io
 from ipython_genutils.path import filefind, ensure_dir_exists
-from traitlets import (
-    Any, Instance, Dict, Unicode, Integer, Bool, DottedObjectName, Type,
-)
+from traitlets import (Any, Instance, Dict, Unicode, Integer, Bool,
+                       DottedObjectName, Type, default)
 from ipython_genutils.importstring import import_item
 from jupyter_core.paths import jupyter_runtime_dir
 from jupyter_client import write_connection_file
@@ -38,7 +37,7 @@
 from .ipkernel import IPythonKernel
 from .parentpoller import ParentPollerUnix, ParentPollerWindows
 from jupyter_client.session import (
-    Session, session_flags, session_aliases,
+        Session, session_flags, session_aliases,
 )
 from .zmqshell import ZMQInteractiveShell
 
@@ -112,7 +111,7 @@ class IPKernelApp(BaseIPythonApplication, InteractiveShellApp,
     poller = Any() # don't restrict this even though current pollers are all Threads
     heartbeat = Instance(Heartbeat, allow_none=True)
     ports = Dict()
-    
+
     subcommands = {
         'install': (
             'ipykernel.kernelspec.InstallIPythonKernelSpecApp',
@@ -122,7 +121,9 @@ class IPKernelApp(BaseIPythonApplication, InteractiveShellApp,
 
     # connection info:
     connection_dir = Unicode()
-    def _connection_dir_default(self):
+
+    @default('connection_dir')
+    def _default_connection_dir(self):
         return jupyter_runtime_dir()
 
     @property
@@ -132,7 +133,6 @@ def abs_connection_file(self):
         else:
             return self.connection_file
 
-
     # streams, etc.
     no_stdout = Bool(False, help="redirect stdout to the null device").tag(config=True)
     no_stderr = Bool(False, help="redirect stderr to the null device").tag(config=True)
@@ -153,7 +153,7 @@ def abs_connection_file(self):
 
     def init_crash_handler(self):
         sys.excepthook = self.excepthook
-    
+
     def excepthook(self, etype, evalue, tb):
         # write uncaught traceback to 'real' stderr, not zmq-forwarder
         traceback.print_exception(etype, evalue, tb, file=sys.__stderr__)
@@ -241,7 +241,7 @@ def init_sockets(self):
         self.control_socket.linger = 1000
         self.control_port = self._bind_socket(self.control_socket, self.control_port)
         self.log.debug("control ROUTER Channel on port: %i" % self.control_port)
-        
+
         self.init_iopub(context)
 
     def init_iopub(self, context):
@@ -253,7 +253,6 @@ def init_iopub(self, context):
         self.iopub_thread.start()
         # backward-compat: wrap iopub socket API in background thread
         self.iopub_socket = self.iopub_thread.background_socket
-        
 
     def init_heartbeat(self):
         """start the heart beating"""
@@ -312,9 +311,9 @@ def init_io(self):
         if self.displayhook_class:
             displayhook_factory = import_item(str(self.displayhook_class))
             sys.displayhook = displayhook_factory(self.session, self.iopub_socket)
-        
+
         self.patch_io()
-    
+
     def patch_io(self):
         """Patch important libraries that can't handle sys.stdout forwarding"""
         try:
@@ -388,7 +387,7 @@ def init_shell(self):
         self.shell = getattr(self.kernel, 'shell', None)
         if self.shell:
             self.shell.configurables.append(self)
-    
+
     def init_extensions(self):
         super(IPKernelApp, self).init_extensions()
         # BEGIN HARDCODED WIDGETS HACK
@@ -434,7 +433,6 @@ def initialize(self, argv=None):
     def start(self):
         if self.subapp is not None:
             return self.subapp.start()
-        
         if self.poller is not None:
             self.poller.start()
         self.kernel.start()