如何在Python中为str子代覆盖ord行为？

Question

我有这样的课程：

class STR(str):

    def __int__(self):
        return 42 

如果我像这样在提示中使用它：

>>> a=STR('8')
>>> ord(a)
56
>>> int(a)
42
>>> chr(a)
'*'

那就是行为。我想将 ord(a) 设为 42。我该怎么做？我应该在 str 类中重写哪个方法？所有这些都记录在任何地方吗？

谢谢！

Answer 1

以下是来自 bltinmodule.c 的当前 SVN 版本：

static PyObject *
builtin_ord(PyObject *self, PyObject* obj)
{
    long ord;
    Py_ssize_t size;

    if (PyString_Check(obj)) {
        size = PyString_GET_SIZE(obj);
        if (size == 1) {
            ord = (long)((unsigned char)*PyString_AS_STRING(obj));
            return PyInt_FromLong(ord);
        }
    } else if (PyByteArray_Check(obj)) {
        size = PyByteArray_GET_SIZE(obj);
        if (size == 1) {
            ord = (long)((unsigned char)*PyByteArray_AS_STRING(obj));
            return PyInt_FromLong(ord);
        }

#ifdef Py_USING_UNICODE
    } else if (PyUnicode_Check(obj)) {
        size = PyUnicode_GET_SIZE(obj);
        if (size == 1) {
            ord = (long)*PyUnicode_AS_UNICODE(obj);
            return PyInt_FromLong(ord);
        }
#endif
    } else {
        PyErr_Format(PyExc_TypeError,
                 "ord() expected string of length 1, but " \
                 "%.200s found", obj->ob_type->tp_name);
        return NULL;
    }

    PyErr_Format(PyExc_TypeError,
             "ord() expected a character, "
             "but string of length %zd found",
             size);
    return NULL;
}

如您所见，它不会对您传入的实例进行任何方法调用。据我所知，如果您不要向其传递显式字符串，因为无法覆盖 ord 的功能。

事实上，它本质上是验证 PyObject 是字符串、字节数组还是 Unicode —— 这就是 PyString_Check 等函数的作用。如果不是这些，您将收到 TypeError 异常。

一种扩展性不太好的解决方法是在全局命名空间中编写您自己的 ord：

>>> class STR(str):
...     def __int__(self):
...             return 42
... 
>>> 
>>> def ord(s):
...     if isinstance(s, STR):
...             return int(s)
...     else:
...             return __builtins__.ord(s)
... 
>>>  
>>> ord(STR('fdsafds'))
42
>>> ord("!")
33

当然，这种扩展性非常糟糕，因为使用您的类的另一个模块可能会直接调用 __builtins__.ord，或者它们甚至可能会覆盖 ord 本身！但是，如果您只需要在单个独立模块上工作，这是一种方法。

Answer 2

有人已经发布了内置的 ord 代码，并且没有您可以拦截的方法调用。

一种解决方案可以覆盖 ord 函数，例如：

backup_ord = ord
def ord(obj):
    if hasattr(obj, '__ord__'):
        return obj.__ord__()
    else:
        return backup_ord(obj)

然后使用 __ord__ 方法定义类并执行以下操作：

class MyStr(str):
    def __ord__(self):
        return 'LOL'

对于测试：

normal_five = '5'
strange_five = MyStr('5')
print ord(normal_five)
print ord(strange_five)

输出：

53
LOL

Answer 3

ord 不可重载，它不会调用字符串对象上的任何方法（至少在 CPython 中 - Python/bltinmodule.c、builtin_ord 函数）。

最好的方法可能是创建 STR.ord 方法并调用它而不是内置方法。

Answer 4

您不能覆盖 ord，但您可以覆盖为变量，例如

备份原始 ord

origord= ord

def ord(x):
    return 42


ord(a)
42

origord(a)
56

但我认为覆盖它不是一个好主意。