def __init__(self,
network_id: int,
max_peers: int = DEFAULT_MAX_PEERS,
data_dir: str = None,
nodekey_path: str = None,
logfile_path: str = None,
nodekey: PrivateKey = None,
sync_mode: str = SYNC_FULL,
port: int = 30303,
preferred_nodes: Tuple[KademliaNode, ...] = None,
bootstrap_nodes: Tuple[KademliaNode, ...] = None) -> None:
self.network_id = network_id
self.max_peers = max_peers
self.sync_mode = sync_mode
self.port = port
if not preferred_nodes and network_id in DEFAULT_PREFERRED_NODES:
self.preferred_nodes = DEFAULT_PREFERRED_NODES[self.network_id]
else:
self.preferred_nodes = preferred_nodes
if bootstrap_nodes is None:
if self.network_id == MAINNET_NETWORK_ID:
self.bootstrap_nodes = tuple(
KademliaNode.from_uri(enode)
for enode in MAINNET_BOOTNODES)
elif self.network_id == ROPSTEN_NETWORK_ID:
self.bootstrap_nodes = tuple(
KademliaNode.from_uri(enode)
for enode in ROPSTEN_BOOTNODES)
else:
self.bootstrap_nodes = bootstrap_nodes
# validation
if nodekey is not None and nodekey_path is not None:
raise ValueError(
"It is invalid to provide both a `nodekey` and a `nodekey_path`"
)
# set values
if data_dir is not None:
self.data_dir = data_dir
else:
self.data_dir = get_data_dir_for_network_id(self.network_id)
if nodekey_path is not None:
self.nodekey_path = nodekey_path
elif nodekey is not None:
self.nodekey = nodekey
if logfile_path is not None:
self.logfile_path = logfile_path
else:
self.logfile_path = get_logfile_path(self.data_dir)
def test_chain_config_computed_properties_custom_xdg(tmpdir, xdg_trinity_root):
alt_xdg_root = tmpdir.mkdir('trinity-custom')
assert not is_under_path(alt_xdg_root, xdg_trinity_root)
data_dir = get_data_dir_for_network_id(1, alt_xdg_root)
chain_config = TrinityConfig(trinity_root_dir=alt_xdg_root, network_id=1)
assert chain_config.network_id == 1
assert chain_config.data_dir == data_dir
assert chain_config.database_dir == data_dir / DATABASE_DIR_NAME / "full"
assert chain_config.nodekey_path == get_nodekey_path(data_dir)
def data_dir(self) -> Path:
"""
The data_dir is the base directory that all chain specific information
for a given chain is stored.
All defaults for chain directories are resolved relative to this
directory.
"""
if self._data_dir is not None:
return self._data_dir
else:
return get_data_dir_for_network_id(self.network_id,
self.trinity_root_dir)
