问题描述
当我编写脚本时,我有一个通用的个人函数库,我通过懒惰地导入整个东西来使用它。所以我想清理它,而不是导入数千行代码,我可以让脚本自动选择该特定脚本所需的函数并将它们复制到文件中以供发布。我使用 ast
和 inspect
的组合来实现这一点,但现在我想复制在我的模块中定义全局变量的确切代码行。
例如,如果我在脚本中使用快捷方式 eprint
,它应该刮掉以下行:eprint = Eprinter(verbose=1).eprint
来创建类方法的快捷方式。
我正在尝试采用我在这里找到的代码:https://stackoverflow.com/a/13230346/11343425
class GetAssignments(ast.NodeVisitor):
def visit_Name(self,node):
if isinstance(node.ctx,ast.Store):
if node.id == 'eprint':
print(node.id,node.lineno)
return node.lineno
code = inspect.getsource(printing)
a = GetAssignments()
b = a.visit(ast.parse(code))
print('b =',b)
这将很好地打印 node.lineno,但它返回 None,我不知道为什么。
解决方法
ast.NodeVisitor 意味着被子类化,这意味着 visit_Name
不能返回任何东西。但是,自定义函数可以是这样的中介:
class GetAssignments(ast.NodeVisitor):
def visit_Name(self,node):
if isinstance(node.ctx,ast.Store):
if node.id == self.expr:
print("Found line number",node.lineno)
self.lineno = node.lineno
def search(self,node,expr):
self.expr = expr
self.visit(node)
return self.lineno
code = inspect.getsource(printing)
lineno = GetAssignments().search(ast.parse(code),'eprint')
line = code.split('\n')[lineno-1]
print(lineno,line)