mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
568 B
Python
22 lines
568 B
Python
5 years ago
|
class Connect:
|
||
|
|
"""
|
||
|
|
This class wraps :func:`~websockets.client.connect` on Python ≥ 3.5.
|
||
|
|
|
||
|
|
This allows using it as an asynchronous context manager.
|
||
|
|
|
||
|
|
"""
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.client = self.__class__.__wrapped__(*args, **kwargs)
|
||
|
|
|
||
|
|
async def __aenter__(self):
|
||
|
|
self.websocket = await self
|
||
|
|
return self.websocket
|
||
|
|
|
||
|
|
async def __aexit__(self, exc_type, exc_value, traceback):
|
||
|
|
await self.websocket.close()
|
||
|
|
|
||
|
|
def __await__(self):
|
||
|
|
return (yield from self.client)
|
||
|
|
|
||
|
|
__iter__ = __await__
|