move files

main_util.py → games/game_util.py

@@ -1,7 +1,7 @@
 from fastapi import HTTPException, status
 
-import snake
-import angrybird
+import games.game_snake as game_snake
+import games.game_angrybird as game_angrybird
 
 
 GAME_ANGRY_BIRD = 'angrybird'
@@ -10,7 +10,7 @@ GAME_SNAKE = 'snake'
 
 def nlp(game: str, command: str):
     if game == GAME_ANGRY_BIRD:
-        return angrybird.process(command)
+        return game_angrybird.process(command)
     if game == GAME_SNAKE:
-        return snake.process(command)
-    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)
+        return game_snake.process(command)
+    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 
-import main_util
+import games.game_util as game_util
 
 app = FastAPI()
 
@@ -11,4 +11,4 @@ async def root():
 
 @app.get("/nlp")
 def test(game: str, command: str):
-    return main_util.nlp(game, command)
+    return game_util.nlp(game, command)