305 lines
12 KiB
Python
Executable file
305 lines
12 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import sys
|
|
import os
|
|
import zipfile
|
|
import shutil
|
|
|
|
libdir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'lib')
|
|
sys.path.append(libdir)
|
|
import utils
|
|
import jdecode
|
|
import cardlib
|
|
from cbow import CBOW
|
|
from namediff import Namediff
|
|
|
|
def main(fname, oname = None, verbose = True, encoding = 'std',
|
|
gatherer = False, for_forum = False, for_mse = False, for_json = False,
|
|
creativity = False, vdump = False, for_html = False):
|
|
|
|
# there is a sane thing to do here (namely, produce both at the same time)
|
|
# but we don't support it yet.
|
|
if for_mse and for_html:
|
|
print 'ERROR - decode.py - incompatible formats "mse" and "html"'
|
|
return
|
|
|
|
fmt_ordered = cardlib.fmt_ordered_default
|
|
|
|
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']:
|
|
pass
|
|
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)
|
|
|
|
cards = jdecode.mtg_open_file(fname, verbose=verbose, fmt_ordered=fmt_ordered)
|
|
|
|
if creativity:
|
|
namediff = Namediff()
|
|
cbow = CBOW()
|
|
if verbose:
|
|
print 'Computing nearest names...'
|
|
nearest_names = namediff.nearest_par(map(lambda c: c.name, cards), n=3)
|
|
if verbose:
|
|
print 'Computing nearest cards...'
|
|
nearest_cards = cbow.nearest_par(cards)
|
|
for i in range(0, len(cards)):
|
|
cards[i].nearest_names = nearest_names[i]
|
|
cards[i].nearest_cards = nearest_cards[i]
|
|
if verbose:
|
|
print '...Done.'
|
|
|
|
def hoverimg(cardname, dist, nd):
|
|
truename = nd.names[cardname]
|
|
code = nd.codes[cardname]
|
|
namestr = ''
|
|
if for_html:
|
|
if code:
|
|
namestr = ('<div class="hover_img"><a href="#">' + truename
|
|
+ '<span><img style="background: url(http://magiccards.info/scans/en/' + code
|
|
+ ');" alt=""/></span></a>' + ': ' + str(dist) + '\n</div>\n')
|
|
else:
|
|
namestr = '<div>' + truename + ': ' + str(dist) + '</div>'
|
|
elif for_forum:
|
|
namestr = '[card]' + truename + '[/card]' + ': ' + str(dist) + '\n'
|
|
else:
|
|
namestr = truename + ': ' + str(dist) + '\n'
|
|
return namestr
|
|
|
|
def writecards(writer):
|
|
if for_mse:
|
|
# have to prepend a massive chunk of formatting info
|
|
writer.write(utils.mse_prepend)
|
|
|
|
if for_html:
|
|
# have to preapend html info
|
|
writer.write(utils.html_prepend)
|
|
# seperate the write function to allow for writing smaller chunks of cards at a time
|
|
segments = sort_colors(cards)
|
|
for i in range(len(segments)):
|
|
# sort color by CMC
|
|
segments[i] = sort_type(segments[i])
|
|
# this allows card boxes to be colored for each color
|
|
# for coloring of each box seperately cardlib.Card.format() must change non-minimaly
|
|
writer.write('<div id="' + utils.segment_ids[i] + '">')
|
|
writehtml(writer, segments[i])
|
|
writer.write("</div><hr>")
|
|
# closing the html file
|
|
writer.write(utils.html_append)
|
|
return #break out of the write cards funcrion to avoid writing cards twice
|
|
|
|
|
|
for card in cards:
|
|
if for_json:
|
|
writer.write(card.to_json())
|
|
elif for_mse:
|
|
writer.write(card.to_mse().encode('utf-8'))
|
|
fstring = ''
|
|
if card.json:
|
|
fstring += 'JSON:\n' + card.json + '\n'
|
|
if card.raw:
|
|
fstring += 'raw:\n' + card.raw + '\n'
|
|
fstring += '\n'
|
|
fstring += card.format(gatherer = gatherer, for_forum = for_forum,
|
|
vdump = vdump) + '\n'
|
|
fstring = fstring.replace('<', '(').replace('>', ')')
|
|
writer.write(('\n' + fstring[:-1]).replace('\n', '\n\t\t'))
|
|
else:
|
|
fstring = card.format(gatherer = gatherer, for_forum = for_forum,
|
|
vdump = vdump, for_html = for_html)
|
|
writer.write((fstring + '\n').encode('utf-8'))
|
|
|
|
if creativity:
|
|
cstring = '~~ closest cards ~~\n'
|
|
nearest = card.nearest_cards
|
|
for dist, cardname in nearest:
|
|
cstring += hoverimg(cardname, dist, namediff)
|
|
cstring += '~~ closest names ~~\n'
|
|
nearest = card.nearest_names
|
|
for dist, cardname in nearest:
|
|
cstring += hoverimg(cardname, dist, namediff)
|
|
if for_mse:
|
|
cstring = ('\n\n' + cstring[:-1]).replace('\n', '\n\t\t')
|
|
writer.write(cstring.encode('utf-8'))
|
|
|
|
writer.write('\n'.encode('utf-8'))
|
|
|
|
if for_mse:
|
|
# more formatting info
|
|
writer.write('version control:\n\ttype: none\napprentice code: ')
|
|
|
|
|
|
def writehtml(writer, card_set):
|
|
for card in card_set:
|
|
fstring = card.format(gatherer = gatherer, for_forum = True,
|
|
vdump = vdump, for_html = for_html)
|
|
if creativity:
|
|
fstring = fstring[:-6] # chop off the closing </div> to stick stuff in
|
|
writer.write((fstring + '\n').encode('utf-8'))
|
|
|
|
if creativity:
|
|
cstring = '~~ closest cards ~~\n<br>\n'
|
|
nearest = card.nearest_cards
|
|
for dist, cardname in nearest:
|
|
cstring += hoverimg(cardname, dist, namediff)
|
|
cstring += "<br>\n"
|
|
cstring += '~~ closest names ~~\n<br>\n'
|
|
nearest = card.nearest_names
|
|
for dist, cardname in nearest:
|
|
cstring += hoverimg(cardname, dist, namediff)
|
|
cstring = '<hr><div>' + cstring + '</div>\n</div>'
|
|
writer.write(cstring.encode('utf-8'))
|
|
|
|
writer.write('\n'.encode('utf-8'))
|
|
|
|
# Sorting by colors
|
|
def sort_colors(card_set):
|
|
# Initialize sections
|
|
red_cards = []
|
|
blue_cards = []
|
|
green_cards = []
|
|
black_cards = []
|
|
white_cards = []
|
|
multi_cards = []
|
|
colorless_cards = []
|
|
lands = []
|
|
for card in card_set:
|
|
if len(card.get_colors())>1:
|
|
multi_cards += [card]
|
|
continue
|
|
if 'R' in card.get_colors():
|
|
red_cards += [card]
|
|
continue
|
|
elif 'U' in card.get_colors():
|
|
blue_cards += [card]
|
|
continue
|
|
elif 'B' in card.get_colors():
|
|
black_cards += [card]
|
|
continue
|
|
elif 'G' in card.get_colors():
|
|
green_cards += [card]
|
|
continue
|
|
elif 'W' in card.get_colors():
|
|
white_cards += [card]
|
|
continue
|
|
else:
|
|
if "land" in card.get_types():
|
|
lands += [card]
|
|
continue
|
|
colorless_cards += [card]
|
|
return[white_cards, blue_cards, black_cards, red_cards, green_cards, multi_cards, colorless_cards, lands]
|
|
|
|
def sort_type(card_set):
|
|
sorting = ["creature", "enchantment", "instant", "sorcery", "artifact", "planeswalker"]
|
|
sorted_cards = [[],[],[],[],[],[],[]]
|
|
sorted_set = []
|
|
for card in card_set:
|
|
types = card.get_types()
|
|
for i in range(len(sorting)):
|
|
if sorting[i] in types:
|
|
sorted_cards[i] += [card]
|
|
break
|
|
else:
|
|
sorted_cards[6] += [card]
|
|
for value in sorted_cards:
|
|
for card in value:
|
|
sorted_set += [card]
|
|
return sorted_set
|
|
|
|
|
|
|
|
def sort_cmc(card_set):
|
|
sorted_cards = []
|
|
sorted_set = []
|
|
for card in card_set:
|
|
# make sure there is an empty set for each CMC
|
|
while len(sorted_cards)-1 < card.get_cmc():
|
|
sorted_cards += [[]]
|
|
# add card to correct set of CMC values
|
|
sorted_cards[card.get_cmc()] += [card]
|
|
# combine each set of CMC valued cards together
|
|
for value in sorted_cards:
|
|
for card in value:
|
|
sorted_set += [card]
|
|
return sorted_set
|
|
|
|
|
|
if oname:
|
|
if for_html:
|
|
print oname
|
|
# if ('.html' != oname[-])
|
|
# oname += '.html'
|
|
if verbose:
|
|
print 'Writing output to: ' + oname
|
|
with open(oname, 'w') as ofile:
|
|
writecards(ofile)
|
|
if for_mse:
|
|
# Copy whatever output file is produced, name the copy 'set' (yes, no extension).
|
|
if os.path.isfile('set'):
|
|
print 'ERROR: tried to overwrite existing file "set" - aborting.'
|
|
return
|
|
shutil.copyfile(oname, 'set')
|
|
# Use the freaky mse extension instead of zip.
|
|
with zipfile.ZipFile(oname+'.mse-set', mode='w') as zf:
|
|
try:
|
|
# Zip up the set file into oname.mse-set.
|
|
zf.write('set')
|
|
finally:
|
|
if verbose:
|
|
print 'Made an MSE set file called ' + oname + '.mse-set.'
|
|
# The set file is useless outside the .mse-set, delete it.
|
|
os.remove('set')
|
|
else:
|
|
writecards(sys.stdout)
|
|
sys.stdout.flush()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('infile', #nargs='?'. default=None,
|
|
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('-g', '--gatherer', action='store_true',
|
|
help='emulate Gatherer visual spoiler')
|
|
parser.add_argument('-f', '--forum', action='store_true',
|
|
help='use pretty mana encoding for mtgsalvation forum')
|
|
parser.add_argument('-c', '--creativity', action='store_true',
|
|
help='use CBOW fuzzy matching to check creativity of cards')
|
|
parser.add_argument('-d', '--dump', action='store_true',
|
|
help='dump out lots of information about invalid cards')
|
|
parser.add_argument('-v', '--verbose', action='store_true',
|
|
help='verbose output')
|
|
parser.add_argument('-mse', '--mse', action='store_true',
|
|
help='use Magic Set Editor 2 encoding; will output as .mse-set file')
|
|
parser.add_argument('-j', '--json', action='store_true',
|
|
help='use JSON encoding; will output as .json file')
|
|
parser.add_argument('-html', '--html', action='store_true', help='create a .html file with pretty forum formatting')
|
|
|
|
args = parser.parse_args()
|
|
|
|
main(args.infile, args.outfile, verbose = args.verbose, encoding = args.encoding,
|
|
gatherer = args.gatherer, for_forum = args.forum, for_mse = args.mse, for_json = args.json,
|
|
creativity = args.creativity, vdump = args.dump, for_html = args.html)
|
|
|
|
exit(0)
|