Python:functools,方法的包装装饰器:当使用'使得ipython返回正确的定义?'

时间:2014-08-01 17:29:06

标签: python functools

我正在使用functools制作一个装饰器,这让我可以 记录方法调用的详细信息。

我在这里得到了很多帮助来写它...这不是我自己的,我现在还在 学习它是如何工作的......

当我使用时,我经常使用ipython:

import Tools
imagetools = Tools.Tools()
imagetools.save_new_image ?

Type:       instancemethod
String Form:<bound method ImageTools.save_new_image of <ImageTools.ImageTools object      at    0x305d1d0>>
File:       /fs/corpus6/dpc/workspace/python/Modules/MiscTools.py
Definition: imagetools.save_new_image(self, *args, **kwargs)
Docstring:  <no docstring>

这是我的问题路线......除此之外我还没写过文字字符串......:|

Definition: imagetools.save_new_image(self, *args, **kwargs)

我想:

save_new_image_clone(self, data, header, outfile,  coordmap)

打包机:

import functools

def log_method(func):

@functools.wraps(func)
def wrapper(self, *args, **kwargs):
    self.__doc__ = func.__doc__
    # allows me to put the wrapper in helper methods that are sometimes major steps
    if check_for_True_in_last_arg(*args):
        log.debug('''Last arg is "True" which could indicate the caller 
intends the log to work on this method. For that behaviour, use the keyword arg:  log_this=True''')

    if 'log_this' in kwargs and kwargs['log_this'] is not False: 
        args_name = inspect.getargspec(func)[0][1:] #drop the 'self' arg
        args_dict = dict(list(itertools_izip(args_name, args)) + list(kwargs.iteritems()))
        method_name = wrapper.__name__
        args_dict_string = '\nMETHOD: ' + method_name + '\n'

        for k,v in args_dict.iteritems():
            if type(v) is not np.ndarray:
                args_dict_string += "%s: %s\n" %(k,v)

            elif type(v) is np.ndarray:
                args_dict_string += "NUMPY ARRAY:%s \nSHAPE:%s\n" %(k, str(v.shape))

        args_dict_string += '\n'
        log.info(args_dict_string)

    return func(self, *args, **kwargs) 

return wrapper

更新: 从ipython中我做到这一点,看看发生了什么,但它对我没有帮助..

help??
Type:       _Helper
String Form:Type help() for interactive help, or help(object) for help about object.
File:       /usr/lib64/python2.7/site.py
Definition: help(self, *args, **kwds)
Source:

class _Helper(object):
"""Define the builtin 'help'.
This is a wrapper around pydoc.help (with a twist).

"""

def __repr__(self):
    return "Type help() for interactive help, " \
           "or help(object) for help about object."
def __call__(self, *args, **kwds):
    import pydoc
    return pydoc.help(*args, **kwds)

Call def:   help(self, *args, **kwds)

1 个答案:

答案 0 :(得分:1)

你想做什么 - 让一个装饰者的呼号与装饰的功能相同 - 从头开始​​做起来并不简单。有一个名为decorator的第三方模块可以执行此操作:

import decorator

@decorator.decorator
def log_method(func, *args, **kwargs):
    # allows me to put the wrapper in helper methods that are sometimes major steps
    if check_for_True_in_last_arg(*args):
        log.debug('''Last arg is "True" which could indicate the caller 
intends the log to work on this method. For that behaviour, use the keyword arg:  log_this=True''')

    if 'log_this' in kwargs and kwargs['log_this'] is not False: 
        args_name = inspect.getargspec(func)[0][1:] #drop the 'self' arg
        args_dict = dict(list(itertools_izip(args_name, args)) + list(kwargs.iteritems()))
        method_name = wrapper.__name__
        args_dict_string = '\nMETHOD: ' + method_name + '\n'

        for k,v in args_dict.iteritems():
            if type(v) is not np.ndarray:
                args_dict_string += "%s: %s\n" %(k,v)

            elif type(v) is np.ndarray:
                args_dict_string += "NUMPY ARRAY:%s \nSHAPE:%s\n" %(k, str(v.shape))

        args_dict_string += '\n'
        log.info(args_dict_string)

    return func(self, *args, **kwargs) 

class Tools(object):
    @log_method
    def save_new_image(self, data, header, outfile, coordmap):
        """
        Jazz the snaz output baz
        """

然后来自IPython:

In [14]: imagetools = Tools()

In [15]: imagetools.save_new_image?
Type:       instancemethod
String Form:<bound method Tools.save_new_image of <__main__.Tools object at 0xae1e20c>>
File:       /tmp/python-2973tNk.py
Definition: imagetools.save_new_image(self, data, header, outfile, coordmap)
Docstring:  Jazz the snaz output baz

如果您想查看实现此目的的Python代码,请参阅the decorator.FunctionMaker class。它使用inspect来计算原始函数的签名,然后使用字符串格式来编写定义修饰函数的def-statement。然后它调用exec code in evaldict来执行字符串。