如何将特殊方法'Mixin'应用于typing.NamedTuple

Question

如何将特殊方法'Mixin'应用于typing.NamedTuple

11

我喜欢 Python 3.6 中的 typing.NamedTuple。但通常情况下，namedtuple 包含一个不可哈希的属性，并且我想将其用作 dict 键或 set 成员。如果 namedtuple 类使用对象标识(id() 作为 __eq__ 和 __hash__) 是有意义的，那么将这些方法添加到类中即可。

然而，现在我的代码中出现了这种模式，我想摆脱繁琐的 __eq__ 和 __hash__ 方法定义。我知道 namedtuple 并不是常规类，我一直无法搞清楚如何解决这个问题。

以下是我尝试过的方法：

from typing import NamedTuple

class ObjectIdentityMixin:
    def __eq__(self, other):
        return self is other

    def __hash__(self):
        return id(self)

class TestMixinFirst(ObjectIdentityMixin, NamedTuple):
    a: int

print(TestMixinFirst(1) == TestMixinFirst(1))  # Prints True, so not using my __eq__

class TestMixinSecond(NamedTuple, ObjectIdentityMixin):
    b: int

print(TestMixinSecond(2) == TestMixinSecond(2))  # Prints True as well

class ObjectIdentityNamedTuple(NamedTuple):
    def __eq__(self, other):
        return self is other

    def __hash__(self):
        return id(self)

class TestSuperclass(ObjectIdentityNamedTuple):
    c: int

TestSuperclass(3)    
"""
Traceback (most recent call last):
  File "test.py", line 30, in <module>
    TestSuperclass(3)
TypeError: __new__() takes 1 positional argument but 2 were given
"""

有没有一种方法，可以让我不必在每个需要“对象标识”（object identity）的NamedTuple中重复这些方法？

- Damon Maria

你的类ObjectIdentityNamedTupl中的__init__()方法是如何实现的？ - developer_hatch

@DamianLattenero 我已经编辑了我的问题，展示了一个完整的代码示例。没有__init__。我尝试添加__new__，但是NamedTuple不允许这样做。 - Damon Maria

1

看起来 Python 3.7 中的 PEP 557 可能会解决所有这些问题。 - Damon Maria

2

你可以在Python 3.7中使用Dataclasses。它的行为类似于NamedTuple，并允许你重写__eq__。只需用dataclasses.dataclass装饰类即可。 - pylang

Dataclasses的文档参考链接：https://docs.python.org/3/library/dataclasses.html https://www.python.org/dev/peps/pep-0557/ - Lekensteyn

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- georgexsh · Accepted Answer

NamedTuple类语法的魔法源泉是其元类metaclassNamedTupleMeta，behind the scene，NamedTupleMeta.__new__为您创建一个新类，而不是典型的类，但是是由collections.namedtuple()创建的类。

问题在于，当NamedTupleMeta创建新类对象时，它会忽略基类，您可以检查TestMixinFirst的MRO，其中没有ObjectIdentityMixin。

>>> print(TestMixinFirst.mro())
[<class '__main__.TestMixinFirst'>, <class 'tuple'>, <class 'object'>]

您需要扩展NamedTupleMeta以处理基类：

import typing


class NamedTupleMetaEx(typing.NamedTupleMeta):

    def __new__(cls, typename, bases, ns):
        cls_obj = super().__new__(cls, typename+'_nm_base', bases, ns)
        bases = bases + (cls_obj,)
        return type(typename, bases, {})


class TestMixin(ObjectIdentityMixin, metaclass=NamedTupleMetaEx):
    a: int
    b: int = 10


t1 = TestMixin(1, 2)
t2 = TestMixin(1, 2)
t3 = TestMixin(1)

assert hash(t1) != hash(t2)
assert not (t1 == t2)
assert t3.b == 10