Implement a /round command

[turbot] / turbot / round.py

diff --git a/turbot/round.py b/turbot/round.py
index 3c4042be6a6d0e94094ba3282c8cc0198ae403b3..3e12f5c17eb560c0650b1d8c1c22435a94f2fb1f 100644 (file)
--- a/turbot/round.py
+++ b/turbot/round.py
@@ -1,7 +1,7 @@
-from turbot.puzzle import puzzle_block
+from turbot.puzzle import puzzle_blocks
 from turbot.blocks import section_block, text_block
 
-def round_blocks(round, puzzles):
+def round_blocks(round, puzzles, omit_header=False):
     """Generate Slack blocks for a round
 
     The 'round' argument should be the name of a round as it appears
@@ -15,17 +15,20 @@ def round_blocks(round, puzzles):
     channels and sheets, etc.).
     """
 
-    round_text = "*Round: {}*".format(round)
+    if omit_header:
+        blocks = []
+    else:
+        round_text = "*Round: {}*".format(round)
 
-    blocks = [
-        section_block(text_block(round_text)),
-    ]
+        blocks = [
+            section_block(text_block(round_text)),
+        ]
 
     for puzzle in puzzles:
         if 'rounds' not in puzzle:
             continue
         if round not in puzzle['rounds']:
             continue
-        blocks.append(puzzle_block(puzzle))
+        blocks += puzzle_blocks(puzzle)
 
     return blocks