Question

我使用NamedTuple来保存数据，我想添加一个可以由多个基于NamedTuple的类继承的方法。但是当我尝试使用多继承或子类化基于NamedTuple的类时，它不起作用。具体来说，我试图自动为我的所有数据类提供一个方法，该方法可以查看类注释，然后根据它调用一些序列化代码。以下是我尝试过的一些例子：

from typing import NamedTuple


class Base1:
    def foo(self):
        print(self.__annotations__)


class Test1(NamedTuple, Base1):
    x: int
    y: int


x = Test1(1, 2)
x.foo() # raises AttributeError


class Base2(NamedTuple):
    def foo(self):
        print(self.__annotations__)


class Test2(Base2):
    x: int
    y: int


x = Test2(1, 2) # TypeError: __new__() takes 1 positional argument but 3 were given

我有办法像这样使用NamedTuple类吗？

Answer 1

问题是typing.NamedTuple使用的元类;此元类忽略所有基类，只生成一个带有添加注释信息的collections.namedtuple()类（复制直接在类上定义的任何其他属性）。

您可以定义自己的元类（必须是typing.NamedTupleMeta的子类），在生成命名元组类之后添加额外的基类：

import typing class MultipleInheritanceNamedTupleMeta(typing.NamedTupleMeta): def __new__(mcls, typename, bases, ns): if typing.NamedTuple in bases: base = super().__new__(mcls, '_base_' + typename, bases, ns) bases = (base, *(b for b in bases if not isinstance(b, typing.NamedTuple))) return super(typing.NamedTupleMeta, mcls).__new__(mcls, typename, bases, ns) class Base1(metaclass=MultipleInheritanceNamedTupleMeta): def foo(self): print(self.__annotations__) class Test1(NamedTuple, Base1): x: int y: int

请注意，这不会让您继承字段！那是因为你必须为任何字段组合生成一个新的namedtuple类。以上结果产生以下结构：

Test1，继承自

_base_Test1 - 实际生成的typing.NamedTuple namedtuple

tuple

Base1

，这可以按要求运作：

>>> x = Test1(1, 2) >>> x.foo() {'x': <class 'int'>, 'y': <class 'int'>}

自定义typing.NamedTuple

1 个答案: