33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
from typing import Dict
|
|
from .base import BaseSpecBuilder
|
|
from ..constants import WHISK
|
|
|
|
|
|
class WhiskSpecBuilder(BaseSpecBuilder):
|
|
fork: str = WHISK
|
|
|
|
@classmethod
|
|
def imports(cls, preset_name: str):
|
|
return f'''
|
|
from eth2spec.capella import {preset_name} as capella
|
|
import curdleproofs
|
|
import json
|
|
'''
|
|
|
|
@classmethod
|
|
def hardcoded_custom_type_dep_constants(cls, spec_object) -> str:
|
|
# Necessary for custom types `WhiskShuffleProof` and `WhiskTrackerProof`
|
|
return {
|
|
'WHISK_MAX_SHUFFLE_PROOF_SIZE': spec_object.preset_vars['WHISK_MAX_SHUFFLE_PROOF_SIZE'].value,
|
|
'WHISK_MAX_OPENING_PROOF_SIZE': spec_object.preset_vars['WHISK_MAX_OPENING_PROOF_SIZE'].value,
|
|
'WHISK_VALIDATORS_PER_SHUFFLE': spec_object.preset_vars['WHISK_VALIDATORS_PER_SHUFFLE'].value,
|
|
'CURDLEPROOFS_N_BLINDERS': spec_object.preset_vars['CURDLEPROOFS_N_BLINDERS'].value,
|
|
}
|
|
|
|
@classmethod
|
|
def hardcoded_ssz_dep_constants(cls) -> Dict[str, str]:
|
|
constants = {
|
|
'EXECUTION_PAYLOAD_GINDEX': 'GeneralizedIndex(41)',
|
|
}
|
|
return {**super().hardcoded_ssz_dep_constants(), **constants}
|