mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-22 00:52:32 +03:00
8291f0221a
* Rework sized bytes for type hinting compatibility
* add a bunch of type: ignores
* this will be handled elsewhere
* noqa E501 instead of changing code
* normalize comment plurality
* @classmethod
* Revert "@classmethod"
This reverts commit 95db80e339.
* add ignore in benchmarks
* just E501 again...
* add some new type: ignores
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
from typing import List
|
|
|
|
from blspy import AugSchemeMPL, G2Element, PrivateKey
|
|
|
|
from chia.types.blockchain_format.sized_bytes import bytes32
|
|
from chia.types.coin_spend import CoinSpend
|
|
from chia.util.condition_tools import conditions_by_opcode, conditions_for_solution, pkm_pairs_for_conditions_dict
|
|
from tests.core.make_block_generator import GROUP_ORDER, int_to_public_key
|
|
from tests.block_tools import test_constants
|
|
|
|
|
|
class KeyTool(dict):
|
|
@classmethod
|
|
def __new__(cls, *args):
|
|
return dict.__new__(*args)
|
|
|
|
def add_secret_exponents(self, secret_exponents: List[int]) -> None:
|
|
for _ in secret_exponents:
|
|
self[bytes(int_to_public_key(_))] = _ % GROUP_ORDER
|
|
|
|
def sign(self, public_key: bytes, message_hash: bytes32) -> G2Element:
|
|
secret_exponent = self.get(public_key)
|
|
if not secret_exponent:
|
|
raise ValueError("unknown pubkey %s" % public_key.hex())
|
|
bls_private_key = PrivateKey.from_bytes(secret_exponent.to_bytes(32, "big"))
|
|
return AugSchemeMPL.sign(bls_private_key, message_hash)
|
|
|
|
def signature_for_solution(self, coin_spend: CoinSpend, additional_data: bytes) -> AugSchemeMPL:
|
|
signatures = []
|
|
err, conditions, cost = conditions_for_solution(
|
|
coin_spend.puzzle_reveal, coin_spend.solution, test_constants.MAX_BLOCK_COST_CLVM
|
|
)
|
|
assert conditions is not None
|
|
conditions_dict = conditions_by_opcode(conditions)
|
|
for public_key, message_hash in pkm_pairs_for_conditions_dict(
|
|
conditions_dict, coin_spend.coin.name(), additional_data
|
|
):
|
|
# TODO: address hint error and remove ignore
|
|
# error: Argument 2 to "sign" of "KeyTool" has incompatible type "bytes"; expected "bytes32"
|
|
# [arg-type]
|
|
signature = self.sign(public_key, message_hash) # type: ignore[arg-type]
|
|
signatures.append(signature)
|
|
return AugSchemeMPL.aggregate(signatures)
|