import io
from typing import Any, Callable, Optional
from multiversx_sdk.abi.interface import ISingleValue
from multiversx_sdk.abi.shared import convert_native_value_to_list
[docs]
class ArrayValue:
def __init__(self,
length: int,
items: Optional[list[ISingleValue]] = None,
item_creator: Optional[Callable[[], ISingleValue]] = None) -> None:
self.length = length
self.items = items or []
if len(self.items):
self.guard_length(self.items)
self.item_creator = item_creator
[docs]
def encode_nested(self, writer: io.BytesIO):
self._encode_list_items(writer)
[docs]
def encode_top_level(self, writer: io.BytesIO):
self._encode_list_items(writer)
[docs]
def decode_nested(self, reader: io.BytesIO):
self.items = []
for _ in range(self.length):
self._decode_list_item(reader)
[docs]
def decode_top_level(self, data: bytes):
reader = io.BytesIO(data)
self.items = []
while reader.tell() < len(data):
self._decode_list_item(reader)
def _encode_list_items(self, writer: io.BytesIO):
for item in self.items:
item.encode_nested(writer)
def _decode_list_item(self, reader: io.BytesIO):
if self.item_creator is None:
raise Exception("cannot decode list: item creator is None")
new_item = self.item_creator()
new_item.decode_nested(reader)
self.items.append(new_item)
[docs]
def set_payload(self, value: Any):
if not self.item_creator:
raise ValueError("populating an array from a native object requires the item creator to be set")
native_items, _ = convert_native_value_to_list(value)
self.guard_length(native_items)
self.items.clear()
for native_item in native_items:
item = self.item_creator()
item.set_payload(native_item)
self.items.append(item)
[docs]
def get_payload(self) -> Any:
return [item.get_payload() for item in self.items]
[docs]
def guard_length(self, items: list[ISingleValue]):
if len(items) != self.length:
raise ValueError(f"wrong length, expected: {self.length}, actual: {len(items)}")
def __eq__(self, other: Any) -> bool:
return (
isinstance(other, ArrayValue)
and self.length == other.length
and self.items == other.items
and self.item_creator == other.item_creator
)
def __iter__(self) -> Any:
return iter(self.items)