120 lines
4.5 KiB
Python
Executable file
120 lines
4.5 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import sys
|
|
import os
|
|
|
|
libdir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'lib')
|
|
sys.path.append(libdir)
|
|
import re
|
|
import random
|
|
import utils
|
|
import jdecode
|
|
import cardlib
|
|
|
|
def main(fname, oname = None, verbose = True, encoding = 'std',
|
|
nolinetrans = False, randomize = False, nolabel = False, stable = False):
|
|
fmt_ordered = cardlib.fmt_ordered_default
|
|
fmt_labeled = None if nolabel else cardlib.fmt_labeled_default
|
|
fieldsep = utils.fieldsep
|
|
line_transformations = not nolinetrans
|
|
randomize_fields = False
|
|
randomize_mana = randomize
|
|
initial_sep = True
|
|
final_sep = True
|
|
|
|
# set the properties of the encoding
|
|
|
|
if encoding in ['std']:
|
|
pass
|
|
elif encoding in ['named']:
|
|
fmt_ordered = cardlib.fmt_ordered_named
|
|
elif encoding in ['noname']:
|
|
fmt_ordered = cardlib.fmt_ordered_noname
|
|
elif encoding in ['rfields']:
|
|
randomize_fields = True
|
|
final_sep = False
|
|
elif encoding in ['old']:
|
|
fmt_ordered = cardlib.fmt_ordered_old
|
|
elif encoding in ['norarity']:
|
|
fmt_ordered = cardlib.fmt_ordered_norarity
|
|
elif encoding in ['vec']:
|
|
pass
|
|
elif encoding in ['custom']:
|
|
## put custom format decisions here ##########################
|
|
|
|
## end of custom format ######################################
|
|
pass
|
|
else:
|
|
raise ValueError('encode.py: unknown encoding: ' + encoding)
|
|
|
|
if verbose:
|
|
print('Preparing to encode:')
|
|
print(' Using encoding ' + repr(encoding))
|
|
if stable:
|
|
print(' NOT randomizing order of cards.')
|
|
if randomize_mana:
|
|
print(' Randomizing order of symobls in manacosts.')
|
|
if not fmt_labeled:
|
|
print(' NOT labeling fields for this run (may be harder to decode).')
|
|
if not line_transformations:
|
|
print(' NOT using line reordering transformations')
|
|
|
|
cards = jdecode.mtg_open_file(fname, verbose=verbose, linetrans=line_transformations)
|
|
|
|
# This should give a random but consistent ordering, to make comparing changes
|
|
# between the output of different versions easier.
|
|
if not stable:
|
|
random.seed(1371367)
|
|
random.shuffle(cards)
|
|
|
|
def writecards(writer):
|
|
for card in cards:
|
|
if encoding in ['vec']:
|
|
writer.write(card.vectorize() + '\n\n')
|
|
else:
|
|
writer.write(card.encode(fmt_ordered = fmt_ordered,
|
|
fmt_labeled = fmt_labeled,
|
|
fieldsep = fieldsep,
|
|
randomize_fields = randomize_fields,
|
|
randomize_mana = randomize_mana,
|
|
initial_sep = initial_sep,
|
|
final_sep = final_sep)
|
|
+ utils.cardsep)
|
|
|
|
if oname:
|
|
if verbose:
|
|
print('Writing output to: ' + oname)
|
|
with open(oname, 'w') as ofile:
|
|
writecards(ofile)
|
|
else:
|
|
writecards(sys.stdout)
|
|
sys.stdout.flush()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('infile',
|
|
help='encoded card file or json corpus to encode')
|
|
parser.add_argument('outfile', nargs='?', default=None,
|
|
help='output file, defaults to stdout')
|
|
parser.add_argument('-e', '--encoding', default='std', choices=utils.formats,
|
|
#help='{' + ','.join(formats) + '}',
|
|
help='encoding format to use',
|
|
)
|
|
parser.add_argument('-r', '--randomize', action='store_true',
|
|
help='randomize the order of symbols in mana costs')
|
|
parser.add_argument('--nolinetrans', action='store_true',
|
|
help="don't reorder lines of card text")
|
|
parser.add_argument('--nolabel', action='store_true',
|
|
help="don't label fields")
|
|
parser.add_argument('-s', '--stable', action='store_true',
|
|
help="don't randomize the order of the cards")
|
|
parser.add_argument('-v', '--verbose', action='store_true',
|
|
help='verbose output')
|
|
|
|
args = parser.parse_args()
|
|
main(args.infile, args.outfile, verbose = args.verbose, encoding = args.encoding,
|
|
nolinetrans = args.nolinetrans, randomize = args.randomize, nolabel = args.nolabel,
|
|
stable = args.stable)
|
|
exit(0)
|