Python標準庫inspect

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

getargspec(func)函數

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

在用__init__參數自動初始化實例屬性的實踐中,是用字節碼對象的co_varnames屬性來獲取函數的位置參數名的:code

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,其它位保留:blog

>>> 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__的**參數了:get

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'}

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

getmembers(object[, predicate])io

返回一個包含對象的全部成員的(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)
相關文章
相關標籤/搜索