✨ Fixed #51

classquiz/db/models.py

@@ -99,10 +99,6 @@ class TokenData(BaseModel):
 
 
 class PlayGame(BaseModel):
-    """
-    For JWT
-    """
-
     quiz_id: uuid.UUID | str
     description: str
     title: str
@@ -110,6 +106,7 @@ class PlayGame(BaseModel):
     game_id: uuid.UUID
     game_pin: str
     started: bool = False
+    captcha_enabled: bool = False
 
 
 class GamePlayer(BaseModel):

classquiz/routers/quiz.py

@@ -6,7 +6,7 @@ from random import randint
 from classquiz.helpers import get_meili_data
 from fastapi import APIRouter, Depends, HTTPException
 from fastapi.responses import JSONResponse
-from pydantic import ValidationError
+from pydantic import ValidationError, BaseModel
 import bleach
 
 from classquiz.auth import get_current_user
@@ -76,7 +76,7 @@ async def get_public_quiz(quiz_id: str):
 
 
 @router.post("/start/{quiz_id}")
-async def start_quiz(quiz_id: str, user: User = Depends(get_current_user)):
+async def start_quiz(quiz_id: str, captcha_enabled: bool = True, user: User = Depends(get_current_user)):
     try:
         quiz_id = uuid.UUID(quiz_id)
     except ValueError:
@@ -94,11 +94,24 @@ async def start_quiz(quiz_id: str, user: User = Depends(get_current_user)):
         game_id=uuid.uuid4(),
         title=quiz.title,
         description=quiz.description,
+        captcha_enabled=captcha_enabled,
     )
     await redis.set(f"game:{str(game.game_pin)}", (game.json()), ex=18000)
     return {**quiz.dict(exclude={"id"}), **game.dict(exclude={"questions"})}
 
 
+class CheckIfCaptchaEnabledResponse(BaseModel):
+    enabled: bool
+
+
+@router.get("/play/check_captcha/{game_pin}", response_model=CheckIfCaptchaEnabledResponse)
+async def check_if_captcha_enabled(game_pin: str):
+    game = await redis.get(f"game:{game_pin}")
+    if game is None:
+        return JSONResponse(status_code=404, content={"detail": "game not found"})
+    return CheckIfCaptchaEnabledResponse(**{"enabled": json.loads(game)["captcha_enabled"]})
+
+
 @router.get("/join/{game_pin}")
 async def get_game_id(game_pin: str):
     redis_res = (await redis.get(f"game:{game_pin}")).decode()

classquiz/socket_server/__init__.py

@@ -13,20 +13,21 @@ settings = settings()
 @sio.event
 async def join_game(sid, data):
     async with aiohttp.ClientSession() as session:
-        try:
-            async with session.post(
-                "https://hcaptcha.com/siteverify",
-                data={"response": data["captcha"], "secret": settings.hcaptcha_key},
-            ) as resp:
-                resp_data = await resp.json()
-                if not resp_data["success"]:
-                    print("CAPTCHA FAILED")
-                    return
-        except KeyError:
-            print("CAPTCHA FAILED")
+        redis_res = await redis.get(f"game:{data['game_pin']}")
+        if json.loads(redis_res)["captcha_enabled"]:
+            try:
+                async with session.post(
+                    "https://hcaptcha.com/siteverify",
+                    data={"response": data["captcha"], "secret": settings.hcaptcha_key},
+                ) as resp:
+                    resp_data = await resp.json()
+                    if not resp_data["success"]:
+                        print("CAPTCHA FAILED")
+                        return
+            except KeyError:
+                print("CAPTCHA FAILED")
 
-            return
-    redis_res = await redis.get(f"game:{data['game_pin']}")
+                return
     if redis_res is None:
         await sio.emit("game_not_found", room=sid)
     else: