hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-18 02:56:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Conversation.cancel() (#1030)

Browse Source
This commit is contained in:
Kyle2142 2018-10-25 15:37:36 +02:00 committed by Lonami
parent 2c61c50671
commit eacfa226fd
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/tl/custom/conversation.py
View File

@ -418,10 +418,14 @@ class Conversation(ChatGetter):
return self return self
def cancel(self):
"""Cancels the current conversation and exits the context manager."""
raise _ConversationCancelled()
def __exit__(self, *args): def __exit__(self, *args):
return self._client.loop.run_until_complete(self.__aexit__(*args)) return self._client.loop.run_until_complete(self.__aexit__(*args))
async def __aexit__(self, *args): async def __aexit__(self, exc_type, exc_val, exc_tb):
chat_id = utils.get_peer_id(self._chat_peer) chat_id = utils.get_peer_id(self._chat_peer)
if self._client._ids_in_conversations[chat_id] == 1: if self._client._ids_in_conversations[chat_id] == 1:
del self._client._ids_in_conversations[chat_id] del self._client._ids_in_conversations[chat_id]
@ -430,3 +434,8 @@ class Conversation(ChatGetter):
del self._client._conversations[self._id] del self._client._conversations[self._id]
self._cancel_all() self._cancel_all()
return isinstance(exc_val, _ConversationCancelled)
class _ConversationCancelled(InterruptedError):
pass