LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-07 13:34:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Documentation

Browse Source
This commit is contained in:
Lonami 2018-10-02 18:38:22 +02:00 committed by GitHub
parent ffa0d31ac7
commit e8215e3c12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
telethon/full_sync.py
View File

@ -77,7 +77,18 @@ def _syncify(*types, loop, thread_ident):
__asyncthread = None __asyncthread = None
def enable(loop=None, executor=None, max_workers=1): def enable(*, loop=None, executor=None, max_workers=1):
"""
Enables the fully synchronous mode. You should enable this at
the beginning of your script, right after importing, only once.
**Please** make sure to call `stop` at the end of your script.
You can define the event loop to use and executor, otherwise
the default loop and ``ThreadPoolExecutor`` will be used, in
which case `max_workers` will be passed to it. If you pass a
custom executor, `max_workers` will be ignored.
"""
global __asyncthread global __asyncthread
if __asyncthread is not None: if __asyncthread is not None:
raise RuntimeError("full_sync can only be enabled once") raise RuntimeError("full_sync can only be enabled once")
@ -91,14 +102,16 @@ def enable(loop=None, executor=None, max_workers=1):
asyncio.set_event_loop(loop) asyncio.set_event_loop(loop)
loop.run_forever() loop.run_forever()
__asyncthread = threading.Thread(target=start, name="__telethon_async_thread__", __asyncthread = threading.Thread(
daemon=True) target=start, name="__telethon_async_thread__", daemon=True
)
__asyncthread.start() __asyncthread.start()
__asyncthread.loop = loop __asyncthread.loop = loop
__asyncthread.executor = executor __asyncthread.executor = executor
TelegramClient.__init__ = functools.partialmethod(TelegramClient.__init__, TelegramClient.__init__ = functools.partialmethod(
loop=loop) TelegramClient.__init__, loop=loop
)
_syncify(TelegramClient, Draft, Dialog, MessageButton, ChatGetter, _syncify(TelegramClient, Draft, Dialog, MessageButton, ChatGetter,
SenderGetter, Forward, Message, InlineResult, Conversation, SenderGetter, Forward, Message, InlineResult, Conversation,
@ -137,6 +150,10 @@ def enable(loop=None, executor=None, max_workers=1):
def stop(): def stop():
"""
Stops the fully synchronous code. You
should call this before your script exits.
"""
global __asyncthread global __asyncthread
if not __asyncthread: if not __asyncthread:
raise RuntimeError("Can't find asyncio thread") raise RuntimeError("Can't find asyncio thread")