55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
# Generate a draft pool for mtgadraft.tk or dr4ft.info
|
|
|
|
import cube_pool
|
|
import cubecobra_csv
|
|
|
|
ROUNDS = 3
|
|
SEATS = 8
|
|
PACKS = ROUNDS * SEATS
|
|
CARDS_PER_PACK = 15
|
|
OUTPUT_POOL_SIZE = PACKS * CARDS_PER_PACK
|
|
OCCASIONALS_SLOTS_PER_ROUND = [2, 1, 0]
|
|
CARDS_FROM_OCCASIONAL = sum(OCCASIONALS_SLOTS_PER_ROUND) * SEATS
|
|
CARDS_FROM_CORE = OUTPUT_POOL_SIZE - CARDS_FROM_OCCASIONAL
|
|
|
|
|
|
CUBE_NAME = 'TheElegantCube_fetched'
|
|
|
|
OUTPUT_FILENAME = 'cards_in_draft.txt'
|
|
|
|
|
|
def main():
|
|
csv_path = cubecobra_csv.request_cube_csv('TheElegantCube_fetched', 'elegant')
|
|
|
|
pools = cube_pool.load_pools(csv_path, {'core', 'occasional'})
|
|
|
|
core_pool = pools['core'].wide_sample(CARDS_FROM_CORE)
|
|
core_pool_names = [card.name for card in core_pool]
|
|
occasional_pool = pools['occasional'].wide_sample(CARDS_FROM_OCCASIONAL)
|
|
occasional_pool_names = [card.name for card in occasional_pool]
|
|
boosters = []
|
|
for r in range(ROUNDS):
|
|
for s in range(SEATS):
|
|
occasionals_per_booster = OCCASIONALS_SLOTS_PER_ROUND[r]
|
|
core_per_booster = CARDS_PER_PACK - occasionals_per_booster
|
|
|
|
booster = occasional_pool_names[-occasionals_per_booster:]
|
|
del occasional_pool_names[-occasionals_per_booster:]
|
|
assert len(booster) == occasionals_per_booster, str(booster)
|
|
|
|
booster += core_pool_names[-core_per_booster:]
|
|
del core_pool_names[-core_per_booster:]
|
|
assert len(booster) == CARDS_PER_PACK, str(booster)
|
|
|
|
boosters.append(booster)
|
|
|
|
with open(OUTPUT_FILENAME, 'wb') as f:
|
|
for booster in boosters:
|
|
booster_contents = '\n'.join(booster).encode('utf-8')
|
|
f.write(booster_contents)
|
|
f.write(b'\n\n')
|
|
print(f'Wrote {len(boosters)} boosters of {CARDS_PER_PACK} cards to {OUTPUT_FILENAME}')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|