teeworlds-network/twnet_parser
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

feat: implement list cast on packet flags

Browse source
This commit is contained in:
ChillerDragon 2024-06-16 10:12:46 +08:00
parent 76b0a89375
commit 22f7b437bd
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
twnet_parser/packet.py
View file

@ -45,6 +45,21 @@ class PacketFlags7(PrettyPrint):
self.compression: Optional[bool] = None self.compression: Optional[bool] = None
self.connless: 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 "<class: '" + str(self.__class__.__name__) + "'>: " + str(list(self))
class PacketFlags6(PrettyPrint): class PacketFlags6(PrettyPrint):
def __init__(self) -> None: def __init__(self) -> None:
self.token: Optional[bool] = None self.token: Optional[bool] = None
@ -53,6 +68,23 @@ class PacketFlags6(PrettyPrint):
self.compression: Optional[bool] = None self.compression: Optional[bool] = None
self.connless: 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 "<class: '" + str(self.__class__.__name__) + "'>: " + str(list(self))
class PacketHeader6(PrettyPrint): class PacketHeader6(PrettyPrint):
def __init__( def __init__(
self, self,