From efe92821671e637b881a73ac52188f340d1e8f08 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Tue, 25 Feb 2025 13:43:44 +0100 Subject: [PATCH] fix: style pb2py [no changelog] --- common/protob/pb2py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/common/protob/pb2py b/common/protob/pb2py index 7382255326..df17855ae5 100755 --- a/common/protob/pb2py +++ b/common/protob/pb2py @@ -239,8 +239,10 @@ class ProtoMessage: def from_message(cls, descriptor: "Descriptor", message): message_type = find_by_name(descriptor.message_type_enum.value, message.name) thp_message_type = None - if not isinstance(descriptor.thp_message_type_enum,tuple): - thp_message_type = find_by_name(descriptor.thp_message_type_enum.value, message.name) + if not isinstance(descriptor.thp_message_type_enum, tuple): + thp_message_type = find_by_name( + descriptor.thp_message_type_enum.value, message.name + ) # use extensions set on the message_type entry (if any) extensions = descriptor.get_extensions(message_type) # override with extensions set on the message itself @@ -359,7 +361,9 @@ class Descriptor: top_level_enums = itertools.chain.from_iterable(f.enum_type for f in self.files) self.message_type_enum = find_by_name(top_level_enums, MESSAGE_TYPE_ENUM, ()) top_level_enums = itertools.chain.from_iterable(f.enum_type for f in self.files) - self.thp_message_type_enum = find_by_name(top_level_enums, THP_MESSAGE_TYPE_ENUM, ()) + self.thp_message_type_enum = find_by_name( + top_level_enums, THP_MESSAGE_TYPE_ENUM, () + ) self.convert_enum_value_names(self.message_type_enum) self.convert_enum_value_names(self.thp_message_type_enum) @@ -430,7 +434,7 @@ class Descriptor: self._nested_types_from_message(nested.orig) def convert_enum_value_names(self, enum): - if isinstance(enum,tuple): + if isinstance(enum, tuple): return for value in enum.value: value.name = strip_enum_prefix(enum.name, value.name)