suite.py

"""run full suite of sims"""
import argparse
import csv
import os
import subprocess
import sys
import yaml


with open("config.yml", "r", encoding="utf8") as ymlfile:
    config = yaml.load(ymlfile, Loader=yaml.FullLoader)


def call_process(process_args):
    """runs a process and constantly monitors for output"""
    subprocess.check_call(process_args, stdout=sys.stdout,
                          stderr=subprocess.STDOUT)


def update_state(directory, sim_type, output_file, script):
    """updates state text file"""
    with open(output_file, 'a+', encoding="utf8") as file:
        file.write(f"{directory},{sim_type},{script},\n")
        file.close()


def check_state(sim_dir, sim_type, output_file, script):
    """opens state file to see if the sim has been ran yet"""
    with open(output_file, 'r', encoding="utf8") as file:
        sims = csv.reader(file, delimiter=',')
        for row in sims:
            if len(row) == 0:
                continue
            if row[0] == sim_dir and row[1] == sim_type and row[2] == script:
                return False
    return True


def generate_args(sim_dir, sim_type, script, ptr):
    """generates arguments for each script based on input"""
    arguments = []
    if sim_type == "composite":
        if ptr:
            arguments = ["python", script, sim_dir, ptr]
        else:
            arguments = ["python", script, sim_dir]
    elif sim_type == "dungeons":
        if ptr:
            arguments = ["python", script, sim_dir, "--dungeons", ptr]
        else:
            arguments = ["python", script, sim_dir, "--dungeons"]
    return arguments


def run_suite(sim_dir, sim_type, output_file, sim, ptr):
    """helper function to orchestrate other calls"""
    if check_state(sim_dir, sim_type, output_file, "profiles"):
        call_process(generate_args(sim_dir, sim_type, "profiles.py", ptr))
        update_state(sim_dir, sim_type, output_file, "profiles")

    if check_state(sim_dir, sim_type, output_file, "sim"):
        print(f"Running sim suite for {sim} - Composite")
        call_process(generate_args(sim_dir, sim_type, "sim.py", ptr))
        update_state(sim_dir, sim_type, output_file, "sim")


def main():
    """main function, runs sim suite"""
    output_file = "internal/suite.csv"

    parser = argparse.ArgumentParser(description="Sims full sim suite")
    parser.add_argument(
        '--exclude', help='Exclude certain sim folders from the suite run',
        choices=config["sims"].keys(), default=["apl", "gear", "talent-builds", "stats", "weights"],
        nargs='+', required=False)
    parser.add_argument(
        '--fresh', help='restart suite from start', action='store_true')
    parser.add_argument(
        '--ptr', help='indicate if the sim should use ptr data.', action='store_true')
    args = parser.parse_args()

    if args.fresh or not os.path.exists(output_file):
        with open(output_file, 'w', encoding="utf8") as file:
            file.write('dir,type,sim,\n')
            file.close()

    ptr = ""
    if args.ptr:
        print("Running sims with PTR data...")
        ptr = "--ptr"

    for sim in config["sims"].keys():
        if sim in args.exclude:
            continue
        sim_dir = (f"{sim}/")
        run_suite(sim_dir, "composite", output_file, sim, ptr)
        run_suite(sim_dir, "dungeons", output_file, sim, ptr)


if __name__ == "__main__":
    main()