57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
from io import BytesIO
|
|
from typing import Union
|
|
|
|
from boilerplate_client.utils import (read, read_uint, read_varint,
|
|
write_varint, UINT64_MAX)
|
|
|
|
|
|
class TransactionError(Exception):
|
|
pass
|
|
|
|
|
|
class Transaction:
|
|
def __init__(self, txType: int, nonce: int, gasPrice: int, gasLimit: int, to: Union[str, bytes], value: int, memo: Union[str, bytes]) -> None:
|
|
self.txType: int = txType
|
|
self.nonce: int = nonce
|
|
self.gasPrice: int = gasPrice
|
|
self.gasLimit: int = gasLimit
|
|
self.to: bytes = bytes.fromhex(to[2:]) if isinstance(to, str) else to
|
|
self.value: int = value
|
|
self.memo: bytes = bytes.fromhex(memo[2:]) if isinstance(memo, str) else memo
|
|
|
|
if not (0 <= self.nonce <= UINT64_MAX):
|
|
raise TransactionError(f"Bad nonce: '{self.nonce}'!")
|
|
|
|
if not (0 <= self.value <= UINT64_MAX):
|
|
raise TransactionError(f"Bad value: '{self.value}'!")
|
|
|
|
if len(self.to) != 20:
|
|
raise TransactionError(f"Bad address: '{self.to}'!")
|
|
|
|
self.lenGP = int((len(hex(self.gasPrice)) - 1) / 2)
|
|
self.lenGL = int((len(hex(self.gasLimit)) - 1) / 2)
|
|
self.lenValue = int((len(hex(self.value)) - 1) / 2)
|
|
|
|
|
|
def serialize(self) -> bytes:
|
|
return b"".join([
|
|
self.txType.to_bytes(1, byteorder="big"),
|
|
|
|
self.nonce.to_bytes(1, byteorder="big"),
|
|
|
|
write_varint(self.lenGP + 0x80),
|
|
self.gasPrice.to_bytes(self.lenGP, byteorder="big"),
|
|
|
|
write_varint(self.lenGL + 0x80),
|
|
self.gasLimit.to_bytes(self.lenGL, byteorder="big"),
|
|
|
|
write_varint(len(self.to) + 0x80),
|
|
self.to,
|
|
|
|
write_varint(self.lenValue + 0x80),
|
|
self.value.to_bytes(self.lenValue, byteorder="big"),
|
|
|
|
self.memo,
|
|
|
|
])
|