Py-Hanabi/test.py

95 lines
2.7 KiB
Python
Raw Permalink Normal View History

import json
2023-05-14 19:11:42 +02:00
import alive_progress
2023-05-14 14:06:01 +02:00
import requests
from variants import Variant
from variants import Suit, variant_name
from site_api import *
2023-05-14 14:06:01 +02:00
from download_data import download_games, detailed_export_game
from check_game import check_game
from compress import link
2023-05-14 19:11:42 +02:00
from database.database import conn, cur
from database.init_database import init_database_tables, populate_static_tables
2023-06-24 17:25:25 +02:00
def find_double_dark_games():
cur.execute("SELECT variants.id, variants.name, count(suits.id) from variants "
"inner join variant_suits on variants.id = variant_suits.variant_id "
"left join suits on suits.id = variant_suits.suit_id "
"where suits.dark = (%s) "
"group by variants.id "
"order by count(suits.id), variants.id",
(True,)
)
cur2 = conn.cursor()
r = []
for (var_id, var_name, num_dark_suits) in cur.fetchall():
if num_dark_suits == 2:
cur2.execute("select count(*) from games where variant_id = (%s)", (var_id,))
games = cur2.fetchone()[0]
cur2.execute("select count(*) from seeds where variant_id = (%s)", (var_id, ))
r.append((var_name, games, cur2.fetchone()[0]))
l = sorted(r, key=lambda e: -e[1])
for (name, games, seeds) in l:
print("{}: {} games on {} seeds".format(name, games, seeds))
def test_suits():
suit = Suit.from_db(55)
print(suit.__dict__)
def test_variant():
var = Variant.from_db(926)
print(var.__dict__)
2023-05-14 14:06:01 +02:00
def check_missing_ids():
2023-05-14 19:11:42 +02:00
# start = 357849
# end = 358154
2023-05-14 14:06:01 +02:00
start = 358393
end = 358687
2023-05-14 19:11:42 +02:00
# broken_ids = [357913, 357914, 357915] # two of these are no variant
# not_supported_ids = [357925, 357957, 358081]
2023-05-14 14:06:01 +02:00
broken_ids = [358627, 358630, 358632]
not_supported_ids = [
]
for game_id in range(start, end):
if game_id in broken_ids or game_id in not_supported_ids:
continue
print(game_id)
detailed_export_game(game_id)
conn.commit()
2023-05-14 19:11:42 +02:00
def export_all_seeds():
cur.execute(
"SELECT id FROM variants ORDER BY ID"
)
var_ids = cur.fetchall()
for var in var_ids:
download_games(*var)
2023-05-14 14:06:01 +02:00
2023-05-14 19:11:42 +02:00
if __name__ == "__main__":
2023-06-24 17:25:25 +02:00
find_double_dark_games()
exit(0)
2023-05-14 19:11:42 +02:00
var_id = 964532
export_all_seeds()
exit(0)
2023-05-14 14:06:01 +02:00
# init_database_tables()
# populate_static_tables()
download_games(1)
print(variant_name(17888))
for page in range(0, 4):
r = api("variants/0?size=20&col[0]=0&page={}".format(page))
ids = []
for game in r['rows']:
ids.append(game['id'])
r['rows'] = None
print(json.dumps(r, indent=2))
print(ids)