判斷繼承關(guān)系

通過內(nèi)建方法 isinstance(object, classinfo) 可以判斷一個(gè)對象是否是某個(gè)類的實(shí)例。這個(gè)關(guān)系可以是直接，間接或抽象。

實(shí)例的檢查是允許重載的，可見文檔customizing-instance-and-subclass-checks 。根據(jù) PEP 3119 的描述：

The primary mechanism proposed here is to allow overloading the built-in functions isinstance() and issubclass(). The overloading works as follows: The call isinstance(x, C) first checks whether C.__instancecheck__ exists, and if so, calls C.__instancecheck__(x) instead of its normal implementation.

這段話的意思是，當(dāng)調(diào)用 isinstance(x, C) 進(jìn)行檢測時(shí)，會(huì)優(yōu)先檢查是否存在 C.__instancecheck__ ，如果存在則調(diào)用 C.__instancecheck__(x) ，返回的結(jié)果便是實(shí)例檢測的結(jié)果，默認(rèn)的判斷方式就沒有了。

這種方式有助于我們來檢查鴨子類型，我用代碼測了一下。

class Sizeable(object):
  def __instancecheck__(cls, instance):
    print("__instancecheck__ call")
    return hasattr(instance, "__len__")

class B(object):
  pass

b = B()
print(isinstance(b, Sizeable)) # output:False

只打印了 False，并且 __instancecheck__ 沒有調(diào)用。 這是怎么回事。

沒有運(yùn)行的 __instancecheck__

可見文檔寫得并不清楚，為了找出問題，我們從 isinstance 源碼開始進(jìn)行跟蹤。

[abstract.c]
int
PyObject_IsInstance(PyObject *inst, PyObject *cls)
{
  _Py_IDENTIFIER(__instancecheck__);
  PyObject *checker;

  /* Quick test for an exact match */
  if (Py_TYPE(inst) == (PyTypeObject *)cls)
    return 1;
  ....
}

Py_TYPE(inst) == (PyTypeObject *)cls 這是一種快速匹配的方式，等價(jià)于 type(inst) is cls ，這種快速的方式匹配成功的話，也不會(huì)去檢查 __instancecheck__ 。所以文檔中的優(yōu)先檢查是否存在 C.__instancecheck__ 有誤。繼續(xù)向下看源碼：

  /* We know what type's __instancecheck__ does. */
  if (PyType_CheckExact(cls)) {
    return recursive_isinstance(inst, cls);
  }

展開宏 PyType_CheckExact :

[object.h]
#define PyType_CheckExact(op) (Py_TYPE(op) == &PyType_Type)

也就是說 cls 是由 type 直接構(gòu)造出來的類，則判斷語言成立。除了類聲明里指定 metaclass 外基本都是由 type 直接構(gòu)造的。從測試代碼中得知判斷成立，進(jìn)入 recursive_isinstance。但是這個(gè)函數(shù)里面我卻沒找到有關(guān) __instancecheck__ 的代碼，recursive_isinstance 的判斷邏輯大致是:

def recursive_isinstance(inst, cls):
  return pyType_IsSubtype(inst, cls)

def pyType_IsSubtype(a, b):
  for mro in a.__mro__:
    if mro is b:
      return True
  return False

是從 __mro__ 繼承順序來判斷的?；氐?PyObject_IsInstance 函數(shù)往下看：

if (PyTuple_Check(cls)) {
  ...
}

這是當(dāng) instance(x, C) 第二個(gè)參數(shù)是元組的情況，里面的處理方式是遞歸調(diào)用 PyObject_IsInstance(inst, item) 。繼續(xù)往下看：

checker = _PyObject_LookupSpecial(cls, &PyId___instancecheck__);
if (checker != NULL) {
  res = PyObject_CallFunctionObjArgs(checker, inst, NULL);
  ok = PyObject_IsTrue(res);
  return ok;
}

顯然，這邊才是獲得 __instancecheck__ 的地方，為了讓檢查流程走到這里，定義的類要指明 metaclass 。剩下就是跟蹤下 _PyObject_LookupSpecial 就可以了：

[typeobject.c]
PyObject *
_PyObject_LookupSpecial(PyObject *self, _Py_Identifier *attrid)
{
  PyObject *res;

  res = _PyType_LookupId(Py_TYPE(self), attrid);
  // 有回調(diào)的話處理回調(diào)
  // ...
  return res;
}

取的是 Py_TYPE(self) ，也就是說指定的 metaclass 里面需要定義 __instancecheck__ 。

總結(jié)

至此，總結(jié)一下要重載 isinstance(x, C) 行為的條件：

1. x 對象不能是由 C 直接實(shí)例化；
2. C 類指定 metaclass ；
3. 指定的 metaclass 類中定義了 __instancecheck__ 。

測試代碼：

class MetaSizeable(type):
  def __instancecheck__(cls, instance):
    print("__instancecheck__ call")
    return hasattr(instance, "__len__")

class Sizeable(metaclass=MetaSizeable):
  pass

class B(object):
  pass

b = B()
print(isinstance(b, Sizeable)) # output: False
print(isinstance([], Sizeable)) # output: True

文檔可能有點(diǎn)老舊了。本次測試的環(huán)境是Python3.6。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評論