diff --git a/twnet_parser/packet.py b/twnet_parser/packet.py index 3654945..2b0f4b5 100644 --- a/twnet_parser/packet.py +++ b/twnet_parser/packet.py @@ -45,6 +45,21 @@ class PacketFlags7(PrettyPrint): self.compression: Optional[bool] = None self.connless: Optional[bool] = None + def __iter__(self): + flags = [] + if self.control: + flags.append('control') + if self.resend: + flags.append('resend') + if self.compression: + flags.append('compression') + if self.connless: + flags.append('connless') + return iter(flags) + + def __repr__(self): + return ": " + str(list(self)) + class PacketFlags6(PrettyPrint): def __init__(self) -> None: self.token: Optional[bool] = None @@ -53,6 +68,23 @@ class PacketFlags6(PrettyPrint): self.compression: Optional[bool] = None self.connless: Optional[bool] = None + def __iter__(self): + flags = [] + if self.token: + flags.append('token') + if self.control: + flags.append('control') + if self.resend: + flags.append('resend') + if self.compression: + flags.append('compression') + if self.connless: + flags.append('connless') + return iter(flags) + + def __repr__(self): + return ": " + str(list(self)) + class PacketHeader6(PrettyPrint): def __init__( self,