Hanabi/streak-hunting-spreadsheet
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

start adjustment to new cli interface: better parsing and default values

Browse source
This commit is contained in:
Maximilian Keßler 2023-11-11 23:09:13 +01:00
parent f0263ab250
commit eb0699e775
Signed by: max
GPG key ID: BCC5A619923C0BA5
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
endgames.py
View file

@ -15,23 +15,26 @@ with open(DATA_FILE, 'r') as f:
def analyze_game(game_id: int):
max_draw_pile_size = 15
probabilities = {}
raw_output = None
try:
result = subprocess.run(['./endgame-analyzer', '-g', str(game_id), '-d', '16', '--interactive', '0', '--quiet', '-r'], stdout=subprocess.PIPE, timeout=30)
result = subprocess.run(['./endgame-analyzer', '-g', str(game_id), '-d', str(max_draw_pile_size), '--interactive', '0', '--quiet', '-r'], stdout=subprocess.PIPE, timeout=10)
raw_output = result.stdout
except subprocess.TimeoutExpired as time_err:
raw_output = time_err.stdout
output = raw_output.decode('utf-8')
print(output)
# Check if the game was just over before reaching the specified draw pile size
if re.match("This given draw pile size \(.*\) cannot be obtained with the specified replay.", output):
for i in range(0,16):
probabilities[str(i)] = 0
return probabilities
if re.search(r'The given draw pile size \({}\) cannot be obtained with the specified replay.'.format(max_draw_pile_size), output):
print('detected empty output')
return {str(size): 0 for size in range(1, max_draw_pile_size + 1)}
# Now, parse all results that we obtained (unclear how many depending on whether we ran into the timeout)
for m in re.finditer('Probability with ([0-9]+) cards left in the deck: .*/.* ~ ([0-9.]+)', output):
x = 'Probability with 13 cards left in deck: 5855/6237 ~ 93.875%'
for m in re.finditer('Probability with ([0-9]+) cards left in deck: .*/.* ~ ([0-9.]+)', output):
probabilities[str(m.group(1))] = m.group(2)
return probabilities
@ -64,10 +67,12 @@ def full_analyze_game_cached(game_id: int):
return result
def analyze_game_cached(game_id: int):
cached = DATA['normal'].get(str(game_id), None)
cached = DATA.get('normal', {}).get(str(game_id), None)
if cached is not None:
return cached
result = analyze_game(game_id)
if 'normal' not in DATA.keys():
DATA['normal'] = {}
DATA['normal'][game_id] = result
save_cache()
return result