mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Use inspect's iscoroutinefunction, not asyncio's

Browse Source
This commit is contained in:
Dan 2020-08-25 09:32:39 +02:00
parent b057dcb39d
commit 65b5229fe8
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyrogram/client.py
View File

@ -18,6 +18,7 @@
import asyncio import asyncio
import functools import functools
import inspect
import logging import logging
import os import os
import re import re
@ -1025,7 +1026,7 @@ class Client(Methods, Scaffold):
offset += limit offset += limit
if progress: if progress:
if asyncio.iscoroutinefunction(progress): if inspect.iscoroutinefunction(progress):
await progress( await progress(
min(offset, file_size) if file_size != 0 else offset, min(offset, file_size) if file_size != 0 else offset,
file_size, file_size,

3
pyrogram/dispatcher.py
View File

@ -17,6 +17,7 @@
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>. # along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
import asyncio import asyncio
import inspect
import logging import logging
from collections import OrderedDict from collections import OrderedDict
@ -203,7 +204,7 @@ class Dispatcher:
continue continue
try: try:
if asyncio.iscoroutinefunction(handler.callback): if inspect.iscoroutinefunction(handler.callback):
await handler.callback(self.client, *args) await handler.callback(self.client, *args)
else: else:
await self.loop.run_in_executor( await self.loop.run_in_executor(

3
pyrogram/methods/advanced/save_file.py
View File

@ -18,6 +18,7 @@
import asyncio import asyncio
import functools import functools
import inspect
import io import io
import logging import logging
import math import math
@ -182,7 +183,7 @@ class SaveFile(Scaffold):
file_part += 1 file_part += 1
if progress: if progress:
if asyncio.iscoroutinefunction(progress): if inspect.iscoroutinefunction(progress):
await progress(min(file_part * part_size, file_size), file_size, *progress_args) await progress(min(file_part * part_size, file_size), file_size, *progress_args)
else: else:
func = functools.partial( func = functools.partial(