自定义类型.NamedTuple

Question

我正在使用NamedTuples来保存数据，并且我想添加一个可以被多个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'>}