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