From d7a817b9ea5593a96ab33a295758df5eac6f836f Mon Sep 17 00:00:00 2001 From: Viv Date: Mon, 31 Jul 2023 08:52:52 -0400 Subject: [PATCH] `classes.py`: Simplify `getattr` calls --- src/tja2fumen/classes.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/tja2fumen/classes.py b/src/tja2fumen/classes.py index 787a124..0556256 100644 --- a/src/tja2fumen/classes.py +++ b/src/tja2fumen/classes.py @@ -339,13 +339,14 @@ class FumenHeader: value_list = [] format_string = self.order for byte_field in fields(self): + value = getattr(self, byte_field.name) if byte_field.name == "order": pass elif byte_field.name == "b000_b431_timing_windows": - value_list.extend(list(getattr(self, byte_field.name))) - format_string += "f" * len(getattr(self, byte_field.name)) + value_list.extend(list(value)) + format_string += "f" * len(value) else: - value_list.append(getattr(self, byte_field.name)) + value_list.append(value) format_string += "i" raw_bytes = struct.pack(format_string, *value_list) assert len(raw_bytes) == 520