some updates to package-puzzles

package-puzzles

@@ -12,15 +12,15 @@ import markdown
 import random
 import zipfile
 
-messageChars = b'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
+import puzzles
-
-def djb2hash(buf):
-    h = 5381
-    for c in buf:
-        h = ((h * 33) + c) & 0xffffffff
-    return h
 
 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:
+if __name__ == '__main__':        
-    def __init__(self, stream):
+    parser = argparse.ArgumentParser(description='Build a puzzle category')
-        self.message = bytes(random.choice(messageChars) for i in range(20))
+    parser.add_argument('seed', help='contest seed')
-        self.fields = {}
+    parser.add_argument('puzzledir', nargs='+', help='Directory of puzzle source')
-        self.answers = []
+    parser.add_argument('outdir', help='Output directory')
-        self.hashes = []
+    args = parser.parse_args()
-
-        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')
+    for puzzledir in args.puzzledir:
-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 = {}
 
@@ -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
+    #        # open and gather some state
-#        zf = zipfile.ZipFile(zipfilename, 'a')
+    #        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])
+            puzzledir = os.path.join(categoryname, 'content', mapping[points])
-        # test write
+            # build/write json
-        write_kv_pairs(zf, os.path.join(puzzledir, 'summary.txt'), summary)
+            ziphandle.writestr(filename, filehandle.read())
-        # build json
+            zf.writestr(os.path.join(puzzledir, 'puzzle.json'), \
-
+                                    json.dumps(puzzle.publish()))
-
+            # write associated files
 
 #vim:py