Настройка набора текста.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'>}