arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-17 11:58:13 +00:00
Code Issues Releases Wiki Activity

add loads, dumps to MessageType and Message

Browse Source
This commit is contained in:
Jan Pochyla 2016-05-19 16:47:35 +02:00 committed by Pavol Rusnak
parent 506f2e0c36
commit f5b802d421
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 24 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/lib/protobuf/protobuf.py
View File

@ -185,6 +185,15 @@ class MessageType:
raise ValueError('The field %s (\'%s\') is required but missing.' % (tag, name)) raise ValueError('The field %s (\'%s\') is required but missing.' % (tag, name))
return message return message
def dumps(self, value):
fp = BytesIO()
self.dump(fp, value)
return fp.getvalue()
def loads(self, buf):
fp = BytesIO(buf)
return self.load(fp)
class Message: class Message:
# Represents a message instance. # Represents a message instance.
@ -196,6 +205,10 @@ class Message:
# Dumps the message into a write-like object. # Dumps the message into a write-like object.
return self.message_type.dump(fp, self) return self.message_type.dump(fp, self)
def dumps(self):
# Dumps the message into bytes
return self.message_type.dumps(self)
# Embedded message. ------------------------------------------------------------ # Embedded message. ------------------------------------------------------------
class EmbeddedMessage: class EmbeddedMessage: