moth/package-puzzles

146 lines
4.0 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 uuid
import zipfile
messageChars = b'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
def djb2hash(buf):
h = 5381
for c in buf:
h = ((h * 33) + c) & 0xffffffff
return h
def write_kv_pairs(ziphandle, filename, kv):
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())
class Puzzle:
def __init__(self, stream):
self.message = bytes(random.choice(messageChars) for i in range(20))
self.fields = {}
self.answers = []
self.hashes = []
body = []
header = True
for line in stream:
if header:
line = line.strip()
if not line.strip():
header = False
continue
key, val = line.split(':', 1)
key = key.lower()
val = val.strip()
self._add_field(key, val)
else:
body.append(line)
self.body = ''.join(body)
def _add_field(self, key, val):
if key == 'answer':
h = djb2hash(val.encode('utf8'))
self.answers.append(val)
self.hashes.append(h)
else:
self.fields[key] = val
def publish(self):
obj = {
'author': self.fields['author'],
'hashes': self.hashes,
'body': markdown.markdown(self.body),
}
return obj
def secrets(self):
obj = {
'answers': self.answers,
'summary': self.fields['summary'],
}
return obj
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 = Puzzle(open(puzzlePath))
puzzles[points] = puzzle
# build mapping, answers, and summary
mapping = {}
answers = {}
summary = {}
for points in sorted(puzzles):
puzzle = puzzles[points]
mapping[points] = str(uuid.uuid4()) # random uuid
answers[points] = puzzle.answers
summary[points] = puzzle.fields['summary']
# write mapping, answers, and summary
write_kv_pairs(zf, os.path.join(categoryname, 'mapping.txt'), mapping)
write_kv_pairs(zf, os.path.join(categoryname, 'answers.txt'), answers)
write_kv_pairs(zf, os.path.join(categoryname, 'summary.txt'), summary)
print(mapping)
print(answers)
print(summary)
# write out puzzles
for points in sorted(puzzles):
puzzle = puzzles[points]
#vim:py