def build_initcode(*, contract_bytecode, contract_abi=[], constructor_args=[]):
constructor_abi = get_constructor_abi(contract_abi)
# The initcode is the bytecode with the encoded arguments appended
if constructor_abi:
return encode_abi(
web3=Web3(),
abi=constructor_abi,
arguments=constructor_args,
data=contract_bytecode,
)
else:
return contract_bytecode
def encodeABI(cls, fn_name, args=None, kwargs=None, data=None):
"""
Encodes the arguments using the Ethereum ABI for the contract function
that matches the given name and arguments..
:param data: defaults to function selector
"""
fn_abi, fn_selector, fn_arguments = get_function_info(
fn_name, cls.web3.codec, contract_abi=cls.abi, args=args, kwargs=kwargs,
)
if data is None:
data = fn_selector
return encode_abi(cls.web3, fn_abi, fn_arguments, data)
def encodeABI(cls, fn_name, args=None, kwargs=None, data=None):
"""
Encodes the arguments using the Ethereum ABI for the contract function
that matches the given name and arguments..
:param data: defaults to function selector
"""
fn_abi, fn_selector, fn_arguments = get_function_info(
fn_name, contract_abi=cls.abi, args=args, kwargs=kwargs,
)
if data is None:
data = fn_selector
return encode_abi(cls.web3, fn_abi, fn_arguments, data)
def _encode_data_in_transaction(self, *args, **kwargs):
constructor_abi = get_constructor_abi(self.abi)
if constructor_abi:
if not args:
args = tuple()
if not kwargs:
kwargs = {}
arguments = merge_args_and_kwargs(constructor_abi, args, kwargs)
data = add_0x_prefix(
encode_abi(self.web3, constructor_abi, arguments, data=self.bytecode)
)
else:
data = to_hex(self.bytecode)
return data
def encode_constructor_arguments(web3: Web3,
constructor_function: ContractConstructor,
*constructor_args,
**constructor_kwargs) -> HexStr:
"""
Takes a web3 constructor function and the arguments passed to it, and produces an encoding hex string
of the constructor arguments, following the standard ABI encoding conventions.
If there's no constructor, it returns None.
"""
constructor_abi = get_constructor_abi(constructor_function.abi)
if constructor_abi:
arguments = merge_args_and_kwargs(constructor_abi, constructor_args,
constructor_kwargs)
data = encode_abi(web3, constructor_abi, arguments)
else:
data = None
return data
def _encode_data_in_transaction(self, *args, **kwargs):
constructor_abi = get_constructor_abi(self.abi)
if constructor_abi:
if not args:
args = tuple()
if not kwargs:
kwargs = {}
arguments = merge_args_and_kwargs(constructor_abi, args, kwargs)
data = add_0x_prefix(
encode_abi(self.web3, constructor_abi, arguments, data=self.bytecode)
)
else:
data = to_hex(self.bytecode)
return data
def _encode_constructor_data(cls, args=None, kwargs=None):
constructor_abi = get_constructor_abi(cls.abi)
if constructor_abi:
if args is None:
args = tuple()
if kwargs is None:
kwargs = {}
arguments = merge_args_and_kwargs(constructor_abi, args, kwargs)
deploy_data = add_0x_prefix(
encode_abi(cls.web3, constructor_abi, arguments, data=cls.bytecode)
)
else:
deploy_data = to_hex(cls.bytecode)
return deploy_data
def from_contract_abi(cls, web3: Web3, fn_sign: str, fn_args: list,
contract_abi):
""" Create a `Calldata` according to the given contract abi """
assert isinstance(web3, Web3)
assert isinstance(fn_sign, str)
assert isinstance(fn_args, list)
fn_split = re.split('[(),]', fn_sign)
fn_name = fn_split[0]
fn_abi, fn_selector, fn_arguments = get_function_info(
fn_name,
abi_codec=web3.codec,
contract_abi=contract_abi,
args=fn_args)
calldata = encode_abi(web3, fn_abi, fn_arguments, fn_selector)
return cls(calldata)
def _encode_constructor_data(cls, args=None, kwargs=None):
constructor_abi = get_constructor_abi(cls.abi)
if constructor_abi:
if args is None:
args = tuple()
if kwargs is None:
kwargs = {}
arguments = merge_args_and_kwargs(constructor_abi, args, kwargs)
deploy_data = add_0x_prefix(
encode_abi(cls.web3, constructor_abi, arguments, data=cls.bytecode)
)
else:
deploy_data = to_hex(cls.bytecode)
return deploy_data
def from_signature(cls, web3: Web3, fn_sign: str, fn_args: list):
""" Allow to create a `Calldata` from a function signature and a list of arguments.
:param fn_sign: the function signature ie. "function(uint256,address)"
:param fn_args: arguments to the function ie. [123, "0x00...00"]
"""
assert isinstance(fn_sign, str)
assert isinstance(fn_args, list)
fn_split = re.split('[(),]', fn_sign)
fn_name = fn_split[0]
fn_args_type = [{"type": type} for type in fn_split[1:] if type]
fn_abi = {"type": "function", "name": fn_name, "inputs": fn_args_type}
fn_abi, fn_selector, fn_arguments = get_function_info("test", abi_codec=web3.codec, fn_abi=fn_abi, args=fn_args)
calldata = encode_abi(web3, fn_abi, fn_arguments, fn_selector)
return cls(calldata)
def _encode_constructor_data(cls, args=None, kwargs=None):
constructor_abi = get_constructor_abi(cls.abi)
if constructor_abi:
if args is None:
args = tuple()
if kwargs is None:
kwargs = {}
arguments = merge_args_and_kwargs(constructor_abi, args, kwargs)
deploy_data = add_0x_prefix(
encode_abi(cls.web3, constructor_abi, arguments, data=cls.bytecode)
)
else:
if args is not None or kwargs is not None:
msg = "Constructor args were provided, but no constructor function was provided."
raise TypeError(msg)
deploy_data = to_hex(cls.bytecode)
return deploy_data
def _encode_transaction_data(cls):
return add_0x_prefix(
encode_abi(cls.web3, cls.abi, cls.arguments, cls.selector))
def _encode_transaction_data(cls):
return add_0x_prefix(encode_abi(cls.web3, cls.abi, cls.arguments, cls.selector))
