✨ Import allows appending to quiz

classquiz/helpers/__init__.py

@@ -111,7 +111,7 @@ async def generate_spreadsheet(quiz_results: dict, quiz: Quiz, player_fields: di
     return storage
 
 
-def handle_import_from_excel(data: BinaryIO, user: User) -> Quiz:
+async def handle_import_from_excel(data: BinaryIO, user: User) -> Quiz:
     try:
         wb = load_workbook(filename=data)
     except KeyError:
@@ -154,6 +154,8 @@ def handle_import_from_excel(data: BinaryIO, user: User) -> Quiz:
         for a, answer in enumerate(answers):
             answers_list.append(ABCDQuizAnswer(answer=answer, right=str(a + 1) in correct_answers))
         questions.append(QuizQuestion(question=question, answers=answers_list, time=str(time)).dict())
+    existing_quiz: Quiz | None = await Quiz.objects.get_or_none(user_id=user, title=title)
+    if existing_quiz is None:
         quiz = Quiz(
             title=title,
             description=description,
@@ -164,6 +166,12 @@ def handle_import_from_excel(data: BinaryIO, user: User) -> Quiz:
             questions=questions,
             imported_from_kahoot=False,
         )
+        await quiz.save()
+    else:
+        existing_quiz.questions = [*existing_quiz.questions, *questions]
+        existing_quiz.updated_at = datetime.now()
+        await existing_quiz.update()
+        quiz = existing_quiz
     return quiz
 
 

classquiz/routers/quiz.py

@@ -250,5 +250,5 @@ async def export_quiz_answers(export_token: str, game_pin: str):
 
 @router.post("/excel-import")
 async def import_from_excel(file: UploadFile = File(), user: User = Depends(get_current_user)) -> Quiz:
-    quiz = handle_import_from_excel(file.file, user)
+    quiz = await handle_import_from_excel(file.file, user)
-    return Quiz.parse_obj((await quiz.save()).dict(exclude={"user_id": ...}))
+    return Quiz.parse_obj(quiz.dict(exclude={"user_id": ...}))