gh-106727: Add __module__ check for inspect.getsource(cls) (#106968)

This commit is contained in:
Tian Gao 2023-07-24 12:12:45 -08:00 committed by GitHub
parent 8ebc9fc321
commit b383703491
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 30 additions and 1 deletions

View file

@ -1078,7 +1078,8 @@ class _ClassFinder(ast.NodeVisitor):
# First, let's see if there are any method definitions
for member in self.cls.__dict__.values():
if isinstance(member, types.FunctionType):
if (isinstance(member, types.FunctionType) and
member.__module__ == self.cls.__module__):
for lineno, end_lineno in self.lineno_found:
if lineno <= member.__code__.co_firstlineno <= end_lineno:
return lineno