装饰器获取内部函数的变量值，这个方法简直太有爱了！

转载于 2012-07-09 15:02:00 发布 · 459 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Phenix_Yang/archive/2012/07/09/2579024.html

文章标签：

#javascript #python #ViewUI

本文介绍了一种利用Python的装饰器和inspect模块来动态获取被装饰函数参数的方法，特别是针对那些具有不定数量参数及默认参数的情况，提供了一个实用的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我的需求是这样的，我有很多个函数，其中大部分函数有一个参数叫arg，并且默认值是True，同时可能会有一些别的不确定参数也就是:

def test(x,y,z,arg=True,a=0,b):

我想要写一个装饰器，根据每个函数中的arg的值来做一些事情，但是上述这种情况的话想要准确获取到我调用test这个方法时arg的值就很困难。

最后在三碗同学的帮助下，找到了一个方法：

inspect.getcallargs

整理后我的代码是这样：

# -*- coding: utf-8 -*-
from inspect import getcallargs

def smart_do(func):
    """每次都需要判断arg来看是否做哪些事情好烦有没有
"""
    def inner(*args, **kwargs):
        all_args = getcallargs(func, *args, **kwargs)
        arg = all_args.get('update')
        do something.....
        return func(*args, **kwargs)
    return inner

@smart_do
def a(a,b,arg=True):
    do something...