asura/main.py

40 lines
951 B
Python
Raw Normal View History

2018-12-31 00:02:21 +00:00
#!/usr/bin/env python3
from sys import argv, exit
from time import time
import os.path as op
from asura import generate, write_rom, write_spoilers
from asura.data import spell_map
if len(argv) < 2 or len(argv) > 4:
print('Usage: {} <filename> [seed [forbidden,spells,...]]'.format(argv[0]))
exit(1)
if len(argv) == 2:
seed = str(time()).split('.')[0]
if len(argv) >= 3:
seed = argv[2]
if len(argv) == 4:
spells = argv[3].split(',')
forbidden = [spell_map[i] for i in spells]
else:
forbidden = []
filename = argv[1]
namebase, ext = op.splitext(filename)
ext = ext[1:]
output_rom = '.'.join([namebase, seed, ext])
output_txt = '.'.join([namebase, seed, 'txt'])
mode = 'snes' if ext != 'gba' else 'gba'
espers = generate(seed, mode, forbidden)
print('Seed: {}\n'.format(seed))
print('Writing ROM...')
write_rom(filename, output_rom, mode, espers)
print('Writing spoilers...')
write_spoilers(output_txt, espers, seed)