1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gh-120200: Fix inspect.iscoroutinefunction(inspect) is True corner case (#120214)

Browse Source
This commit is contained in:
Nikita Sobolev 2024-06-07 18:48:31 +03:00 committed by GitHub
parent 9d6604222e
commit 10fb1b8f36
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Lib/inspect.py
View File

@ -403,13 +403,13 @@ def isgeneratorfunction(obj):
return _has_code_flag(obj, CO_GENERATOR) return _has_code_flag(obj, CO_GENERATOR)
# A marker for markcoroutinefunction and iscoroutinefunction. # A marker for markcoroutinefunction and iscoroutinefunction.
_is_coroutine_marker = object() _is_coroutine_mark = object()
def _has_coroutine_mark(f): def _has_coroutine_mark(f):
while ismethod(f): while ismethod(f):
f = f.__func__ f = f.__func__
f = functools._unwrap_partial(f) f = functools._unwrap_partial(f)
return getattr(f, "_is_coroutine_marker", None) is _is_coroutine_marker return getattr(f, "_is_coroutine_marker", None) is _is_coroutine_mark
def markcoroutinefunction(func): def markcoroutinefunction(func):
""" """
@ -417,7 +417,7 @@ def markcoroutinefunction(func):
""" """
if hasattr(func, '__func__'): if hasattr(func, '__func__'):
func = func.__func__ func = func.__func__
func._is_coroutine_marker = _is_coroutine_marker func._is_coroutine_marker = _is_coroutine_mark
return func return func
def iscoroutinefunction(obj): def iscoroutinefunction(obj):

1
Lib/test/test_inspect/test_inspect.py
View File

@ -235,6 +235,7 @@ class TestPredicates(IsTestBase):
gen_coroutine_function_example)))) gen_coroutine_function_example))))
self.assertFalse(inspect.iscoroutinefunction(gen_coro_pmi)) self.assertFalse(inspect.iscoroutinefunction(gen_coro_pmi))
self.assertFalse(inspect.iscoroutinefunction(gen_coro_pmc)) self.assertFalse(inspect.iscoroutinefunction(gen_coro_pmc))
self.assertFalse(inspect.iscoroutinefunction(inspect))
self.assertFalse(inspect.iscoroutine(gen_coro)) self.assertFalse(inspect.iscoroutine(gen_coro))
self.assertTrue( self.assertTrue(