upstream/ipython Files · IPython/kernel/inprocess/socket.py

fix some unicode in zmqhandlers...

fix some unicode in zmqhandlers prevented session identity from being set properly, which in turn prevented stdin from working in the notebook in Python 3. closes

MinRK - - Load All Authors

File last commit:

r9375:4d245182


                r11330:c60a91d4

Download file

             socket.py
        
                    63 lines
            
             | 2.1 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / kernel / inprocess / socket.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        epatters
    
Implement EmbeddedKernel.

              r8411
            
      """ Defines a dummy socket implementing (part of) the zmq.Socket interface. """

      #-----------------------------------------------------------------------------

      #  Copyright (C) 2012  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

      #-----------------------------------------------------------------------------

      # Standard library imports.

        epatters
    
Add abstract base class (ABC) for sockets used in kernel.

              r8418
            
      import abc

        epatters
    
Implement EmbeddedKernel.

              r8411
            
      import Queue

      # System library imports.

      import zmq

      # Local imports.

      from IPython.utils.traitlets import HasTraits, Instance, Int

      #-----------------------------------------------------------------------------

        epatters
    
Add abstract base class (ABC) for sockets used in kernel.

              r8418
            
      # Generic socket interface

      #-----------------------------------------------------------------------------

      class SocketABC(object):

          __metaclass__ = abc.ABCMeta

          @abc.abstractmethod

          def recv_multipart(self, flags=0, copy=True, track=False):

              raise NotImplementedError

          @abc.abstractmethod

          def send_multipart(self, msg_parts, flags=0, copy=True, track=False):

              raise NotImplementedError

      SocketABC.register(zmq.Socket)

      #-----------------------------------------------------------------------------

        epatters
    
Implement EmbeddedKernel.

              r8411
            
      # Dummy socket class

      #-----------------------------------------------------------------------------

      class DummySocket(HasTraits):

          """ A dummy socket implementing (part of) the zmq.Socket interface. """

          queue = Instance(Queue.Queue, ())

          message_sent = Int(0) # Should be an Event

          #-------------------------------------------------------------------------

        epatters
    
Add abstract base class (ABC) for sockets used in kernel.

              r8418
            
          # Socket interface

        epatters
    
Implement EmbeddedKernel.

              r8411
            
          #-------------------------------------------------------------------------

          def recv_multipart(self, flags=0, copy=True, track=False):

              return self.queue.get_nowait()

          def send_multipart(self, msg_parts, flags=0, copy=True, track=False):

              msg_parts = map(zmq.Message, msg_parts)

              self.queue.put_nowait(msg_parts)

              self.message_sent += 1

        epatters
    
Add abstract base class (ABC) for sockets used in kernel.

              r8418
            
      SocketABC.register(DummySocket)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

epatters Implement EmbeddedKernel.	r8411	""" Defines a dummy socket implementing (part of) the zmq.Socket interface. """

		#-----------------------------------------------------------------------------
		# Copyright (C) 2012 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
		#-----------------------------------------------------------------------------

		# Standard library imports.
epatters Add abstract base class (ABC) for sockets used in kernel.	r8418	import abc
epatters Implement EmbeddedKernel.	r8411	import Queue

		# System library imports.
		import zmq

		# Local imports.
		from IPython.utils.traitlets import HasTraits, Instance, Int

		#-----------------------------------------------------------------------------
epatters Add abstract base class (ABC) for sockets used in kernel.	r8418	# Generic socket interface
		#-----------------------------------------------------------------------------

		class SocketABC(object):
		__metaclass__ = abc.ABCMeta

		@abc.abstractmethod
		def recv_multipart(self, flags=0, copy=True, track=False):
		raise NotImplementedError

		@abc.abstractmethod
		def send_multipart(self, msg_parts, flags=0, copy=True, track=False):
		raise NotImplementedError

		SocketABC.register(zmq.Socket)

		#-----------------------------------------------------------------------------
epatters Implement EmbeddedKernel.	r8411	# Dummy socket class
		#-----------------------------------------------------------------------------

		class DummySocket(HasTraits):
		""" A dummy socket implementing (part of) the zmq.Socket interface. """

		queue = Instance(Queue.Queue, ())
		message_sent = Int(0) # Should be an Event

		#-------------------------------------------------------------------------
epatters Add abstract base class (ABC) for sockets used in kernel.	r8418	# Socket interface
epatters Implement EmbeddedKernel.	r8411	#-------------------------------------------------------------------------

		def recv_multipart(self, flags=0, copy=True, track=False):
		return self.queue.get_nowait()

		def send_multipart(self, msg_parts, flags=0, copy=True, track=False):
		msg_parts = map(zmq.Message, msg_parts)
		self.queue.put_nowait(msg_parts)
		self.message_sent += 1
epatters Add abstract base class (ABC) for sockets used in kernel.	r8418
		SocketABC.register(DummySocket)