LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-09-21 03:08:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support using the client in with blocks

Browse Source
This commit is contained in:
Lonami Exo 2018-06-26 11:26:01 +02:00
parent d4b508bf48
commit 32786d16fd
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
telethon/client/auth.py
View File

@ -1,8 +1,7 @@
import getpass
import hashlib
import sys
import os
import sys
from .messageparse import MessageParseMethods
from .users import UserMethods
@ -459,3 +458,19 @@ class AuthMethods(MessageParseMethods, UserMethods):
))
# endregion
# region with blocks
def __enter__(self):
return self.start()
async def __aenter__(self):
return await self.start()
def __exit__(self, *args):
self.disconnect()
async def __aexit__(self, *args):
await self.disconnect()
# endregion