1
0
mirror of synced 2025-02-03 13:13:26 +01:00

classes.py: Simplify getattr calls

This commit is contained in:
Viv 2023-07-31 08:52:52 -04:00
parent cd519f5a26
commit d7a817b9ea

View File

@ -339,13 +339,14 @@ class FumenHeader:
value_list = [] value_list = []
format_string = self.order format_string = self.order
for byte_field in fields(self): for byte_field in fields(self):
value = getattr(self, byte_field.name)
if byte_field.name == "order": if byte_field.name == "order":
pass pass
elif byte_field.name == "b000_b431_timing_windows": elif byte_field.name == "b000_b431_timing_windows":
value_list.extend(list(getattr(self, byte_field.name))) value_list.extend(list(value))
format_string += "f" * len(getattr(self, byte_field.name)) format_string += "f" * len(value)
else: else:
value_list.append(getattr(self, byte_field.name)) value_list.append(value)
format_string += "i" format_string += "i"
raw_bytes = struct.pack(format_string, *value_list) raw_bytes = struct.pack(format_string, *value_list)
assert len(raw_bytes) == 520 assert len(raw_bytes) == 520