56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import logging
|
|
import json
|
|
import os
|
|
|
|
logg = logging.getLogger()
|
|
|
|
class AccountRegistry:
|
|
|
|
abi = None
|
|
|
|
def __init__(self, w3, address, signer_address=None):
|
|
if AccountRegistry.abi == None:
|
|
moddir = os.path.dirname(__file__)
|
|
datadir = os.path.join(moddir, 'data')
|
|
f = open(os.path.join(datadir, 'registry.abi.json'), 'r')
|
|
AccountRegistry.abi = json.load(f)
|
|
f.close()
|
|
self.contract = w3.eth.contract(abi=AccountRegistry.abi, address=address)
|
|
self.w3 = w3
|
|
if signer_address != None:
|
|
self.signer_address = signer_address
|
|
else:
|
|
if type(self.w3.eth.defaultAccount).__name__ == 'Empty':
|
|
self.w3.eth.defaultAccount = self.w3.eth.accounts[0]
|
|
self.signer_address = self.w3.eth.defaultAccount
|
|
|
|
|
|
def add(self, address):
|
|
gasPrice = self.w3.eth.gasPrice;
|
|
tx_hash = self.contract.functions.add(address).transact({
|
|
'gasPrice': gasPrice,
|
|
'gas': 100000,
|
|
'from': self.signer_address,
|
|
})
|
|
|
|
|
|
def count(self):
|
|
return self.contract.functions.count().call()
|
|
|
|
|
|
def have(self, address):
|
|
r = self.contract.functions.accountsIndex(address).call()
|
|
return r != 0
|
|
|
|
|
|
def last(self, n):
|
|
c = self.count()
|
|
lo = c - n - 1
|
|
if lo < 0:
|
|
lo = 0
|
|
accounts = []
|
|
for i in range(c - 1, lo, -1):
|
|
a = self.contract.functions.accounts(i).call()
|
|
accounts.append(a)
|
|
return accounts
|