hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-22 21:16:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove code to syncify async generator functions

Browse Source
This commit is contained in:
Lonami Exo 2019-02-27 10:19:08 +01:00
parent 5b8e6531fa
commit 49d8a3fb33
1 changed files with 5 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
telethon/sync.py
View File

@ -14,8 +14,6 @@ import asyncio
import functools import functools
import inspect import inspect
from async_generator import isasyncgenfunction
from .client.telegramclient import TelegramClient from .client.telegramclient import TelegramClient
from .tl.custom import ( from .tl.custom import (
Draft, Dialog, MessageButton, Forward, Message, InlineResult, Conversation Draft, Dialog, MessageButton, Forward, Message, InlineResult, Conversation
@ -24,22 +22,7 @@ from .tl.custom.chatgetter import ChatGetter
from .tl.custom.sendergetter import SenderGetter from .tl.custom.sendergetter import SenderGetter
class _SyncGen: def _syncify_wrap(t, method_name):
def __init__(self, gen):
self.gen = gen
def __iter__(self):
return self
def __next__(self):
try:
return asyncio.get_event_loop() \
.run_until_complete(self.gen.__anext__())
except StopAsyncIteration:
raise StopIteration from None
def _syncify_wrap(t, method_name, gen):
method = getattr(t, method_name) method = getattr(t, method_name)
@functools.wraps(method) @functools.wraps(method)
@ -48,8 +31,6 @@ def _syncify_wrap(t, method_name, gen):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
if loop.is_running(): if loop.is_running():
return coro return coro
elif gen:
return _SyncGen(coro)
else: else:
return loop.run_until_complete(coro) return loop.run_until_complete(coro)
@ -64,13 +45,14 @@ def syncify(*types):
into synchronous, which return either the coroutine or the result into synchronous, which return either the coroutine or the result
based on whether ``asyncio's`` event loop is running. based on whether ``asyncio's`` event loop is running.
""" """
# Our asynchronous generators all are `RequestIter`, which already
# provide a synchronous iterator variant, so we don't need to worry
# about asyncgenfunction's here.
for t in types: for t in types:
for name in dir(t): for name in dir(t):
if not name.startswith('_') or name == '__call__': if not name.startswith('_') or name == '__call__':
if inspect.iscoroutinefunction(getattr(t, name)): if inspect.iscoroutinefunction(getattr(t, name)):
_syncify_wrap(t, name, gen=False) _syncify_wrap(t, name)
elif isasyncgenfunction(getattr(t, name)):
_syncify_wrap(t, name, gen=True)
syncify(TelegramClient, Draft, Dialog, MessageButton, syncify(TelegramClient, Draft, Dialog, MessageButton,