mirror of
https://github.com/codex-storage/bittorrent-benchmarks.git
synced 2025-02-03 23:03:42 +00:00
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
|
|
from pydantic_core import ValidationError
|
|
|
|
from benchmarks.core.config import ConfigParser, ExperimentBuilder
|
|
from benchmarks.core.experiments.experiments import Experiment
|
|
from benchmarks.deluge.config import DelugeExperimentConfig
|
|
|
|
config_parser = ConfigParser()
|
|
config_parser.register(DelugeExperimentConfig)
|
|
|
|
|
|
def cmd_list(experiments: Dict[str, ExperimentBuilder[Experiment]], _):
|
|
"""
|
|
Lists the experiments available in CONFIG.
|
|
"""
|
|
print(f'Available experiments are:')
|
|
for experiment in experiments.keys():
|
|
print(f' - {experiment}')
|
|
|
|
|
|
def cmd_run(experiments: Dict[str, ExperimentBuilder[Experiment]], args):
|
|
"""
|
|
Runs the experiment with name EXPERIMENT.
|
|
"""
|
|
if args.experiment not in experiments:
|
|
print(f'Experiment {args.experiment} not found.')
|
|
sys.exit(-1)
|
|
experiments[args.experiment].build().run()
|
|
|
|
|
|
def _parse_config(config: Path) -> Dict[str, ExperimentBuilder[Experiment]]:
|
|
if not config.exists():
|
|
print(f'Config file {config} does not exist.')
|
|
sys.exit(-1)
|
|
|
|
with config.open(encoding='utf-8') as infile:
|
|
try:
|
|
return config_parser.parse(infile)
|
|
except ValidationError as e:
|
|
print(f'There were errors parsing the config file.')
|
|
for error in e.errors():
|
|
print(f' - {error["loc"]}: {error["msg"]} {error["input"]}')
|
|
sys.exit(-1)
|
|
|
|
|
|
def _init_logging():
|
|
import logging
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('config', type=Path, help="Path to the experiment configuration file.")
|
|
|
|
commands = parser.add_subparsers(required=True)
|
|
list_cmd = commands.add_parser('list', help='Lists available experiments.')
|
|
list_cmd.set_defaults(func=cmd_list)
|
|
|
|
run_cmd = commands.add_parser('run')
|
|
run_cmd.add_argument('experiment', type=str, help='Name of the experiment to run.')
|
|
run_cmd.set_defaults(func=cmd_run)
|
|
|
|
args = parser.parse_args()
|
|
|
|
_init_logging()
|
|
|
|
args.func(_parse_config(args.config), args)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|