mirror of https://github.com/dirtbags/moth.git
some updates to package-puzzles
This commit is contained in:
slackish
parent
6470cbd9d6
commit
0b335066a4
|
|
@ -12,15 +12,15 @@ import markdown
|
|||
import random
|
||||
import zipfile
|
||||
|
||||
messageChars = b'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
||||
|
||||
def djb2hash(buf):
|
||||
h = 5381
|
||||
for c in buf:
|
||||
h = ((h * 33) + c) & 0xffffffff
|
||||
return h
|
||||
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([]):
|
||||
|
|
@ -31,61 +31,15 @@ def write_kv_pairs(ziphandle, filename, kv):
|
|||
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
|
||||
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()
|
||||
|
||||
|
||||
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:
|
||||
for puzzledir in args.puzzledir:
|
||||
puzzles = {}
|
||||
secrets = {}
|
||||
|
||||
|
|
@ -101,10 +55,10 @@ for puzzledir in args.puzzledir:
|
|||
# create zipfile
|
||||
zipfilename = os.path.join(args.outdir, "%s.zip" % categoryname)
|
||||
if os.path.exists(zipfilename):
|
||||
# assume blank state for now
|
||||
# assume blank state for now
|
||||
pass
|
||||
# # open and gather some state
|
||||
# zf = zipfile.ZipFile(zipfilename, 'a')
|
||||
# # open and gather some state
|
||||
# zf = zipfile.ZipFile(zipfilename, 'a')
|
||||
else:
|
||||
# create and read in state
|
||||
zf = zipfile.ZipFile(zipfilename, 'w')
|
||||
|
|
@ -114,7 +68,7 @@ for puzzledir in args.puzzledir:
|
|||
filename = os.path.basename(puzzlePath)
|
||||
points, ext = os.path.splitext(filename)
|
||||
points = int(points)
|
||||
puzzle = Puzzle(open(puzzlePath))
|
||||
puzzle = puzzles.Puzzle(open(puzzlePath))
|
||||
puzzles[points] = puzzle
|
||||
|
||||
# build mapping, answers, and summary
|
||||
|
|
@ -123,10 +77,6 @@ for puzzledir in args.puzzledir:
|
|||
summary = {}
|
||||
for points in sorted(puzzles):
|
||||
puzzle = puzzles[points]
|
||||
print(args.seed)
|
||||
print(categoryname)
|
||||
print(points)
|
||||
|
||||
hashmap = hashlib.sha1(args.seed.encode('utf-8'))
|
||||
hashmap.update(categoryname.encode('utf-8'))
|
||||
hashmap.update(str(points).encode('utf-8'))
|
||||
|
|
@ -142,11 +92,11 @@ for puzzledir in args.puzzledir:
|
|||
# write out puzzles
|
||||
for points in sorted(puzzles):
|
||||
puzzle = puzzles[points]
|
||||
puzzledir = os.path.join(categoryname, "content", mapping[points])
|
||||
# test write
|
||||
write_kv_pairs(zf, os.path.join(puzzledir, 'summary.txt'), summary)
|
||||
# build json
|
||||
|
||||
|
||||
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
|
||||
|
|
|
|||
Loading…
Reference in New Issue