def program_from_multi_pubkey(m: int, pub_keys: []) -> bytes:
n = len(pub_keys)
if m <= 0:
raise SDKException(ErrorCode.other_error(f'Param error: m == {m}'))
if m > n:
raise SDKException(
ErrorCode.other_error(f'Param error: m == {m} n == {n}'))
if n > MULTI_SIG_MAX_PUBKEY_SIZE:
raise SDKException(
ErrorCode.other_error(
f'Param error: n == {n} > {MULTI_SIG_MAX_PUBKEY_SIZE}'))
builder = ParamsBuilder()
builder.emit_push_int(m)
pub_keys = ProgramBuilder.sort_public_keys(pub_keys)
for pk in pub_keys:
builder.emit_push_bytearray(pk)
builder.emit_push_int(n)
builder.emit(CHECKMULTISIG)
return builder.to_bytes()
def push_param(param, builder: ParamsBuilder):
if isinstance(param, bytearray) or isinstance(param, bytes):
builder.emit_push_bytearray(param)
elif isinstance(param, str):
builder.emit_push_bytearray(bytes(param.encode()))
elif isinstance(param, bool):
builder.emit_push_bool(param)
elif isinstance(param, int):
builder.emit_push_int(param)
elif isinstance(param, dict):
# builder.emit_push_byte_array(BuildParams.get_map_bytes(dict(param)))
BuildParams.push_map(param, builder)
elif isinstance(param, list):
BuildParams.create_code_params_script_builder(param, builder)
builder.emit_push_int(len(param))
builder.emit(PACK)
elif isinstance(param, Struct):
BuildParams.push_struct(param, builder)
else:
raise SDKException(
ErrorCode.other_error('parameter type is error'))
def build_native_invoke_code(contract_address: bytes, version: bytes, method: str, params):
builder = ParamsBuilder()
build_neo_vm_param(builder, params)
builder.emit_push_bytearray(method.encode())
builder.emit_push_bytearray(contract_address)
builder.emit_push_int(int.from_bytes(version, 'little'))
builder.emit(SYSCALL)
builder.emit_push_bytearray(b'Ontology.Native.Invoke')
return builder.to_bytes()
def get_map_bytes(param_dict: dict):
builder = ParamsBuilder()
builder.emit(BuildParams.Type.dict_type.value)
builder.emit(utils.big_int_to_neo_bytearray(len(param_dict)))
for key, value in param_dict.items():
builder.emit(BuildParams.Type.bytearray_type.value)
builder.emit_push_bytearray(str(key).encode())
if isinstance(value, bytearray) or isinstance(value, bytes):
builder.emit(BuildParams.Type.bytearray_type.value)
builder.emit_push_bytearray(bytearray(value))
elif isinstance(value, str):
builder.emit(BuildParams.Type.bytearray_type.value)
builder.emit_push_bytearray(value.encode())
elif isinstance(value, bool):
builder.emit(BuildParams.Type.bool_type.value)
builder.emit_push_bool(value)
elif isinstance(value, int):
builder.emit(BuildParams.Type.int_type.value)
builder.emit_push_int(int(value))
else:
raise Exception("param error")
return builder.to_bytes()
def from_public_key(cls, public_key: bytes):
builder = ParamsBuilder()
builder.emit_push_bytearray(bytearray(public_key))
builder.emit(CHECKSIG)
return cls.__from_byte_script(builder.to_bytes())
def address_from_bytes_pubkey(public_key: bytes):
builder = ParamsBuilder()
builder.emit_push_bytearray(bytearray(public_key))
builder.emit(CHECKSIG)
addr = Address(Address.to_script_hash(builder.to_bytes()))
return addr
def program_from_pubkey(public_key):
builder = ParamsBuilder()
builder.emit_push_bytearray(public_key)
builder.emit(CHECKSIG)
return builder.to_bytes()
