mirror of https://github.com/dirtbags/moth.git
103 lines
3.4 KiB
Python
Executable File
103 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import base64
|
|
import glob
|
|
import hashlib
|
|
import hmac
|
|
import io
|
|
import json
|
|
import os
|
|
import markdown
|
|
import random
|
|
import zipfile
|
|
|
|
import puzzles
|
|
|
|
def write_kv_pairs(ziphandle, filename, kv):
|
|
""" Write out a sorted map to file
|
|
:param ziphandle: a zipfile object
|
|
:param filename: The filename to write within the zipfile object
|
|
:param kv: the map to write out
|
|
:return:
|
|
"""
|
|
filehandle = io.StringIO()
|
|
for key in sorted(kv.keys()):
|
|
if type(kv[key]) == type([]):
|
|
for val in kv[key]:
|
|
filehandle.write("%s: %s%s" % (key, val, os.linesep))
|
|
else:
|
|
filehandle.write("%s: %s%s" % (key, kv[key], os.linesep))
|
|
filehandle.seek(0)
|
|
ziphandle.writestr(filename, filehandle.read())
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Build a puzzle category')
|
|
parser.add_argument('seed', help='contest seed')
|
|
parser.add_argument('puzzledir', nargs='+', help='Directory of puzzle source')
|
|
parser.add_argument('outdir', help='Output directory')
|
|
args = parser.parse_args()
|
|
|
|
|
|
for puzzledir in args.puzzledir:
|
|
puzzles = {}
|
|
secrets = {}
|
|
|
|
|
|
categoryname = os.path.basename(puzzledir.strip(os.sep))
|
|
|
|
# build puzzle seed
|
|
puzzle_seed = hashlib.new('sha1')
|
|
puzzle_seed.update(categoryname.encode('utf-8'))
|
|
puzzle_seed.update(args.seed.encode('utf-8'))
|
|
puzzle_seed = puzzle_seed.hexdigest()
|
|
|
|
# create zipfile
|
|
zipfilename = os.path.join(args.outdir, "%s.zip" % categoryname)
|
|
if os.path.exists(zipfilename):
|
|
# assume blank state for now
|
|
pass
|
|
# # open and gather some state
|
|
# zf = zipfile.ZipFile(zipfilename, 'a')
|
|
else:
|
|
# create and read in state
|
|
zf = zipfile.ZipFile(zipfilename, 'w')
|
|
|
|
# read in puzzle details (will be pflarr in future)
|
|
for puzzlePath in glob.glob(os.path.join(puzzledir, "*.moth")):
|
|
filename = os.path.basename(puzzlePath)
|
|
points, ext = os.path.splitext(filename)
|
|
points = int(points)
|
|
puzzle = puzzles.Puzzle(open(puzzlePath))
|
|
puzzles[points] = puzzle
|
|
|
|
# build mapping, answers, and summary
|
|
mapping = {}
|
|
answers = {}
|
|
summary = {}
|
|
for points in sorted(puzzles):
|
|
puzzle = puzzles[points]
|
|
hashmap = hashlib.sha1(args.seed.encode('utf-8'))
|
|
hashmap.update(categoryname.encode('utf-8'))
|
|
hashmap.update(str(points).encode('utf-8'))
|
|
mapping[points] = hashmap.hexdigest()
|
|
answers[points] = puzzle.answers
|
|
summary[points] = puzzle.fields['summary']
|
|
|
|
# write mapping, answers, and summary
|
|
write_kv_pairs(zf, os.path.join(categoryname, 'map.txt'), mapping)
|
|
write_kv_pairs(zf, os.path.join(categoryname, 'answers.txt'), answers)
|
|
write_kv_pairs(zf, os.path.join(categoryname, 'summary.txt'), summary)
|
|
|
|
# write out puzzles
|
|
for points in sorted(puzzles):
|
|
puzzle = puzzles[points]
|
|
puzzledir = os.path.join(categoryname, 'content', mapping[points])
|
|
# build/write json
|
|
ziphandle.writestr(filename, filehandle.read())
|
|
zf.writestr(os.path.join(puzzledir, 'puzzle.json'), \
|
|
json.dumps(puzzle.publish()))
|
|
# write associated files
|
|
|
|
#vim:py
|