add reset

main.py

@@ -3,5 +3,5 @@ from routers import root, snake, watermelon
 
 app = FastAPI()
 app.include_router(root.router)
-app.include_router(snake.router)
 app.include_router(watermelon.router)
+app.include_router(snake.router)

routers/game_util.py

@@ -10,14 +10,21 @@ debug = False
 
 
 def op(game: str, code: str, crud: str, command: str, result="", desc="") -> Any:
-    if crud not in "crud":
-        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+    # check passcode
     _code = os.getenv("hfs_code")
     if _code is None:
         raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
     if code != _code:
         raise HTTPException(status.HTTP_511_NETWORK_AUTHENTICATION_REQUIRED)        
 
+    # handle reset
+    if crud == 'reset':
+        raise HTTPException(status_code=status.HTTP_501_NOT_IMPLEMENTED)
+    
+    # handle c/r/u/d
+    if crud not in "crud":
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+    
     # url = f'https://{os.getenv("hfs_space")}.hf.space/games/{game}'
     url = f'https://{os.getenv("hfs_space")}.hf.space/games/{game}/nlp/{crud}'
     if debug:

routers/watermelon.py

@@ -25,3 +25,8 @@ def update(code: str=Body(), command: str=Body()):
 @router.post("/d")
 def delete(code: str=Body(), command: str=Body()):
     return game_util.op(GAME, code, "d", command)
+
+
+@router.post("/reset")
+def reset(code: str=Body()):
+    return game_util.op(GAME, code, "reset", "")