status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update `imports_and_predefinitions`

This commit is contained in:
Hsiao-Wei Wang 2021-04-09 21:28:58 +08:00
parent 658ede2191
commit 1ffa436836
No known key found for this signature in database
GPG Key ID: 1111A8A81778319E
1 changed files with 23 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
setup.py
View File

@ -179,25 +179,22 @@ class SpecAdjustment(ABC):
class Phase0SpecAdjustment(SpecAdjustment):
@classmethod
def imports_and_predefinitions(cls) -> str:
return '''from eth2spec.config.config_util import apply_constants_config
from typing import (
Any, Callable, Dict, Set, Sequence, Tuple, Optional, TypeVar
)
return '''from lru import LRU
from dataclasses import (
dataclass,
field,
)
from typing import (
Any, Callable, Dict, Set, Sequence, Tuple, Optional, TypeVar
)
from lru import LRU
from eth2spec.config.config_util import apply_constants_config
from eth2spec.utils.ssz.ssz_impl import hash_tree_root, copy, uint_to_bytes
from eth2spec.utils.ssz.ssz_typing import (
View, boolean, Container, List, Vector, uint8, uint32, uint64,
Bytes1, Bytes4, Bytes32, Bytes48, Bytes96, Bitlist, Bitvector,
)
from eth2spec.utils import bls
from eth2spec.utils.hash_function import hash
SSZObject = TypeVar('SSZObject', bound=View)
@ -297,19 +294,18 @@ get_attesting_indices = cache_this(
class AltairSpecAdjustment(Phase0SpecAdjustment):
@classmethod
def imports_and_predefinitions(cls) -> str:
return '''from eth2spec.phase0 import spec as phase0
from eth2spec.config.config_util import apply_constants_config
from typing import (
Any, Dict, Set, Sequence, NewType, Tuple, TypeVar, Callable, Optional, Union
)
return '''from lru import LRU
from dataclasses import (
dataclass,
field,
)
from typing import (
Any, Dict, Set, Sequence, NewType, Tuple, TypeVar, Callable, Optional, Union
)
from lru import LRU
from eth2spec.config.config_util import apply_constants_config
from eth2spec.phase0 import spec as phase0
from eth2spec.utils.ssz.ssz_impl import hash_tree_root, copy, uint_to_bytes
from eth2spec.utils.ssz.ssz_typing import (
View, boolean, Container, List, Vector, uint8, uint32, uint64,
@ -320,7 +316,7 @@ from eth2spec.utils import bls
from eth2spec.utils.hash_function import hash
# Whenever altair is loaded, make sure we have the latest phase0
# Whenever this spec version is loaded, make sure we have the latest phase0
from importlib import reload
reload(phase0)
@ -363,19 +359,17 @@ assert (
class MergeSpecAdjustment(Phase0SpecAdjustment):
@classmethod
def imports_and_predefinitions(cls):
return '''from eth2spec.phase0 import spec as phase0
from eth2spec.config.config_util import apply_constants_config
from typing import (
Any, Callable, Dict, Set, Sequence, Tuple, Optional, TypeVar
)
return '''from lru import LRU
from dataclasses import (
dataclass,
field,
)
from typing import (
Any, Callable, Dict, Set, Sequence, Tuple, Optional, TypeVar
)
from lru import LRU
from eth2spec.phase0 import spec as phase0
from eth2spec.config.config_util import apply_constants_config
from eth2spec.utils.ssz.ssz_impl import hash_tree_root, copy, uint_to_bytes
from eth2spec.utils.ssz.ssz_typing import (
View, boolean, Container, List, Vector, uint8, uint32, uint64, uint256,
@ -383,9 +377,13 @@ from eth2spec.utils.ssz.ssz_typing import (
ByteList, ByteVector
)
from eth2spec.utils import bls
from eth2spec.utils.hash_function import hash
# Whenever this spec version is loaded, make sure we have the latest phase0
from importlib import reload
reload(phase0)
SSZObject = TypeVar('SSZObject', bound=View)
CONFIG_NAME = 'mainnet'