support worksheets with multiple tasks in them

mathgenerator/mathgen.py

@@ -11,10 +11,28 @@ def genById(id):
     return (generator())
 
 
-def makePdf(id, count):
-    generator = genList[id][2]
-    title = genList[id][1]
-    ws = Worksheet(title=title)
+def make_worksheet(title):
+    return Worksheet(title=title)
+
+
+def write_pdf(worksheet):
+    worksheet.write_pdf()
+
+def add_section_title(worksheet, task_id):
+    title = genList[task_id][1]
+    worksheet.add_instruction(title)
+
+def add_task_to_worksheet(worksheet, task_id, count):
+    generator = genList[task_id][2]
     for i in range(count):
-        ws.add_problem(generator()[0])
-    ws.write_pdf()
+        worksheet.add_problem(generator()[0])
+
+def add_section_with_task_to_worksheet(worksheet, task_id, count):
+    add_section_title(worksheet, task_id)
+    add_task_to_worksheet(worksheet, task_id, count)
+
+def make_pdf(task_id, count):
+    title = genList[task_id][1]
+    worksheet = make_worksheet(title)
+    add_task_to_worksheet(worksheet, task_id, count)
+    write_pdf(worksheet)