mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 18:14:19 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Collection
|
|
|
|
from chia_rs import FullBlock, HeaderBlock, SpendBundleConditions
|
|
from chia_rs.sized_bytes import bytes32
|
|
from chia_rs.sized_ints import uint64
|
|
from chiabip158 import PyBIP158
|
|
|
|
from chia.types.blockchain_format.coin import Coin
|
|
|
|
|
|
def get_block_header(
|
|
block: FullBlock, removals_and_additions: tuple[Collection[bytes32], Collection[Coin]] | None = None
|
|
) -> HeaderBlock:
|
|
"""
|
|
Returns a HeaderBlock from a FullBlock.
|
|
If `removals_and_additions` is not None, account for them, as well as
|
|
reward coins, in the creation of the transactions filter, otherwise create
|
|
an empty one.
|
|
"""
|
|
# Guard against passing removals and additions for a non-transaction block.
|
|
assert removals_and_additions is None or block.is_transaction_block()
|
|
# Create an empty filter to begin with
|
|
byte_array_tx: list[bytearray] = []
|
|
if removals_and_additions is not None and block.is_transaction_block():
|
|
removals_names, tx_addition_coins = removals_and_additions
|
|
for coin in tx_addition_coins:
|
|
byte_array_tx.append(bytearray(coin.puzzle_hash))
|
|
for coin in block.get_included_reward_coins():
|
|
byte_array_tx.append(bytearray(coin.puzzle_hash))
|
|
for name in removals_names:
|
|
byte_array_tx.append(bytearray(name))
|
|
|
|
bip158: PyBIP158 = PyBIP158(byte_array_tx)
|
|
encoded_filter: bytes = bytes(bip158.GetEncoded())
|
|
|
|
return HeaderBlock(
|
|
block.finished_sub_slots,
|
|
block.reward_chain_block,
|
|
block.challenge_chain_sp_proof,
|
|
block.challenge_chain_ip_proof,
|
|
block.reward_chain_sp_proof,
|
|
block.reward_chain_ip_proof,
|
|
block.infused_challenge_chain_ip_proof,
|
|
block.foliage,
|
|
block.foliage_transaction_block,
|
|
encoded_filter,
|
|
block.transactions_info,
|
|
)
|
|
|
|
|
|
def tx_removals_and_additions(results: SpendBundleConditions | None) -> tuple[list[bytes32], list[Coin]]:
|
|
"""
|
|
Doesn't return farmer and pool reward.
|
|
"""
|
|
|
|
removals: list[bytes32] = []
|
|
additions: list[Coin] = []
|
|
|
|
# build removals list
|
|
if results is None:
|
|
return [], []
|
|
for spend in results.spends:
|
|
removals.append(bytes32(spend.coin_id))
|
|
for puzzle_hash, amount, _ in spend.create_coin:
|
|
additions.append(Coin(spend.coin_id, puzzle_hash, uint64(amount)))
|
|
|
|
return removals, additions
|