start charting results
This commit is contained in:
64
multi-play.py
Normal file
64
multi-play.py
Normal 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))
|
Reference in New Issue
Block a user