class DemoClient(object):
"""
"""
def __init__(self, rpc_addr: str, wallet_file: str, is_maker: bool, pwd,
avm_file, abi_file):
self.sdk = OntologySdk()
self.sdk.rpc.set_address(rpc_addr)
self.sdk.open_wallet(wallet_file)
self.is_maker = is_maker # 可以查询一个合约存不存在，存在就可以让maker是True，而不必自己指定了
self.__pwd = pwd
self.__get_code_from_avm(avm_file)
self.__get_abi_from_file(abi_file)
@staticmethod
def is_transaction_success(txid):
if len(txid) != 64:
return False
return True
def start_game(self, acc_index):
if not self.is_maker:
return
acc = self.get_account_by_index(acc_index)
txid = self.deploy_contract(need_storage=True,
name="sixsixsix",
code_ver="1.0",
author="tomzou",
email="*****@*****.**",
desp="demo",
acc=acc,
gas_limit=20200000,
gas_price=500)
if not self.is_transaction_success(txid):
raise Exception(txid)
return txid
def get_account_by_index(self, acc_index):
wm = self.sdk.get_wallet_manager()
return wm.get_account(
wm.get_wallet().get_account_by_index(acc_index).get_address(),
self.__pwd[acc_index])
def deploy_contract(self, need_storage, name, code_ver, author, email,
desp, acc, gas_limit, gas_price):
tx = self.sdk.neo_vm().make_deploy_transaction(
self.code,
need_storage=need_storage,
name=name,
code_version=code_ver,
author=author,
email=email,
desp=desp,
payer=acc.get_address_base58(),
gas_limit=gas_limit,
gas_price=gas_price)
self.sdk.sign_transaction(tx, acc)
return self.sdk.get_rpc().send_raw_transaction(tx)
def __get_code_from_avm(self, avm_file):
f = open(avm_file)
self.code = f.readline()
code_addr_obj = Address.address_from_vm_code(self.code)
self.code_addr = code_addr_obj.to_reverse_hex_str()
self.contract_acc = code_addr_obj.to_base58()
self.contract_addr = code_addr_obj.to_byte_array()
f.close()
def __get_abi_from_file(self, abi_file):
f = open(abi_file)
abi_str = f.readline()
abi = json.loads(abi_str,
object_hook=lambda d: namedtuple('X', d.keys())
(*d.values()))
self.abi_info = AbiInfo("0x" + self.code_addr, "main", abi.functions,
[])
def luck_roll(self, amount, acc_index):
"""
玩家通过本方法参与游戏
:param amount:
:param acc_index:
:return:
"""
self.join_game(amount, acc_index)
if self.bet(amount, acc_index) is True:
print('I win, you bitch')
else:
print('Go back and work, FOOL')
def join_game(self, amount, acc_index):
acc = self.get_account_by_index(acc_index)
# 调用相应的函数，需写完合约再回头搞
func_join = self.abi_info.get_function("join")
func_join.set_params_value(acc.get_address_base58(), amount)
func_read_res = self.abi_info.get_function("is_game_ready")
txid = self.sdk.neo_vm().send_transaction(self.contract_addr, acc, acc,
2000000, 500, func_join,
False)
print(txid)
if not self.is_transaction_success(txid):
raise Exception(txid)
before = time.time()
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(txid)
while tx_info == 'unknown transaction' or tx_info is None:
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(
txid)
print('cost %f seconds to confirm tx' % (time.time() - before))
try:
notify_info = tx_info['Notify'][0]['States']
except Exception:
raise Exception(tx_info)
print(notify_info)
if notify_info == '4e6f206d6f72652073656174':
raise Exception("No more seat")
elif notify_info == '6a6f696e207375636365737366756c6c79':
# 等待游戏结束，不停滴向合约pull结果
while True:
time.sleep(1)
res = self.sdk.neo_vm().send_transaction(
self.contract_addr, acc, acc, 2000000, 500, func_read_res,
True)
if res is True:
print(res)
break
def bet(self, amount, acc_index):
"""
人都坐好了，开赌！
掷出骰子并等待结果，返回输赢
:param amount: 需与join相同
:param acc_index: 使用钱包中第几个账户，需和join方法统一
:return: True代表胜利，False代表输了。。
"""
acc = self.get_account_by_index(acc_index)
tx = self.sdk.native_vm().asset().new_transfer_transaction(
'ont', acc.get_address_base58(), self.contract_acc, amount,
acc.get_address_base58(), 2000000, 500)
self.sdk.sign_transaction(tx, acc)
txid = self.sdk.get_rpc().send_raw_transaction(tx)
if not self.is_transaction_success(txid):
raise Exception('Some thing wrong with tx:%s' % txid)
func_get_amounts = self.abi_info.get_function('get_amounts')
sum_amounts = self.sdk.neo_vm().send_transaction(
self.contract_addr, acc, acc, 2000000, 500, func_get_amounts, True)
contract_balance = self.sdk.get_rpc().get_balance(self.contract_acc)
while sum_amounts > contract_balance:
contract_balance = self.sdk.get_rpc().get_balance(
self.contract_acc)
time.sleep(0.5)
# 直接开始游戏，获取随机数
func_roll_dice = self.abi_info.get_function('roll_dice')
func_roll_dice.set_params_value(acc.get_address_base58())
txid = self.sdk.neo_vm().send_transaction(self.contract_addr, acc, acc,
2000000, 500, func_roll_dice,
False)
if not self.is_transaction_success(txid):
raise Exception('Some thing wrong with tx:%s' % txid)
before = time.time()
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(txid)
while tx_info == 'unknown transaction' or tx_info is None:
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(
txid)
print('cost %f seconds to confirm tx' % (time.time() - before))
try:
notify_info = tx_info['Notify'][0]['States']
except Exception:
raise Exception(tx_info)
if type(notify_info) is int:
print('your game is done and your dice number is %d' % notify_info)
else:
raise Exception(bytes.fromhex(notify_info).decode())
func_get_res = self.abi_info.get_function('get_res')
res = self.sdk.neo_vm().send_transaction(self.contract_addr, acc, acc,
2000000, 500, func_get_res,
True)
while res is False:
res = self.sdk.neo_vm().send_transaction(self.contract_addr, acc,
acc, 2000000, 500,
func_get_res, True)
time.sleep(0.5)
if res == acc.get_address_base58():
return True
else:
return False
class DemoClient(object):
"""
"""
def __init__(self, rpc_addr: str, wallet_file: str, is_maker: bool, pwd, avm_file, abi_file):
self.sdk = OntologySdk()
self.sdk.rpc.set_address(rpc_addr)
self.sdk.open_wallet(wallet_file)
self.is_maker = is_maker # 可以查询一个合约存不存在，存在就可以让maker是True，而不必自己指定了
self.__pwd = pwd
self.__get_code_from_avm(avm_file)
self.__get_abi_from_file(abi_file)
@staticmethod
def is_transaction_success(txid):
if len(txid) != 64:
return False
return True
def start_game(self, acc_index):
if not self.is_maker:
return
acc = self.get_account_by_index(acc_index)
txid = self.deploy_contract(need_storage=True, name="sixsixsix", code_ver="1.0", author="tomzou",
email="*****@*****.**", desp="demo", acc=acc, gas_limit=20200000, gas_price=500)
if not self.is_transaction_success(txid):
raise Exception(txid)
return txid
def get_account_by_index(self, acc_index):
wm = self.sdk.get_wallet_manager()
return wm.get_account(wm.get_wallet().get_account_by_index(acc_index).get_address(), self.__pwd[acc_index])
def deploy_contract(self, need_storage, name, code_ver, author, email, desp, acc, gas_limit, gas_price):
tx = self.sdk.neo_vm().make_deploy_transaction(self.code, need_storage=need_storage, name=name,
code_version=code_ver, author=author, email=email, desp=desp,
payer=acc.get_address_base58(), gas_limit=gas_limit, gas_price=gas_price)
self.sdk.sign_transaction(tx, acc)
return self.sdk.get_rpc().send_raw_transaction(tx)
def __get_code_from_avm(self, avm_file):
f = open(avm_file)
self.code = f.readline()
code_addr_obj = Address.address_from_vm_code(self.code)
self.code_addr = code_addr_obj.to_reverse_hex_str()
self.contract_acc = code_addr_obj.to_base58()
self.contract_addr = code_addr_obj.to_byte_array() # 应该是这个地址
f.close()
def __get_abi_from_file(self, abi_file):
f = open(abi_file)
abi_str = f.readline()
abi = json.loads(abi_str, object_hook=lambda d: namedtuple('X', d.keys())(*d.values()))
self.abi_info = AbiInfo("0x" + self.code_addr, "main", abi.functions, [])
def luck_roll(self, amount, acc_index):
"""
玩家通过本方法参与游戏
:param amount:
:param acc_index:
:return:
"""
self.join_game(amount, acc_index)
def join_game(self, amount, acc_index):
acc = self.get_account_by_index(acc_index)
func_join = self.abi_info.get_function("join")
func_join.set_params_value(acc.get_address().to_byte_array(), amount)
txid = self.sdk.neo_vm().send_transaction(self.contract_addr, acc, acc, 2000000, 500, func_join, False)
if not self.is_transaction_success(txid):
raise Exception(txid)
before = time.time()
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(txid)
while tx_info == 'unknown transaction' or tx_info is None:
tx_info = self.sdk.get_rpc().get_smart_contract_event_by_txhash(txid)
print('cost %f seconds to confirm tx' %(time.time() - before))
try:
notify_info = tx_info['Notify'][0]['States']
except Exception:
raise Exception(tx_info)
if notify_info == '626574207472616e666572206661696c6564':
raise Exception('bet tranfer failed')
elif notify_info == '736574746c65207472616e73666572206661696c6564':
raise Exception('settle transfer failed')
else:
print(bytes.fromhex(notify_info).decode())
