def __init__(self):
self.funding_goal = 10000 # 众筹目标
self.funding = 0 # 已众筹金额
self.max_block_number = block.number() + 1000 # 众筹最高区块
self.vote_dict = {} # 众筹情况数据存储
self.on_sale = True # 众筹状态
self.owner = msg.sender # 众筹所有者
def withdraw(self, public_key):
"""
提取质押金
:param public_key:
:return:
"""
# 是否有质押金冻结记录
require(public_key in self.deregister_list)
lock_stake_block_count = 5
return_stake_infos = []
for info in self.deregister_list[public_key]:
if (block.number() - info["deregisterBlockNumber"] >
lock_stake_block_count and info["owner"] == msg.sender):
return_stake_infos.append(info)
# 是否有已解冻质押金未提取，并且检查owner
require(len(return_stake_infos) > 0)
for info in return_stake_infos:
self.deregister_list[public_key].remove(info)
# 删除空KEY
if len(self.deregister_list[public_key]) <= 0:
self.deregister_list.pop(public_key)
return_stake = 0
for info in return_stake_infos:
return_stake += info["stake"]
# 转账
assert return_stake > 0
assert account.get_balance(this) > return_stake
account.transfer(msg.sender, return_stake)
def register(self, public_key, miner_type, vrf_public_key=""):
"""
申请成为矿工，转账金额做为质押金
:param public_key: 公钥
:param miner_type: 0=轻节点 1=重节点
:param vrf_public_key: 重节点的vrf公钥，轻节点不需要
:return:
"""
# 账户余额检查
# require(account.get_balance(msg.sender) >= msg.value)
# 质押金检查
miner_light_stake = 10
miner_heavy_stake = 50
if miner_type == 0:
require(msg.value >= miner_light_stake)
elif miner_type == 1:
require(msg.value >= miner_heavy_stake)
else:
assert False
# 检查是否已注册
require(public_key not in self.register_list)
#
info = {
"registerBlockNumber": block.number(),
"stake": msg.value,
"type": miner_type,
"vrfPk": vrf_public_key,
"owner": msg.sender
}
self.register_list[public_key] = info
def deregister(self, public_key):
"""
注销矿工资格
:param public_key:
:return:
"""
if public_key in self.register_list:
info = self.register_list.pop(public_key)
info["deregisterBlockNumber"] = block.number()
infos = []
if public_key in self.deregister_list:
infos = self.deregister_list[public_key]
infos.append(info)
self.deregister_list[public_key] = infos
def sale(self):
if self.max_block_number < block.number():
self.on_sale = False
if not self.on_sale:
raise Exception("not on sale")
value = msg.value
sender = msg.sender
self.funding = self.funding + value
if self.funding > self.funding_goal:
value = self.funding_goal - self.funding
self.on_sale = False
account.transfer(sender, value)
return
balance = self.vote_dict.get(sender, 0)
self.vote_dict[sender] = balance + value
print(self.vote_dict)