parse python 函数作为装饰器中的字符串

Question

我正在尝试编写一个函数调试装饰器，它将查看：

def foo(baz):
  bar = 1
  bar = 2
  return bar

并将其包装到：

def foo(baz):
  bar = 1
  print 'bar: {}'.format(bar)
  bar = 2
  print 'bar: {}'.format(bar)
  return bar

我需要像文本一样使用函数，以获取“\w+(?=\s*[=])”，但不知道如何访问它。我有一个装饰器，我从一个有效的博客修改而来，但我只是尝试将其更改为：

class decorator_string_check(object):

   def __init__(self, func):
        self.func = func
        wraps(func)(self)

   def __call__(self, *args, **kwargs):
        print dir(self.func)
        print dir(self.func.__code__)
        print self.func.__code__.__str__()
        ret = self.func(*args, **kwargs)
        return ret

@decorator_string_check
def fake(x):
    y = 6
    y = x
    return y

y = fake(9)

我没有得到任何有价值的东西，即：

['__call__', '__class__', '__closure__', '__code__', '__defaults__', '__delattr__', '__dict__', '__doc__', '__format__', '__get__', '__getattribute__', '__globals__', '__hash__', '__init__', '__module__', '__name__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'func_closure', 'func_code', 'func_defaults', 'func_dict', 'func_doc', 'func_globals', 'func_name']
['__class__', '__cmp__', '__delattr__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__le__', '__lt__', '__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'co_argcount', 'co_cellvars', 'co_code', 'co_consts', 'co_filename', 'co_firstlineno', 'co_flags', 'co_freevars', 'co_lnotab', 'co_name', 'co_names', 'co_nlocals', 'co_stacksize', 'co_varnames']
<code object fake at 0x7f98b8b1d030, file "./logging.py", line 48>

如何使用实际的 "func" 文本，在其上使用运行正则表达式并在装饰器 class 对象中找到我需要的东西？谢谢

Answer 1

首先，我建议你不要做那样的事情。很难获得工作代码，也很难制作正确的版本。

此外，我真的不知道你到底想做什么。这个装饰器是否应该在每次赋值后添加一个 print 语句并显示更新后的值？或者只跟踪给定的变量子集？

也就是说，要获取某些东西的源代码，您可以使用 inspect module in particular the getsource 函数：

In [1]: def test():
   ...:     a = 1
   ...:     b = 2
   ...:     

In [2]: import inspect

In [3]: inspect.getsource(test)
Out[3]: 'def test():\n    a = 1\n    b = 2\n'
In [4]: print(inspect.getsource(test))
def test():
    a = 1
    b = 2

您可以根据需要修改和检查源代码，最后 compile() 新的源代码。

但请注意：

修改源代码时必须小心，因为很容易创建语法无效的代码（想想：多行表达式等）
编译时您希望在与原始函数相同的范围内进行编译。 inspect 模块有一些函数可以让你获得调用你的装饰器的栈帧，你可以从那里获得环境。阅读 here 了解如何处理堆栈。
源代码可能不可用。代码可以编译成字节码，原始文件可以删除。在这种情况下 getsource 只会引发 OSError.

更 "sane" 的解决方案是不查看源代码，而是查看 字节码 。您可以使用 dis 模块来执行此操作。您可以尝试查看变量值何时更改并插入一些将打印该变量的字节码。

请注意，dis 模块在 python3.4+ 中得到了极大增强，因此对于 python 的早期版本，这可能很难。

在尝试此操作之前，您可能应该阅读 Python bytecode hacks, gotos revisited 等文章。他们让您了解如何查看字节码并使用它。

这可能更安全（例如，即使机器上不存在源文件，字节码仍然可以访问），但我仍然认为您的想法是不是一件好事，除了作为练习。

正如 jsbueno 指出的做你想做的事情的正确方法（即 python 调试器）是使用 sys.settrace.

此函数允许您设置跟踪函数，每次执行 "code" 时都会调用该函数。该函数将知道何时调用函数、输入新块等。它使您可以访问将执行代码的框架，因此您应该能够找到您感兴趣的值。

您应该检查 lnotab_notes.txt 文件以了解作为参数提供给此函数的数据如何映射到源代码位置以了解何时执行赋值。

_{等我有空的时候（大概是下周末）我会尝试实现一些基于这种方法的东西来演示它。}

parse python 函数作为装饰器中的字符串

parse python functions as a string within decorator

python

decorator

python-decorators

python-magic