start charting results

This commit is contained in:
2024-09-23 16:46:39 +01:00
parent 885a44be11
commit a9bd47b6bd
3 changed files with 117 additions and 2 deletions

64
multi-play.py Normal file
View File

@ -0,0 +1,64 @@
import subprocess
runs = []
with open("runs.csv", "r") as data:
try:
runs = data.readlines()
print("Successfully opened runs.csv")
except:
print("runs.csv doesn't exist")
number_of_runs = 100
def do_a_run(i):
print(f"Run {i}")
print("Playing Game")
command = "python main.py"
output = subprocess.check_output(command, shell=True, text=True)
output_lines = output.splitlines()
laps = 0
p1_sp = 0
p1_lead = 0
p2_sp = 0,
p2_lead = 0
winner = None
line_cache = None
for line in output_lines:
line = line.strip()
if line_cache == "p1":
p1_sp = int(line.split("SP: ")[1])
line_cache = None
if line_cache == "p2":
p2_sp = int(line.split("SP: ")[1])
line_cache = None
if line.startswith("Laps:"):
laps = int(line.split("Laps:")[1].strip())
if line.startswith("WINNER:"):
winner_line = line.split("WINNER: ")[1]
if "by" in winner_line:
winner_line = winner_line.split(" by ")
winner = winner_line[0]
if winner == "Player 1":
p1_lead = int(winner_line[1])
else:
p2_lead = int(winner_line[1])
else:
winner = "draw"
if line.startswith("--- Player 1"):
line_cache = "p1"
if line.startswith("--- Player 2"):
line_cache = "p2"
runs.append(f"{laps},{p1_sp},{p1_lead},{p2_sp},{p2_lead},{winner}")
if i < number_of_runs:
do_a_run(i + 1)
do_a_run(1)
with open("runs.csv", "w") as data:
data.write("\n".join(runs))