improve docs and code
parent
33f96c066f
commit
b25419fcd9
|
@ -53,6 +53,8 @@ You can use them like that:
|
|||
:members:
|
||||
.. autoclass:: i2plib.StreamConnection
|
||||
:members:
|
||||
.. autoclass:: i2plib.StreamAcceptor
|
||||
:members:
|
||||
|
||||
Utilities
|
||||
---------
|
||||
|
|
|
@ -212,23 +212,10 @@ class StreamConnection:
|
|||
async def __aenter__(self):
|
||||
self.reader, self.writer = await stream_connect(self.session_name,
|
||||
self.destination, sam_address=self.sam_address, loop=self.loop)
|
||||
self.read = self.reader.read
|
||||
self.write = self.writer.write
|
||||
return self
|
||||
|
||||
async def read(self, length):
|
||||
"""Read data from socket
|
||||
|
||||
:param length: buffer length
|
||||
:return: data
|
||||
"""
|
||||
return await self.reader.read(length)
|
||||
|
||||
def write(self, data):
|
||||
"""Write data to socket
|
||||
|
||||
:param data: data
|
||||
"""
|
||||
self.writer.write(data)
|
||||
|
||||
async def __aexit__(self, exc_type, exc, tb):
|
||||
### TODO handle exceptions
|
||||
self.writer.close()
|
||||
|
@ -250,23 +237,10 @@ class StreamAcceptor:
|
|||
async def __aenter__(self):
|
||||
self.reader, self.writer = await stream_accept(self.session_name,
|
||||
sam_address=self.sam_address, loop=self.loop)
|
||||
self.read = self.reader.read
|
||||
self.write = self.writer.write
|
||||
return self
|
||||
|
||||
async def read(self, length):
|
||||
"""Read data from socket
|
||||
|
||||
:param length: buffer length
|
||||
:return: data
|
||||
"""
|
||||
return await self.reader.read(length)
|
||||
|
||||
def write(self, data):
|
||||
"""Write data to socket
|
||||
|
||||
:param data: data
|
||||
"""
|
||||
self.writer.write(data)
|
||||
|
||||
async def __aexit__(self, exc_type, exc, tb):
|
||||
### TODO handle exceptions
|
||||
self.writer.close()
|
||||
|
|
Loading…
Reference in New Issue