• decorate all function in all module


    需求:

    有package db_api,其下有很多 module 如 plane.py ship.py ufo.py。这些module内定义了方法如 plane.fly(), ship.float(),ufo.siu()。现在希望有一个装饰器来给装饰这些函数,让其打印log,log中包含调用的参数以及return。

    解决办法

    首先定义 decorator

    def db_log(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            func_name = str(func) if func else ''
            log_args = str(args) if args else ''
            log_kwargs = str(kwargs) if kwargs else ''
            eventLog.info("call %s" % func_name)
            eventLog.info("args are %s" % log_args)
            eventLog.info("kwargs are %s " % log_kwargs)
    
            ret = func(*args, **kwargs)
            log_ret = str(ret) if ret else ''
            eventLog.info("ret are %s" % log_ret)
            return ret
    
        return wrapper
    

    这没什么问题,问题是,接下来你要手动的去装饰之前的每一个方法如 fly 等。 这里我们用一段代码动态的patch 这些函数

    在db_api/init.py 中会import 所有module

    import plane
    import ufo
    import ship
    

    然后用下面的代码动态装饰

    for module_obj in dir():
        if isinstance(eval(module_obj),type(plane)) and eval(module_obj).__package__ == 'db_api':
            # Patch call ables
            module_obj = eval(module_obj)
            for func_obj in dir(module_obj):
                if callable(getattr(module_obj, func_obj)) and not func_obj.startswith('_') and isinstance(getattr(module_obj, func_obj),type(lambda x: x)):
                    # func_obj = db_log(func_obj)
                    setattr(module_obj,func_obj,db_log(getattr(module_obj, func_obj)))
  • 相关阅读:
    数据中心基础设施自动化运维之puppet专项
    centos git版本服务器配置
    安装PHP出现make: *** [sapi/cli/php] Error 1 解决办法
    php5.5 yum源
    CentOS下源码安装Apache2.4+PHP5.4+MySQL5.5
    php升级到5.4
    puppet重申证书
    passwnger
    Http状态消息
    UVA 10168 Summation of Four Primes(数论)
  • 原文地址:https://www.cnblogs.com/kramer/p/6775418.html
走看看 - 开发者的网上家园