35 lines
806 B
Python
35 lines
806 B
Python
from twisted.protocols import basic
|
|
from twisted.internet.error import ConnectionDone
|
|
from eventlet.api import spawn
|
|
from eventlet.twistedutil.protocol import BaseBuffer
|
|
|
|
class LineOnlyReceiver(basic.LineOnlyReceiver):
|
|
|
|
def lineReceived(self, line):
|
|
spawn(self.channel.send, line)
|
|
|
|
def connectionLost(self, reason):
|
|
spawn(self.channel.send_exception, reason.value)
|
|
|
|
class LineOnlyReceiverBuffer(BaseBuffer):
|
|
|
|
protocol_class = LineOnlyReceiver
|
|
|
|
def readline(self):
|
|
return self.channel.receive()
|
|
|
|
def sendline(self, line):
|
|
self.protocol.sendLine(line)
|
|
|
|
# iterator protocol:
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def next(self):
|
|
try:
|
|
return self.readline()
|
|
except ConnectionDone:
|
|
raise StopIteration
|
|
|