improve docs and code

master
Viktor Villainov 2018-12-27 02:08:32 -05:00
parent 33f96c066f
commit b25419fcd9
No known key found for this signature in database
GPG Key ID: 8EB38B46F33BAF2F
2 changed files with 6 additions and 30 deletions

View File

@ -53,6 +53,8 @@ You can use them like that:
:members:
.. autoclass:: i2plib.StreamConnection
:members:
.. autoclass:: i2plib.StreamAcceptor
:members:
Utilities
---------

View File

@ -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()