##// END OF EJS Templates
require pyzmq >= 2.0.10.1...
require pyzmq >= 2.0.10.1 closes gh-268

File last commit:

r3269:8c0b3978
r3318:729f8adf
Show More
iostream.py
84 lines | 2.7 KiB | text/x-python | PythonLexer
Brian Granger
Separating kernel into smaller pieces.
r2754 import sys
import time
from cStringIO import StringIO
from session import extract_header, Message
Fernando Perez
Add missing flush of output streams on execute
r2938 from IPython.utils import io
Brian Granger
Separating kernel into smaller pieces.
r2754 #-----------------------------------------------------------------------------
# Stream classes
#-----------------------------------------------------------------------------
class OutStream(object):
"""A file like object that publishes the stream to a 0MQ PUB socket."""
# The time interval between automatic flushes, in seconds.
flush_interval = 0.05
def __init__(self, session, pub_socket, name):
self.session = session
self.pub_socket = pub_socket
self.name = name
self.parent_header = {}
self._new_buffer()
def set_parent(self, parent):
self.parent_header = extract_header(parent)
def close(self):
self.pub_socket = None
def flush(self):
Fernando Perez
Add missing flush of output streams on execute
r2938 #io.rprint('>>>flushing output buffer: %s<<<' % self.name) # dbg
Brian Granger
Separating kernel into smaller pieces.
r2754 if self.pub_socket is None:
raise ValueError(u'I/O operation on closed file')
else:
data = self._buffer.getvalue()
if data:
content = {u'name':self.name, u'data':data}
MinRK
all sends/recvs now via Session.send/recv....
r3269 msg = self.session.send(self.pub_socket, u'stream', content=content,
Brian Granger
Separating kernel into smaller pieces.
r2754 parent=self.parent_header)
Fernando Perez
Add missing flush of output streams on execute
r2938 io.raw_print(msg)
Brian Granger
Separating kernel into smaller pieces.
r2754
self._buffer.close()
self._new_buffer()
def isatty(self):
return False
def next(self):
raise IOError('Read not supported on a write only stream.')
def read(self, size=-1):
raise IOError('Read not supported on a write only stream.')
def readline(self, size=-1):
raise IOError('Read not supported on a write only stream.')
def write(self, string):
if self.pub_socket is None:
raise ValueError('I/O operation on closed file')
else:
Fernando Perez
Unicode fixes, basic input/printing of unicode works.
r3038 # We can only send raw bytes, not unicode objects, so we encode
# into utf-8 for all frontends if we get unicode inputs.
if type(string) == unicode:
string = string.encode('utf-8')
Brian Granger
Separating kernel into smaller pieces.
r2754 self._buffer.write(string)
current_time = time.time()
if self._start <= 0:
self._start = current_time
elif current_time - self._start > self.flush_interval:
self.flush()
def writelines(self, sequence):
if self.pub_socket is None:
raise ValueError('I/O operation on closed file')
else:
for string in sequence:
self.write(string)
def _new_buffer(self):
self._buffer = StringIO()
Fernando Perez
Add missing flush of output streams on execute
r2938 self._start = -1