|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = [ 'Client', 'Listener', 'Pipe' ] |
|
|
|
from queue import Queue |
|
|
|
|
|
families = [None] |
|
|
|
|
|
class Listener(object): |
|
|
|
def __init__(self, address=None, family=None, backlog=1): |
|
self._backlog_queue = Queue(backlog) |
|
|
|
def accept(self): |
|
return Connection(*self._backlog_queue.get()) |
|
|
|
def close(self): |
|
self._backlog_queue = None |
|
|
|
@property |
|
def address(self): |
|
return self._backlog_queue |
|
|
|
def __enter__(self): |
|
return self |
|
|
|
def __exit__(self, exc_type, exc_value, exc_tb): |
|
self.close() |
|
|
|
|
|
def Client(address): |
|
_in, _out = Queue(), Queue() |
|
address.put((_out, _in)) |
|
return Connection(_in, _out) |
|
|
|
|
|
def Pipe(duplex=True): |
|
a, b = Queue(), Queue() |
|
return Connection(a, b), Connection(b, a) |
|
|
|
|
|
class Connection(object): |
|
|
|
def __init__(self, _in, _out): |
|
self._out = _out |
|
self._in = _in |
|
self.send = self.send_bytes = _out.put |
|
self.recv = self.recv_bytes = _in.get |
|
|
|
def poll(self, timeout=0.0): |
|
if self._in.qsize() > 0: |
|
return True |
|
if timeout <= 0.0: |
|
return False |
|
with self._in.not_empty: |
|
self._in.not_empty.wait(timeout) |
|
return self._in.qsize() > 0 |
|
|
|
def close(self): |
|
pass |
|
|
|
def __enter__(self): |
|
return self |
|
|
|
def __exit__(self, exc_type, exc_value, exc_tb): |
|
self.close() |
|
|
