112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
import re
|
|
import codecs
|
|
import sys
|
|
|
|
# returns back a dictionary mapping the names of classes of cards
|
|
# to lists of cards in those classes
|
|
def sortcards(cards):
|
|
classes = {
|
|
'multicards' : [],
|
|
|
|
'X cards' : [],
|
|
'counter cards' : [],
|
|
'choice cards' : [],
|
|
'equipment' : [],
|
|
'levelers' : [],
|
|
'legendary' : [],
|
|
|
|
'planeswalkers' : [],
|
|
'lands' : [],
|
|
'instants' : [],
|
|
'sorceries' : [],
|
|
'enchantments' : [],
|
|
'noncreature artifacts' : [],
|
|
'creatures' : [],
|
|
'other' : [],
|
|
}
|
|
|
|
for card in cards:
|
|
# special classes
|
|
if '|\n|' in card:
|
|
classes['multicards'] += [card]
|
|
continue
|
|
|
|
# inclusive classes
|
|
if 'X' in card:
|
|
classes['X cards'] += [card]
|
|
if '#' in card:
|
|
classes['counter cards'] += [card]
|
|
if 'choose one ~' in card or 'choose two ~' in card or '=' in card:
|
|
classes['choice cards'] += [card]
|
|
if '|equipment|' in card or 'equip {' in card:
|
|
classes['equipment'] += [card]
|
|
if 'level up' in card or 'level &' in card:
|
|
classes['levelers'] += [card]
|
|
if '|legendary|' in card:
|
|
classes['legendary'] += [card]
|
|
|
|
# exclusive classes
|
|
if '|planeswalker|' in card:
|
|
classes['planeswalkers'] += [card]
|
|
elif '|land|' in card:
|
|
classes['lands'] += [card]
|
|
elif '|instant|' in card:
|
|
classes['instants'] += [card]
|
|
elif '|sorcery|' in card:
|
|
classes['sorceries'] += [card]
|
|
elif '|enchantment|' in card:
|
|
classes['enchantments'] += [card]
|
|
elif '|artifact|' in card:
|
|
classes['noncreature artifacts'] += [card]
|
|
elif '|creature|' in card or 'artifact creature' in card:
|
|
classes['creatures'] += [card]
|
|
else:
|
|
classes['other'] += [card]
|
|
|
|
return classes
|
|
|
|
def main(fname, oname = None, verbose = True):
|
|
if verbose:
|
|
print 'Opening encoded card file: ' + fname
|
|
|
|
f = open(fname, 'r')
|
|
text = f.read()
|
|
f.close()
|
|
|
|
# we get rid of the first and last because they are probably partial
|
|
cards = text.split('\n\n')[1:-1]
|
|
classes = sortcards(cards)
|
|
|
|
if not oname == None:
|
|
if verbose:
|
|
print 'Writing output to: ' + oname
|
|
ofile = codecs.open(oname, 'w', 'utf-8')
|
|
|
|
for cardclass in classes:
|
|
print cardclass + ': ' + str(len(classes[cardclass]))
|
|
|
|
if oname == None:
|
|
outputter = sys.stdout
|
|
else:
|
|
outputter = ofile
|
|
|
|
for cardclass in classes:
|
|
outputter.write('[spoiler=' + cardclass + ']\n')
|
|
for card in classes[cardclass]:
|
|
outputter.write(card + '\n\n')
|
|
outputter.write('[/spoiler]')
|
|
|
|
if not oname == None:
|
|
ofile.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
if len(sys.argv) == 2:
|
|
main(sys.argv[1])
|
|
elif len(sys.argv) == 3:
|
|
main(sys.argv[1], oname = sys.argv[2])
|
|
else:
|
|
print 'Usage: ' + sys.argv[0] + ' ' + '<encoded file> [output filename]'
|
|
exit(1)
|
|
|