Correct last commit

This commit is contained in:
rocky
2020-02-06 20:19:29 -05:00
parent 3438e76865
commit 73951840b6

View File

@@ -2356,6 +2356,7 @@ class SourceWalker(GenericASTTraversal, object):
# The function defining a class normally returns locals().
# We don't want this to show up in the source, so remove the node.
if len(ast):
if ast == "stmts" and ast[-1] == "sstmt":
return_locals_parent = ast[-1]
parent_index = 0
@@ -2366,6 +2367,8 @@ class SourceWalker(GenericASTTraversal, object):
if return_locals == RETURN_LOCALS:
if self.hide_internal:
del return_locals_parent[parent_index]
pass
pass
# else:
# print stmt[-1]