/ src / main.py
main.py
 1  # Python Imports
 2  import typer
 3  
 4  # Project Imports
 5  from src.plotter.run_plotter import run_plotter
 6  from src.simulation_runner.runner import SimulationRunner
 7  from src.utilities.files.json.simulation_config_parser import SimulationConfigParser
 8  
 9  
10  def main(
11          run_type: str = typer.Option(..., "--run", "-r"),
12          configuration_file: str = typer.Option(..., "--configuration-file", "-cf")
13  ):
14      parser = SimulationConfigParser(configuration_file)
15      arguments_config, simulation_config, plotter_config = parser.read_content()
16      # Calls depending on json config
17      if run_type == "simulation":
18          runner = SimulationRunner(arguments_config, simulation_config)
19          runner.run_simulation()
20      elif run_type == "plotter":
21          run_plotter(arguments_config, plotter_config)
22      else:
23          runner = SimulationRunner(arguments_config, simulation_config)
24          runner.run_simulation()
25          run_plotter()
26  
27  
28  if __name__ == '__main__':
29      typer.run(main)