mirror of
https://github.com/GoodStartLabs/AI_Diplomacy.git
synced 2026-04-19 12:58:09 +00:00
109 lines
3.7 KiB
Python
109 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Orchestration script to run all three analysis scripts in sequence:
|
|
1. 1_make_longform_orders_data.py
|
|
2. 2_make_convo_data.py
|
|
3. 3_make_phase_data.py
|
|
|
|
This script passes the same CLI arguments to each script.
|
|
|
|
eg
|
|
|
|
# Basic usage with all parameters
|
|
python analysis/make_all_analysis_data.py --selected_game game1 game2 --game_data_folder "/path/to/Game Data" --output_folder "/path/to/Game Data - Analysis" --analysis_folder "/path/to/analysis"
|
|
|
|
# Using output_folder as analysis_folder
|
|
python analysis/make_all_analysis_data.py --selected_game game1 --game_data_folder "/path/to/Game Data" --output_folder "/path/to/Game Data - Analysis"
|
|
|
|
# or leave out to process all games in the data folder
|
|
python analysis/make_all_analysis_data.py --game_data_folder "/path/to/Game Data" --output_folder "/path/to/Game Data - Analysis"
|
|
"""
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
def run_script(script_name, args_dict):
|
|
"""Run a script with the provided arguments."""
|
|
print(f"\n=== Running {script_name} ===")
|
|
|
|
cmd = [sys.executable, script_name]
|
|
|
|
# Add all arguments that apply to this script
|
|
for arg_name, arg_value in args_dict.items():
|
|
if arg_value: # Only add if value exists
|
|
if isinstance(arg_value, list):
|
|
cmd.append(f"--{arg_name}")
|
|
cmd.extend(arg_value)
|
|
else:
|
|
cmd.append(f"--{arg_name}")
|
|
cmd.append(str(arg_value))
|
|
|
|
print(f"Command: {' '.join(cmd)}")
|
|
|
|
result = subprocess.run(cmd, text=True)
|
|
|
|
# Print output
|
|
if result.stdout:
|
|
print("\nOutput:")
|
|
print(result.stdout)
|
|
|
|
# Check for errors
|
|
if result.returncode != 0:
|
|
print("\nERROR:")
|
|
print(result.stderr)
|
|
return False
|
|
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Run all three analysis scripts in sequence with the same arguments.")
|
|
|
|
parser.add_argument(
|
|
"--selected_game",
|
|
type=str,
|
|
nargs='*',
|
|
help="One or more specific games to process. If not provided, all games in the data folder will be processed."
|
|
)
|
|
parser.add_argument(
|
|
"--game_data_folder",
|
|
type=str,
|
|
required=True,
|
|
help="The folder where game data is stored."
|
|
)
|
|
parser.add_argument(
|
|
"--analysis_folder",
|
|
type=str,
|
|
required=True,
|
|
help="The folder to save the new analysis folders and files"
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Convert namespace to dictionary
|
|
args_dict = vars(args)
|
|
args_dict["analysis_folder"] = Path(args_dict["analysis_folder"])
|
|
args_dict["game_data_folder"] = Path(args_dict["game_data_folder"])
|
|
|
|
# Prepare arguments for each script
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
script1_path = os.path.join(script_dir, "1_make_longform_orders_data.py")
|
|
script2_path = os.path.join(script_dir, "2_make_convo_data.py")
|
|
script3_path = os.path.join(script_dir, "3_make_phase_data.py")
|
|
|
|
# Run scripts in order
|
|
if not run_script(script1_path, {k: v for k, v in args_dict.items()}):
|
|
print("ERROR: Failed to run 1_make_longform_orders_data.py. Stopping.")
|
|
sys.exit(1)
|
|
|
|
if not run_script(script2_path, {k: v for k, v in args_dict.items()}):
|
|
print("ERROR: Failed to run 2_make_convo_data.py. Stopping.")
|
|
sys.exit(1)
|
|
|
|
if not run_script(script3_path, {k: v for k, v in args_dict.items()}):
|
|
print("ERROR: Failed to run 3_make_phase_data.py. Stopping.")
|
|
sys.exit(1)
|
|
|
|
print("\n=== All analysis scripts completed successfully! ===")
|