TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-17 21:22:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for "async with" context manager

Browse Source
This commit is contained in:
Dan 2019-03-16 19:56:25 +01:00
parent a329e56259
commit a06885dd14
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/client/client.py
View File

@ -243,6 +243,12 @@ class Client(Methods, BaseClient):
def __exit__(self, *args): def __exit__(self, *args):
self.stop() self.stop()
async def __aenter__(self):
return await self.start()
async def __aexit__(self, *args):
await self.stop()
@property @property
def proxy(self): def proxy(self):
return self._proxy return self._proxy