Fix style error

M1nd3r/thp5
M1nd3r 2 months ago
parent 15cf9da533
commit 2bbe55f063

@ -197,11 +197,11 @@ class ChannelContext(Context):
message = message_handler.wrap_protobuf_load(buf, expected_type) message = message_handler.wrap_protobuf_load(buf, expected_type)
print(message) print(message)
# ------------------------------------------------TYPE ERROR------------------------------------------------ # ------------------------------------------------TYPE ERROR------------------------------------------------
session_message: ThpCreateNewSession = message assert isinstance(message, ThpCreateNewSession)
print("passphrase:", session_message.passphrase) print("passphrase:", message.passphrase)
# await thp_messages.handle_CreateNewSession(message) # await thp_messages.handle_CreateNewSession(message)
if session_message.passphrase is not None: if message.passphrase is not None:
self.create_new_session(session_message.passphrase) self.create_new_session(message.passphrase)
else: else:
self.create_new_session() self.create_new_session()
except Exception as e: except Exception as e:

@ -2,7 +2,7 @@ import ustruct # pyright:ignore[reportMissingModuleSource]
from storage.cache_thp import BROADCAST_CHANNEL_ID from storage.cache_thp import BROADCAST_CHANNEL_ID
from trezor import protobuf from trezor import protobuf
from trezor.messages import CreateNewSession from trezor.messages import ThpCreateNewSession
from .. import message_handler from .. import message_handler
from ..protocol_common import Message from ..protocol_common import Message
@ -90,7 +90,7 @@ def decode_message(buffer: bytes, msg_type: int) -> protobuf.MessageType:
return x return x
async def handle_CreateNewSession(msg: CreateNewSession) -> None: async def handle_CreateNewSession(msg: ThpCreateNewSession) -> None:
print(msg.passphrase) print(msg.passphrase)
print(msg.on_device) print(msg.on_device)
pass pass

@ -183,7 +183,7 @@ class Field:
class _MessageTypeMeta(type): class _MessageTypeMeta(type):
def __init__(cls, name: str, bases: tuple, d: dict) -> None: def __init__(cls, name: str, bases: tuple, d: dict) -> None:
super().__init__(name, bases, d) super().__init__(name, bases, d) # type: ignore [Expected 1 positional argument]
if name != "MessageType": if name != "MessageType":
cls.__init__ = MessageType.__init__ # type: ignore ["__init__" is obscured by a declaration of the same name;;Cannot assign member "__init__" for type "_MessageTypeMeta"] cls.__init__ = MessageType.__init__ # type: ignore ["__init__" is obscured by a declaration of the same name;;Cannot assign member "__init__" for type "_MessageTypeMeta"]

Loading…
Cancel
Save