Incorporate pat suggestions

tools/devel-server.py

@@ -151,7 +151,7 @@ you are a fool.
             # List all point values in a category
             title = "Puzzles in category `{}`".format(parts[2])
             body.write("<ul>")
-            for points in cat.pointvals:
+            for points in cat.pointvals():
                 body.write('<li><a href="/puzzles/{cat}/{points}/">puzzles/{cat}/{points}/</a></li>'.format(cat=parts[2], points=points))
             body.write("</ul>")
         elif len(parts) == 4:

tools/moth.py

@@ -29,6 +29,13 @@ def pushd(newdir):
     finally:
         os.chdir(curdir)
 
+
+def loadmod(name, path):
+    abspath = os.path.abspath(path)
+    loader = importlib.machinery.SourceFileLoader(name, abspath)
+    return loader.load_module()
+
+
 # Get a big list of clean words for our answer file.
 ANSWER_WORDS = [w.strip() for w in open(os.path.join(os.path.dirname(__file__),
                                                      'answer_words.txt'))]
@@ -112,9 +119,7 @@ class Puzzle:
 
     def read_directory(self, path):
         try:
-            fn = os.path.abspath(os.path.join(path, "puzzle.py"))
+            puzzle_mod = loadmod("puzzle", os.path.join(path, "puzzle.py"))
-            loader = importlib.machinery.SourceFileLoader('puzzle_mod', fn)
-            puzzle_mod = loader.load_module()
         except FileNotFoundError:
             puzzle_mod = None
 
@@ -255,22 +260,24 @@ class Category:
     def __init__(self, path, seed):
         self.path = path
         self.seed = seed
-        self.pointvals = []
         self.catmod = None
 
-        if os.path.exists(os.path.join(path, 'category.py')):
+        try:
-            with pushd(path):
+            self.catmod = loadmod('category', os.path.join(path, 'category.py'))
-                fn = os.path.abspath('category.py')
+        except FileNotFoundError:
-                loader = importlib.machinery.SourceFileLoader('category', fn)
+            self.catmod = None
-                self.catmod = loader.load_module()
+
-                self.pointvals.extend(self.catmod.pointvals())
+    def pointvals(self):
+        if self.catmod:
+            with pushd(self.path):
+                pointvals = self.catmod.pointvals()
         else:
-            for fpath in glob.glob(os.path.join(path, "[0-9]*")):
+            pointvals = []
+            for fpath in glob.glob(os.path.join(self.path, "[0-9]*")):
                 pn = os.path.basename(fpath)
                 points = int(pn)
-                self.pointvals.append(points)
+                pointvals.append(points)
-
+        return sorted(pointvals)
-        self.pointvals.sort()
 
     def puzzle(self, points):
         puzzle = Puzzle(self.seed, points)
@@ -283,5 +290,5 @@ class Category:
         return puzzle
 
     def __iter__(self):
-        for points in self.pointvals:
+        for points in self.pointvals():
             yield self.puzzle(points)