121 lines
4.6 KiB
Python
121 lines
4.6 KiB
Python
import json
|
|
from time import sleep
|
|
from site_api import get, api, replay
|
|
from sat import COLORS, solve
|
|
from database import Game, store, load, commit, conn
|
|
from download_data import export_game
|
|
from variants import num_suits, VARIANTS
|
|
from alive_progress import alive_bar
|
|
|
|
|
|
def update_seeds_db():
|
|
cur2 = conn.cursor()
|
|
with conn.cursor() as cur:
|
|
cur.execute("SELECT num_players, seed, variant_id from games;")
|
|
for (num_players, seed, variant_id) in cur:
|
|
cur2.execute("SELECT COUNT(*) from seeds WHERE seed = (%s);", (seed,))
|
|
if cur2.fetchone()[0] == 0:
|
|
print("new seed {}".format(seed))
|
|
cur2.execute("INSERT INTO seeds"
|
|
"(seed, num_players, variant_id)"
|
|
"VALUES"
|
|
"(%s, %s, %s)",
|
|
(seed, num_players, variant_id)
|
|
)
|
|
conn.commit()
|
|
else:
|
|
print("seed {} already found in DB".format(seed))
|
|
|
|
|
|
def get_decks_of_seeds():
|
|
cur = conn.cursor()
|
|
cur2 = conn.cursor()
|
|
cur.execute("SELECT seed FROM seeds WHERE deck is NULL")
|
|
for (seed,) in cur:
|
|
cur2.execute("SELECT id FROM games WHERE seed = (%s)", (seed,))
|
|
(game_id,) = cur2.fetchone()
|
|
print("Exporting game {} for seed {}.".format(game_id, seed))
|
|
export_game(game_id)
|
|
conn.commit()
|
|
|
|
|
|
def update_trivially_feasible_games():
|
|
cur = conn.cursor()
|
|
for var in VARIANTS:
|
|
cur.execute("SELECT COUNT(*) FROM seeds WHERE variant_id = (%s) AND feasible is null", (var['id'],))
|
|
num_seeds = cur.fetchone()[0]
|
|
cur.execute("SELECT seed, id FROM games WHERE score = (%s) AND variant_id = (%s) ORDER BY seed;", (5 * len(var['suits']), var['id']))
|
|
res = cur.fetchall()
|
|
print('Checking variant {} (id {}), found {} seeds with {} max-score games to check...'.format(var['name'], var['id'], num_seeds, len(res)))
|
|
cur_seed = None
|
|
seed_finished = False
|
|
with alive_bar(num_seeds) as bar:
|
|
for (seed, game_id) in res:
|
|
if seed_finished and cur_seed == seed:
|
|
print('skipping further game of seed {}'.format(seed))
|
|
continue
|
|
if cur_seed != seed:
|
|
bar()
|
|
cur_seed = seed
|
|
seed_finished = False
|
|
cur.execute("SELECT deck_plays, one_extra_card, one_less_card, all_or_nothing FROM games WHERE id = (%s)", (game_id,))
|
|
cheat_options = cur.fetchall()[0]
|
|
valid = None
|
|
if None in cheat_options:
|
|
print('Game {} not found in database, exporting...'.format(game_id))
|
|
succ, valid = export_game(game_id)
|
|
if not succ:
|
|
print('Error exporting game {}.'.format(game_id))
|
|
continue
|
|
else:
|
|
valid = not any(cheat_options)
|
|
print('Game {} already in database, valid: {}'.format(game_id, valid))
|
|
if valid:
|
|
print('Seed {:9} (variant {} / {}) found to be feasible via game {:6}'.format(seed, var['id'], var['name'], game_id))
|
|
cur.execute("UPDATE seeds SET feasible = (%s) WHERE seed = (%s)", (True, seed))
|
|
conn.commit()
|
|
seed_finished = True
|
|
else:
|
|
print('Cheaty game found')
|
|
return
|
|
for (seed,) in cur:
|
|
cur2.execute("UPDATE seeds SET feasible=TRUE WHERE seed=(%s)", (seed,))
|
|
print("Seed {} found to be feasible")
|
|
conn.commit()
|
|
|
|
update_trivially_feasible_games()
|
|
exit(0)
|
|
|
|
|
|
|
|
def solve_unknown_seeds():
|
|
for var in VARIANTS:
|
|
cur.execute("SELECT deck FROM games WHERE seed = (%s)", (a,))
|
|
|
|
|
|
|
|
print('looked at {} games'.format(num_games))
|
|
for i in range(2,7):
|
|
print("Found {} seeds in {}-player in database".format(len(seeds[i]), i))
|
|
|
|
hard_seeds = []
|
|
for seed in seeds[3]:
|
|
if not known_solvable(seed):
|
|
# print("seed {} has no solve in online database".format(seed))
|
|
hard_seeds.append(seed)
|
|
print("Found {} seeds with no solve in database, attacking each with SAT solver"
|
|
.format(len(hard_seeds)))
|
|
|
|
for seed in hard_seeds:
|
|
r = replay(seed)
|
|
if not r:
|
|
continue
|
|
s, sol = solvable(r)
|
|
if s:
|
|
print("Seed {} was found to be solvable".format(seed))
|
|
# print(sol)
|
|
else:
|
|
print("==============================")
|
|
print("Found non-solvable seed {}", seed)
|
|
print("==============================")
|