Upload workbook_server.py

workbook_server.py

@@ -25,9 +25,20 @@ def get_db():
 def get_chapters():
     db = get_db()
     cursor = db.cursor()
-    cursor.execute("SELECT chapter, chapter_name FROM chapter")
+    cursor.execute("""
+        SELECT c.chapter, c.chapter_name, COUNT(q.question) as n_questions
+        FROM chapter c
+        LEFT JOIN question q ON c.chapter = q.chapter
+        GROUP BY c.chapter, c.chapter_name
+    """)
     chapters = cursor.fetchall()
-    chapter_list = [{"chapter_number": row[0], "chapter_name": row[1]} for row in chapters]
+    chapter_list = [
+        {
+            "chapter_number": row[0],
+            "chapter_name": row[1],
+            "n_questions": row[2]
+        } for row in chapters
+    ]
     return JSONResponse(content=chapter_list)
 
 class ChapterRequest(BaseModel):