48 lines
1.6 KiB
Python
Executable File
48 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.abspath('apps/blockchain-node/src'))
|
|
|
|
from sqlmodel import select
|
|
from aitbc_chain.database import session_scope, init_db
|
|
from aitbc_chain.models import Block
|
|
from datetime import datetime
|
|
import hashlib
|
|
|
|
def compute_block_hash(chain_id: str, height: int, parent_hash: str, timestamp: datetime) -> str:
|
|
data = f"{chain_id}{height}{parent_hash}{timestamp}".encode()
|
|
return "0x" + hashlib.sha256(data).hexdigest()
|
|
|
|
def create_genesis(chain_id: str):
|
|
print(f"Creating genesis block for {chain_id}...")
|
|
|
|
with session_scope() as session:
|
|
existing = session.exec(select(Block).where(Block.chain_id == chain_id).order_by(Block.height.desc()).limit(1)).first()
|
|
if existing:
|
|
print(f"Genesis block already exists for {chain_id}: #{existing.height}")
|
|
return
|
|
|
|
timestamp = datetime.utcnow()
|
|
genesis_hash = compute_block_hash(chain_id, 0, "0x00", timestamp)
|
|
genesis = Block(
|
|
chain_id=chain_id,
|
|
height=0,
|
|
hash=genesis_hash,
|
|
parent_hash="0x00",
|
|
proposer=f"{chain_id}-proposer",
|
|
timestamp=timestamp,
|
|
tx_count=0,
|
|
state_root=None,
|
|
)
|
|
session.add(genesis)
|
|
session.commit()
|
|
print(f"Genesis block created for {chain_id}: #{genesis.height}")
|
|
print(f"Hash: {genesis.hash}")
|
|
print(f"Proposer: {genesis.proposer}")
|
|
print(f"Timestamp: {genesis.timestamp}")
|
|
|
|
if __name__ == "__main__":
|
|
init_db()
|
|
for chain in ["ait-testnet", "ait-devnet"]:
|
|
create_genesis(chain)
|