问题描述
我有一个使用struct解码二进制数据并将其存储在NamedTuple中的类,如下所示:
class HEADER1(NamedTuple):
name: str
u2: int
tracetime: int
u4: int
u5: int
u6: int
u7: int
struct = Struct('<8s6L')
@classmethod
def unpack(cls,data):
return cls(*cls.struct.unpack(data))
这可以正常工作,我可以按以下方式使用:
h = HEADER1.unpack(b'svpt_str\x03\x01\x00\x00\xae\xa6i_\xd0\x03\xfe3\x00\x00\x00\x00P\xa0\xdc3\x00\x00\x00\x00')
但是,如果我尝试将其更改为继承类方法,则失败:
class NamedTupleUnpack(NamedTuple):
struct = Struct('x')
@classmethod
def unpack(cls,data):
return cls(*cls.struct.unpack(data))
class HEADER1(NamedTupleUnpack):
name: str
u2: int
tracetime: int
u4: int
u5: int
u6: int
u7: int
struct = Struct('<8s6L')
然后错误TypeError: __new__() takes 1 positional argument but 8 were given
。
我了解从NamedTuple继承有问题,但想知道是否可以解决吗?
编辑:正如下面其他人所暗示的,数据类似乎是解决之道:A way to subclass NamedTuple for purposes of typechecking
解决方法
typing.NamedTuple
没有提供所需的功能,因为将字段添加到namedtuple类的子类中会与namedtuple的设计意图相冲突。
设计意图是,如果Foo
是具有n
字段的namedtuple类,则Foo
的实例应该是和行为类似于n
元素元组。例如,如果为n==3
,那么您应该能够采用Foo
的任意实例并执行
a,b,c = foo
添加字段可解决此问题。如果您可以使用第四个字段创建子类class Bar(Foo)
,则Bar
的实例将是您无法做到的Foo
的实例
a,c = bar
您的NamedTupleUnpack
是具有0个字段的namedtuple类。您不能在HEADER1
子类中添加字段。
您可能应该使用常规类或dataclasses
,或将unpack
放入mixin。
例如,我尝试使用mixin:
class HEADER1(Unpack,NamedTuple):
# Mixins evaluate right to left so NamedTuple is the base
...
但是,NameTuple也不能与mixins一起使用,并且在派生的元组中unpack方法不可用。
因此,我使用了确实有效的数据类(并添加了更多功能):
from typing import NamedTuple
from struct import Struct
from dataclasses import dataclass,fields
from abc import ABCMeta,abstractproperty
from collections.abc import Sequence
class Unpack(Sequence,metaclass=ABCMeta):
@abstractproperty
def struct(self):
pass
@classmethod
def unpack(cls,data):
return cls(*cls.struct.unpack(data))
def __getitem__(self,i):
return getattr(self,fields(self)[i].name)
def __len__(self):
return len(fields(self))
@dataclass(order=True)
class HEADER1(Unpack):
# Note Mixin's evalute right to left so NamedTuple is the base
name: str
u2: int
tracetime: int
u4: int
u5: int
u6: int
u7: int
struct = Struct('<8s6L')
h = HEADER1.unpack(b'svpt_str\x03\x01\x00\x00\xae\xa6i_\xd0\x03\xfe3\x00\x00\x00\x00P\xa0\xdc3\x00\x00\x00\x00')
h
HEADER1(name=b'svpt_str',u2=259,tracetime=1600759470,u4=872285136,u5=0,u6=870096976,u7=0)