moth/package-puzzles

128 lines
4.5 KiB
Plaintext
Raw Normal View History

2016-10-18 09:36:24 -06:00
#!/usr/bin/env python3
import argparse
import binascii
2016-10-18 09:36:24 -06:00
import glob
import hashlib
import io
import json
import os
import shutil
import sys
2016-10-18 09:36:24 -06:00
import zipfile
2016-10-18 15:42:42 -06:00
import puzzles
2016-10-18 09:36:24 -06:00
TMPFILE = "%s.tmp"
2016-10-18 09:36:24 -06:00
def write_kv_pairs(ziphandle, filename, kv):
2016-10-18 15:42:42 -06:00
""" 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:
"""
2016-10-18 09:36:24 -06:00
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())
2016-10-18 15:42:42 -06:00
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Build a category package')
parser.add_argument('categorydirs', nargs='+', help='Directory of category source')
2016-10-18 15:42:42 -06:00
parser.add_argument('outdir', help='Output directory')
args = parser.parse_args()
2016-10-18 09:36:24 -06:00
for categorydir in args.categorydirs:
puzzles_dict = {}
2016-10-18 15:42:42 -06:00
secrets = {}
2016-10-18 09:36:24 -06:00
categoryname = os.path.basename(categorydir.strip(os.sep))
2016-10-18 09:36:24 -06:00
2016-10-18 15:42:42 -06:00
# create zipfile
zipfilename = os.path.join(args.outdir, "%s.zip" % categoryname)
if os.path.exists(zipfilename):
# open and gather some state
zf = zipfile.ZipFile(zipfilename, 'r')
try:
category_seed = zf.open(os.path.join(categoryname, "category_seed.txt")).read().strip()
except:
pass
zf.close()
zf = zipfile.ZipFile(TMPFILE % zipfilename, 'w')
if 'category_seed' not in locals():
category_seed = binascii.b2a_hex(os.urandom(20))
2016-10-18 15:42:42 -06:00
# read in category details (will be pflarr in future)
for categorypath in glob.glob(os.path.join(categorydir, "*", "puzzle.moth")):
points = categorypath.split(os.sep)[-2] # directory before '/puzzle.moth'
categorypath = os.path.dirname(categorypath)
2016-10-18 16:06:11 -06:00
print(categorypath)
try:
points = int(points)
except:
print("Failed to identify points on: %s" % categorypath, file=sys.stderr)
continue
2016-10-18 16:06:11 -06:00
puzzle = puzzles.Puzzle(category_seed, categorypath)
puzzles_dict[points] = puzzle
2016-10-18 15:42:42 -06:00
# build mapping, answers, and summary
mapping = {}
answers = {}
summary = {}
for points in sorted(puzzles_dict):
puzzle = puzzles_dict[points]
hashmap = hashlib.sha1(category_seed)
2016-10-18 15:42:42 -06:00
hashmap.update(str(points).encode('utf-8'))
mapping[points] = hashmap.hexdigest()
answers[points] = puzzle['answer']
if len(puzzle['summary']) > 0:
summary[points] = puzzle['summary']
2016-10-18 15:42:42 -06:00
# write mapping, answers, summary, category_seed
2016-10-18 15:42:42 -06:00
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)
zf.writestr(os.path.join(categoryname, "category_seed.txt"), category_seed)
2016-10-18 15:42:42 -06:00
# write out puzzles
for points in sorted(puzzles_dict):
puzzle = puzzles_dict[points]
2016-10-18 15:42:42 -06:00
puzzledir = os.path.join(categoryname, 'content', mapping[points])
2016-10-22 10:35:55 -06:00
puzzledict = {
'author': puzzle.author,
'hashes': puzzle.hashes(),
'files': [f.name for f in puzzle.files if f.visible],
'body': puzzle.html_body(),
}
secretsdict = {
'summary': puzzle.summary,
'answers': puzzle.answers,
}
2016-10-18 15:42:42 -06:00
# write associated files
assoc_files = []
for fobj in puzzle['files']:
if fobj.visible == True:
assoc_files.append(fobj.name)
zf.writestr(os.path.join(puzzledir, fobj.name), \
fobj.handle.read())
if len(assoc_files) > 0:
puzzlejson["associated_files"] = assoc_files
# non-optimal writing of file-like objects, but it works
zf.writestr(os.path.join(puzzledir, 'puzzle.json'), \
json.dumps(puzzle.publish()))
# clean up
zf.close()
shutil.move(TMPFILE % zipfilename, zipfilename)
2016-10-18 09:36:24 -06:00
#vim:py