import domdiv
import domdiv.main
from zipfile import ZipFile, ZIP_DEFLATED

prefix = 'generated/sumpfork_dominion_tabs_'
postfix = 'v' + domdiv.__version__ + '.pdf'


def doit(args, main):
    args = args + ' --outfile ' + prefix + main + postfix
    args = args.split()
    fname = args[-1]
    print args
    print ':::Generating ' + fname
    options = domdiv.main.parse_opts(args)
    options = domdiv.main.clean_opts(options)
    domdiv.main.generate(options)
    return fname


argsets = [
    ('', ''), ('--orientation=vertical', 'vertical_'),
    ('--papersize=A4', 'A4_'),
    ('--papersize=A4 --orientation=vertical', 'vertical_A4_'),
    ('--size=sleeved', 'sleeved_'),
    ('--size=sleeved --orientation=vertical', 'vertical_sleeved_')
]
additional = ['--expansion_dividers']

fnames = [doit(args[0] + ' ' + ' '.join(additional), args[1])
          for args in argsets]
print fnames

zip = ZipFile('generated/sumpfork_dominion_tabs_v' + domdiv.__version__ + '.zip', 'w',
              ZIP_DEFLATED)
for f in fnames:
    zip.write(f)
zip.close()