from ..tl.functions import Raw_GetAccountState, Raw_GetTransactions
from ..tl.types import Internal_TransactionId
from ..utils import sha256, contracts
class StateMethods:
async def load_state(self, **kwargs):
self.state = await self.client.tonlib_wrapper.execute(
Raw_GetAccountState(self.account_address), **kwargs
)
async def get_state(self, force=False, **kwargs):
if self.state is None or force:
await self.load_state(**kwargs)
return self.state
async def detect_type(self, **kwargs):
state = await self.get_state(**kwargs)
return contracts.get(sha256(state.code), None)
async def get_balance(self, **kwargs):
return int(
(await self.get_state(force=True, **kwargs)).balance # nanoTONs
)
async def get_transactions(self, from_transaction_lt=None, from_transaction_hash=None, to_transaction_lt=0, limit=10):
if from_transaction_lt is None or from_transaction_hash is None:
state = await self.get_state(force=True)
from_transaction_lt, from_transaction_hash = state.last_transaction_id.lt, state.last_transaction_id.hash
reach_lt = False
all_transactions = []
current_tx = Internal_TransactionId(
from_transaction_lt, from_transaction_hash
)
while not reach_lt and len(all_transactions) < limit:
raw_transactions = await self.client.tonlib_wrapper.execute(
Raw_GetTransactions(
self.account_address,
current_tx
)
)
transactions, current_tx = raw_transactions.transactions, raw_transactions.__dict__.get("previous_transaction_id", None)
for tx in transactions:
tlt = int(tx.transaction_id.lt)
if tlt <= to_transaction_lt:
reach_lt = True
break
all_transactions.append(tx)
if current_tx is None:
break
if int(current_tx.lt) == 0:
break
return all_transactions
