diff --git a/bitcoin/messages.py b/bitcoin/messages.py index 59bd55cc..b295e71f 100644 --- a/bitcoin/messages.py +++ b/bitcoin/messages.py @@ -51,11 +51,13 @@ def msg_ser(self, f): def msg_deser(cls, f, protover=PROTO_VERSION): raise NotImplementedError - def to_bytes(self): + def to_bytes(self, params=None): + if params is None: + params = bitcoin.params f = _BytesIO() self.msg_ser(f) body = f.getvalue() - res = bitcoin.params.MESSAGE_START + res = params.MESSAGE_START res += self.command res += b"\x00" * (12 - len(self.command)) res += struct.pack(b"