transpub/Spectrum2/iochannel.py

47 lines
1 KiB
Python
Raw Normal View History

2016-04-09 22:21:58 +00:00
# use unicode encoding for all literals by default (for python2.x)
from __future__ import unicode_literals
2013-05-31 18:53:21 +00:00
import asyncore, socket
2015-02-27 16:06:30 +00:00
import logging
import sys
2013-05-31 18:53:21 +00:00
class IOChannel(asyncore.dispatcher):
def __init__(self, host, port, callback, closeCallback):
2013-05-31 18:53:21 +00:00
asyncore.dispatcher.__init__(self)
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
self.connect((host, port))
self.logger = logging.getLogger(self.__class__.__name__)
2013-05-31 18:53:21 +00:00
self.callback = callback
self.closeCallback = closeCallback
2016-04-10 20:48:04 +00:00
self.buffer = bytes("")
2013-05-31 18:53:21 +00:00
def sendData(self, data):
self.buffer += data
def handle_connect(self):
pass
def handle_close(self):
self.close()
def handle_read(self):
data = self.recv(65536)
self.callback(data)
def handle_write(self):
sent = self.send(self.buffer)
self.buffer = self.buffer[sent:]
def handle_close(self):
self.logger.info('Connection to backend closed, terminating.')
self.close()
self.closeCallback()
2013-05-31 18:53:21 +00:00
def writable(self):
return (len(self.buffer) > 0)
def readable(self):
return True