Python標準庫inspect

inspect模塊用於收集python對象的信息，可以獲取類或函數的參數的信息，源碼，解析堆棧，對對象進行類型檢查等等，有幾個好用的方法：

getargspec(func)

返回一個命名元組ArgSpect(args, varargs, keywords, defaults)，args是函數位置參數名列表，varargs是*參數名，keywords是**參數名，defaults是預設參數值的元組。

在用__init__參數自動初始化實例屬性的實踐中，是用位元組碼對象的co_varnames屬性來獲取函數的位置參數名的：

def attr_from_locals(locals_dict):
    self = locals_dict.pop('self')
    code = self.__init__.__func__.__code__
    args = code.co_varnames[1:code.co_argcount]
    for k in args:
        setattr(self, k, locals_dict[k])
         
class Foo(object):
    def __init__(self, name, color, num=1):
        x = 1
        attr_from_locals(locals())

而當__init__方法使用**特殊參數接收任意數量的關鍵字參數時，上述代碼是不適用的。可行的辦法是使用位元組碼的co_flags屬性來判斷**參數是否存在。

函數使用*args語法來接受任意數量的位置參數時，co_flags置位0x04，使用**kwargs語法時，置位0x08，函數為一個生成器時，置位0x2000，其它位保留：

>>> def foo(x, *args, **kwargv):
	    pass

>>> foo.__code__.co_varnames
('x', 'args', 'kwargv')
>>> foo.__code__.co_flags & 0x04
4
>>> foo.__code__.co_flags & 0x08
8

inspect模塊的getargspec()方法正是用此判斷來獲取函數的特殊參數的。現在可以方便的獲取__init__的**參數了：

import inspect


def attr_from_locals(locals_dict):
    self = locals_dict.pop('self')
    args = inspect.getargspec(self.__init__.__func__).args[1:]
    for k in args:
        setattr(self, k, locals_dict[k])
    keywords = inspect.getargspec(self.__init__.__func__).keywords
    if keywords:
        keywords_dict = locals_dict[keywords]
        for k in keywords_dict:
            setattr(self, k, keywords_dict[k])

       
class Foo(object):
    def __init__(self, name, **kwargv):
        attr_from_locals(locals())


f = Foo('bar', color='yellow', num=1)
print f.__dict__

結果為：

{'color': 'yellow', 'num': 1, 'name': 'bar'}

對象已經正確的初始化了。

getmembers(object[, predicate])

返回一個包含對象的所有成員的(name, value)列表。返回的內容比對象的__dict__包含的內容多，源碼是通過dir()實現的。

predicate是一個可選的函數參數，被此函數判斷為True的成員才被返回。

getmodule(object)

返回定義對象的模塊

getsource(object)

返回對象的源代碼

getsourcelines(object)

返回一個元組，元組第一項為對象源代碼行的列表，第二項是第一行源代碼的行號

ismodule,isclass,ismethod,isfunction,isbuiltin

一系列判斷對象類型的方法，大都是包裝了isinstance(object, types.FunctionType)之類語句的函數。

現在可以用類型判斷來返回一個類的方法了：

class Foo(object):
    '''Foo doc'''
    def __init__(self, name):
        self.__name = name

    def getname(self):
        return self.__name

inspect.getmembers(Foo, inspect.ismethod)