Add support for an argument to the /hunt command

[turbot] / turbot / hunt.py

diff --git a/turbot/hunt.py b/turbot/hunt.py
index 77c179771a17cd7e01eed6f5d3230bf554a225b6..6d44662b3d44f53e8ca5c654d4c5ca6177e09673 100644 (file)
--- a/turbot/hunt.py
+++ b/turbot/hunt.py
@@ -23,13 +23,21 @@ def find_hunt_for_hunt_id(turb, hunt_id):
     else:
         return None
 
-def hunt_blocks(turb, hunt):
+def hunt_blocks(turb, hunt, puzzle_status='unsolved'):
     """Generate Slack blocks for a hunt
 
     The hunt argument should be a dictionary as returned from the
-    database. The return value can be used in a Slack command
-    expecting blocks to provide all the details of a hunt, (puzzles,
-    their state, solution, links to channels and sheets, etc.).
+    database.
+
+    Option 'puzzle_status' indicates which puzzles to include. If
+    either 'solved' or 'unsolved' only puzzles with that status from
+    the hunt will be included in the result. If any other value, all
+    puzzles in the hunt will be included.
+
+    The return value can be used in a Slack command expecting blocks to
+    provide all the details of a hunt, (puzzles, their state,
+    solution, links to channels and sheets, etc.).
+
     """
 
     name = hunt['name']
@@ -44,6 +52,10 @@ def hunt_blocks(turb, hunt):
     )
     puzzles = response['Items']
 
+    # Filter the set of puzzles according the the requested puzzle_status
+    if puzzle_status in ('solved', 'unsolved'):
+        puzzles = [p for p in puzzles if p['status'] == puzzle_status]
+
     # Compute the set of rounds across all puzzles
     rounds = set()
     for puzzle in puzzles: