52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
|
|
||
|
|
|
||
|
|
class Socket(object):
|
||
|
|
def __init__(self, socket, on_read, encoding="utf8"):
|
||
|
|
self.socket = socket
|
||
|
|
self._on_read = on_read
|
||
|
|
self.encoding = encoding
|
||
|
|
|
||
|
|
self._write_buffer = b""
|
||
|
|
self._read_buffer = b""
|
||
|
|
self.delimiter = None
|
||
|
|
self.length = None
|
||
|
|
self.connected = True
|
||
|
|
|
||
|
|
def fileno(self):
|
||
|
|
return self.socket.fileno()
|
||
|
|
|
||
|
|
def disconnect(self):
|
||
|
|
self.connected = False
|
||
|
|
|
||
|
|
def _decode(self, s):
|
||
|
|
return s.decode(self.encoding) if self.encoding else s
|
||
|
|
def _encode(self, s):
|
||
|
|
return s.encode(self.encoding) if self.encoding else s
|
||
|
|
|
||
|
|
def read(self):
|
||
|
|
data = self.socket.recv(1024)
|
||
|
|
if not data:
|
||
|
|
return None
|
||
|
|
|
||
|
|
data = self._read_buffer+data
|
||
|
|
self._read_buffer = b""
|
||
|
|
if not self.delimiter == None:
|
||
|
|
data_split = data.split(delimiter)
|
||
|
|
if data_split[-1]:
|
||
|
|
self._read_buffer = data_split.pop(-1)
|
||
|
|
return [self._decode(data) for data in data_split]
|
||
|
|
return [data.decode(self.encoding)]
|
||
|
|
|
||
|
|
def parse_data(self, data):
|
||
|
|
self._on_read(self, data)
|
||
|
|
|
||
|
|
def send(self, data):
|
||
|
|
self._write_buffer += self._encode(data)
|
||
|
|
|
||
|
|
def _send(self):
|
||
|
|
self._write_buffer = self._write_buffer[self.socket.send(
|
||
|
|
self._write_buffer):]
|
||
|
|
|
||
|
|
def waiting_send(self):
|
||
|
|
return bool(len(self._write_buffer))
|