Python 函数内省

函数内省(function introspection)

除了__doc__属性, 函数对象还有很多属性,对于下面的函数,可以使用dir()查看函数具有的属性:

def factorial(n):
    return 1 if n == 1 else n*factorial(n - 1)
>>> dir(factorial)
['__annotations__', '__call__', '__class__', '__closure__', '__code__', 
'__defaults__', '__delattr__', '__dict__', '__dir__', '__doc__', '__eq__', 
'__format__', '__ge__', '__get__', '__getattribute__', '__globals__', '__gt__',
 '__hash__', '__init__', '__kwdefaults__', '__le__', '__lt__', '__module__', 
'__name__', '__ne__', '__new__', '__qualname__', '__reduce__', 
'__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', 
'__subclasshook__']

其中大多数是Python常规类都有的属性,下面重点看看常规对象没有而函数对象有的属性:

>>> class C:pass
... 
>>> obj = C()
>>> def func():pass
... 
>>> sorted(set(dir(func)) - set(dir(obj))) # 计算差集,然后排序
['__annotations__', '__call__', '__closure__', '__code__', '__defaults__', '__get__', '__globals__', '__kwdefaults__', '__name__', '__qualname__']

对于上面列出的函数特有属性,说明如下:

接下来我们讨论__defaults__, __code__, __annotations__的作用

(一) __defaults__, __code__:

函数对象有个__defaults__属性,他的值是一个元组,里面保存着定位参数和关键字参数的默认值。然而,参数的名称在__code__属性中, 它的值是一个code对象引用,自身也有很多属性

仅限关键字参数的默认值在__kwdefaults__属性中,

我们借用下面的示例说明这些属性的用途:

def clip(text=None, max_length=80):
    """
    Return text clipped at the last space before or after max_len
    """
    end = None
    if len(text) > max_length:
        space_before = text.rfind(' ', 0, max_length)
        if space_before >= 0:
            end = space_before
        else:
            space_after = text.rfind(' ', max_length)
            if space_after >= 0:
                end = space_after

        if end is None:
            end = len(text)

        return text[:end].rstrip()

提取clip函数参数的信息:

>>> from Example5_15 import clip
>>> clip.__defaults__
(None, 80)
>>> clip.__code__
<code object clip at 0x7f8e50658f60, file "/root/python_demo/Example5_15.py", line 1>
>>> clip.__code__.co_varnames
('text', 'max_length', 'end', 'space_before', 'space_after')
>>> clip.__code__.co_argcount
2

从控制台信息可以看出,__code__.co_varnames包含函数参数,并且还包含函数体中的局部变量,函数参数个数由__code__.co_argcount确定(这里不包含前缀为*或者**的变长参数)

从这些属性提取参数信息并不是很方便,我们有更好的方式-------使用inspect模块:

>>> from inspect import signature
>>> sig = signature(clip)
>>> sig
<Signature (text=None, max_length=80)>
>>> for name, param in sig.parameters.items():
...     print(param.kind, ':', name, '=', param.default)
... 
POSITIONAL_OR_KEYWORD : text = None
POSITIONAL_OR_KEYWORD : max_length = 80

这样就好多了, Inspect.signature 函数返回一个inspect.Signature对象,它有一个parameter属性,这是一个有序映射,把参数名和inspect.Parameter对象对应起来,各个Parameter属性也有自己的属性,比如name, default, kind

如果是非默认值参数,特殊的inspect._empty值表示没有默认值,考虑到上面例子的None是有效默认值,这样处理是合理的

(二) __annotations__属性:

我们把上面的clip函数添加注解,如下所示:

def clip(text: str =None, max_length: 'int > 0'=80) -> str:
    """
    :return text clipped at the last space before or after max_len
    :param text:
    :param max_length:
    """
    end = None
    if len(text) > max_length:
        space_before = text.rfind(' ', 0, max_length)
        if space_before >= 0:
            end = space_before
        else:
            space_after = text.rfind(' ', max_length)
            if space_after >= 0:
                end = space_after

        if end is None:
            end = len(text)

        return text[:end].rstrip()

注意函数头部的第一行

函数声明中的各个参数可以在:之后添加注解,添加注解的方法:

(1)参数有默认值,注解放在参数名和=号之间

(2)返回值,在)和函数声明尾部的:之间添加->和一个表达式,表达式可以是任何类型

对于注解,Python解释器不做任何处理,只是存储在__annotations__属性(字典形式)中:

>>> from Example5_15 import clip
>>> clip.__annotations__
{'return': <class 'str'>, 'max_length': 'int > 0', 'text': <class 'str'>}

 

posted on 2017-11-06 15:57  _Joshua  阅读(3996)  评论(0编辑  收藏  举报

导航