Python：扩展预定义的命名元组

Question

我有以下命名元组：

from collections import namedtuple
ReadElement = namedtuple('ReadElement', 'address value')

然后我想要以下内容：

LookupElement = namedtuple('LookupElement', 'address value lookups')

两个名称元组之间有重复，我如何子类ReadElement以包含一个附加字段？

class LookupElement(ReadElement):
    def __new__(self, address, value, lookups):
        self = super(LookupElement, self).__new__(address, value)
        l = list(self)
        l.append(lookups)
        return tuple(l)

然而元组是创建在那里的，那么在新语句中，如果我修改自我成为一个列表，我会失去类型信息，我该如何避免这种情况？

Answer 1

很容易将一些东西组合在一起，允许您从其他名称元组组成名称元组以及引入新字段。

def extended_namedtuple(name, source_fields):
    assert isinstance(source_fields, list)
    new_type_fields = []
    for f in source_fields:
        try:
            new_type_fields.extend(f._fields)
        except:
            new_type_fields.append(f) 
    return namedtuple(name, new_type_fields) 

# source types
Name = namedtuple('Name', ['first_name', 'last_name'])
Address = namedtuple('Address', ['address_line1', 'city'])
# new type uses source types and adds additional ID field
Customer = extended_namedtuple('Customer', ['ID', Name, Address])
# using the new type
cust1 = Customer(1, 'Banana', 'Man', '29 Acacia Road', 'Nuttytown')
print(cust1)

这将输出以下内容：

Customer(ID=1, first_name='Banana', last_name='Man', address_line1='29 Acacia Road', city='Nuttytown')

Answer 2

扩展Martijn Pieters的答案：有一种方法可以使新的namedtuple类成为另一个类的子类，而无需破解。只需单独创建新的namedtuple，然后使用其__new__方法而不是使用Super：

from collections import namedtuple

class ReadElement(namedtuple('ReadElement', ('address', 'value'))):
    def compute(self):
        return self.value + 1

_LookupElement = namedtuple('_LookupElement', ReadElement._fields + ('lookups',))

class LookupElement(_LookupElement, ReadElement):
    def __new__(self, address, value, lookups):
        return _LookupElement.__new__(LookupElement, address, value, lookups)

assert issubclass(LookupElement, ReadElement)
l = LookupElement('ad', 1, dict())
assert isinstance(l, ReadElement)
assert l.compute() == 2

这似乎也可以在没有覆盖__new__的情况下工作！

from collections import namedtuple

class ReadElement(namedtuple('ReadElement', ('address', 'value'))):
    def compute(self):
        return self.value + 1

class LookupElement(namedtuple('LookupElement', ReadElement._fields + ('lookups',)),
                    ReadElement):
    """nothing special to do"""
    pass

Answer 3

您可以子类化< code>namedtuple生成的类，但是您需要更仔细地研究生成的类。您需要添加另一个具有额外字段的< code>__slots__属性，更新< code>_fields属性，创建新的< code>__repr__和< code>_replace方法(它们硬编码字段列表和类名)，并为额外的字段添加额外的< code>property对象。请参见文档中的示例。

这一切都有点太多了。而不是子类，我只是重用源类型的somenamedtuple._fields属性：

LookupElement = namedtuple('LookupElement', ReadElement._fields + ('lookups',))

< code>namedtuple()构造函数的< code>field_names参数不一定是字符串，也可以是字符串序列。只需获取< code>_fields并通过连接一个新元组来添加更多元素。

演示:

>>> from collections import namedtuple
>>> ReadElement = namedtuple('ReadElement', 'address value')
>>> LookupElement = namedtuple('LookupElement', ReadElement._fields + ('lookups',))
>>> LookupElement._fields
('address', 'value', 'lookups')
>>> LookupElement('addr', 'val', 'lookup') 
LookupElement(address='addr', value='val', lookups='lookup')

这意味着扩展类型不是基类型的子类。如果您必须有一个类层次结构，那么与其尝试使命名元组适合该模型，不如改为使用数据类。在大多数使用元组的用例中，数据类可以起到相同的作用，但可以很容易地进行子类化。