arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-26 14:22:06 +00:00
Code Issues Releases Wiki Activity

fix: style pb2py

Browse Source 
[no changelog]
This commit is contained in:
M1nd3r 2025-02-25 13:43:44 +01:00
parent 9813e63104
commit efe9282167
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
common/protob/pb2py
View File

@ -240,7 +240,9 @@ class ProtoMessage:
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)
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)