4

我正在编写一个新的扩展类型，但在设置数字运算（例如加法/减法/乘法）时遇到问题。我已经设法设置了一些就地操作，而没有调用正常操作。

例如，我有以下功能：

static PyObject *
MyType_Mul(PyObject *v, PyObject *w)
{
    PyErr_SetString(PyExc_ValueError, "testing");
    return NULL;
}

我在这样的数字方法中设置它：

static PyNumberMethods my_type_as_number = {
    0,  /* nb_add */
    0,  /* nb_sub */
    (binaryfunc)MyType_Mul,  /* nb_mul */
    ...
    0,  /* nb_in_place_add */
    0,  /* nb_in_place_sub */
    (binaryfunc)MyType_Mul,  /* nb_in_place_mul */
    ...
};

现在，当我尝试使用我的类型时，我得到了这种行为：

>>> from mytype import MyType
>>> a = MyType()
>>> a * 2
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for *: 'mytype.MyType' and 'int'
>>> 2 * a
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for *: 'int' and 'mytype.MyType'

但是，如果我使用就地运算符：

>>> a *= 2
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: testing

如果我dir()在对象上使用，我可以看到__mul__and__rmul__方法（这意味着 python 看到它们），但似乎根本没有调用它们。使用a.__mul__(2)退货NotImplemented。

还：

>>> a.__mul__
<method-wrapper '__mul__' of mytype.MyType object at 0x7fc2ecc50468>
>>> a.__imul__
<method-wrapper '__imul__' of mytype.MyType object at 0x7fc2ecc50468>

所以，正如你所看到的，它们是完全一样的。

这是怎么回事？为什么相同的功能适用于就地运算符而不适用于“正常”运算符？我也认为我可能使用了错误的插槽，但我仔细检查了它是正确的，并且将它设置为nb_add等nb_sub也不起作用。

4

1 回答 1

3

感谢 nneonneo 的评论，我明白出了什么问题。基本上我忘了设置Py_TPFLAGS_CHECKTYPES标志。

在我给出的描述中，有一些关于这种缺失的线索：

  1. 这些方法是同一个对象，但它对就地操作的作用不同
  2. 在评论中，我还说过，例如，做a*a会产生正确的结果，而做a*different-type不会。

这显然意味着解释器在执行非就地操作时正在检查参数的类型，如果类型为 ，则调用我的函数MyType，否则返回NotImplemented。

在文档中搜索一下很容易看出这是数字方法的默认行为。

如果参数类型不属于同一类，则假定操作未实现。

要允许不同类型一起“操作”，您必须在以下位置设置Py_TPFLAGS_CHECKTYPES标志MyType：

static PyTypeObject MyType = {
    PyObject_HEAD_INIT(&PyType_Type)
    0,                         /*ob_size*/
    "mytype.MyType",           /*tp_name*/
    sizeof(MyTypeObject),      /*tp_basicsize*/
    0,                         /*tp_itemsize*/
    ...
    0,                         /*tp_repr*/
    &mytype_as_number,         /*tp_as_number*/
    0,                         /*tp_as_sequence*/
    ...
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES,/*tp_flags*/
    ...
};

设置此标志后，解释器将不会检查类型，因此您必须手动处理它们。

相反，就地运算符总是允许不同的类型。为什么会这样，我不知道。

于 2012-09-16T18:09:40.410 回答