From 401709d040df302cdf3cd4a7e0d7703c90ff2d9e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edgar=20Ram=C3=ADrez-Mondrag=C3=B3n?= Date: Thu, 17 Oct 2024 16:28:46 -0600 Subject: [PATCH] Use `inspect.iscoroutinefunction` instead of `asyncio.iscoroutinefunction` --- backoff/_async.py | 13 +++++++------ backoff/_decorator.py | 6 +++--- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/backoff/_async.py b/backoff/_async.py index 82fd477..c24587c 100644 --- a/backoff/_async.py +++ b/backoff/_async.py @@ -1,5 +1,6 @@ # coding:utf-8 import datetime +import inspect import functools import asyncio from datetime import timedelta @@ -8,7 +9,7 @@ def _ensure_coroutine(coro_or_func): - if asyncio.iscoroutinefunction(coro_or_func): + if inspect.iscoroutinefunction(coro_or_func): return coro_or_func else: @functools.wraps(coro_or_func) @@ -47,10 +48,10 @@ def retry_predicate(target, wait_gen, predicate, on_giveup = _ensure_coroutines(on_giveup) # Easy to implement, please report if you need this. - assert not asyncio.iscoroutinefunction(max_tries) - assert not asyncio.iscoroutinefunction(jitter) + assert not inspect.iscoroutinefunction(max_tries) + assert not inspect.iscoroutinefunction(jitter) - assert asyncio.iscoroutinefunction(target) + assert inspect.iscoroutinefunction(target) @functools.wraps(target) async def retry(*args, **kwargs): @@ -124,8 +125,8 @@ def retry_exception(target, wait_gen, exception, giveup = _ensure_coroutine(giveup) # Easy to implement, please report if you need this. - assert not asyncio.iscoroutinefunction(max_tries) - assert not asyncio.iscoroutinefunction(jitter) + assert not inspect.iscoroutinefunction(max_tries) + assert not inspect.iscoroutinefunction(jitter) @functools.wraps(target) async def retry(*args, **kwargs): diff --git a/backoff/_decorator.py b/backoff/_decorator.py index 77ed8c2..ca5d0ff 100644 --- a/backoff/_decorator.py +++ b/backoff/_decorator.py @@ -1,5 +1,5 @@ # coding:utf-8 -import asyncio +import inspect import logging import operator from typing import Any, Callable, Iterable, Optional, Type, Union @@ -98,7 +98,7 @@ def decorate(target): log_level=giveup_log_level ) - if asyncio.iscoroutinefunction(target): + if inspect.iscoroutinefunction(target): retry = _async.retry_predicate else: retry = _sync.retry_predicate @@ -198,7 +198,7 @@ def decorate(target): log_level=giveup_log_level, ) - if asyncio.iscoroutinefunction(target): + if inspect.iscoroutinefunction(target): retry = _async.retry_exception else: retry = _sync.retry_exception