45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
from eth2spec.phase0 import spec
|
||
|
|
from eth2spec.utils.merkle_minimal import get_merkle_root, calc_merkle_tree_from_leaves, get_merkle_proof
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
|
||
|
|
def create_genesis_state(deposits: List[spec.Deposit]) -> spec.BeaconState:
|
||
|
|
deposit_root = get_merkle_root((tuple([spec.hash(dep.data.serialize()) for dep in deposits])))
|
||
|
|
|
||
|
|
return spec.get_genesis_beacon_state(
|
||
|
|
deposits,
|
||
|
|
genesis_time=0,
|
||
|
|
genesis_eth1_data=spec.Eth1Data(
|
||
|
|
deposit_root=deposit_root,
|
||
|
|
deposit_count=len(deposits),
|
||
|
|
block_hash=spec.ZERO_HASH,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def create_deposits(pubkeys: List[spec.BLSPubkey], withdrawal_cred: List[spec.Bytes32]) -> List[spec.Deposit]:
|
||
|
|
|
||
|
|
# Mock proof of possession
|
||
|
|
proof_of_possession = b'\x33' * 96
|
||
|
|
|
||
|
|
deposit_data = [
|
||
|
|
spec.DepositData(
|
||
|
|
pubkey=pubkeys[i],
|
||
|
|
withdrawal_credentials=spec.BLS_WITHDRAWAL_PREFIX_BYTE + withdrawal_cred[i][1:],
|
||
|
|
amount=spec.MAX_DEPOSIT_AMOUNT,
|
||
|
|
proof_of_possession=proof_of_possession,
|
||
|
|
) for i in range(len(pubkeys))
|
||
|
|
]
|
||
|
|
|
||
|
|
# Fill tree with existing deposits
|
||
|
|
deposit_data_leaves = [spec.hash(data.serialize()) for data in deposit_data]
|
||
|
|
tree = calc_merkle_tree_from_leaves(tuple(deposit_data_leaves))
|
||
|
|
|
||
|
|
return [
|
||
|
|
spec.Deposit(
|
||
|
|
proof=list(get_merkle_proof(tree, item_index=i)),
|
||
|
|
index=i,
|
||
|
|
data=deposit_data[i]
|
||
|
|
) for i in range(len(deposit_data))
|
||
|
|
]
|