polish

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Body
+from fastapi import FastAPI
 from routers import root, snake, watermelon
 
 app = FastAPI()

routers/game_util.py

@@ -16,17 +16,20 @@ def read(game: str, code: str, command: str):
         raise HTTPException(status.HTTP_511_NETWORK_AUTHENTICATION_REQUIRED)        
     
     url = f'https://{os.getenv("hfs_space")}.hf.space/games/{game}'
-    print(url)
     myobj = {
         'command': command,
     }
-    headers = {'Authorization': f'Bearer {os.getenv("hfs_token")}'}
+    headers = {
+        'Authorization': f'Bearer {os.getenv("hfs_token")}',
+    }
 
     x = requests.post(url, json=myobj, headers=headers)
-    if x.status_code == status.HTTP_200_OK:
-        return json.loads(x.text)
-    else:
-        raise HTTPException(status_code=x.status_code, detail=json.loads(x.text)['detail'])
+    if x.status_code != status.HTTP_200_OK:
+        raise HTTPException(
+            status_code=x.status_code, 
+            detail=json.loads(x.text)['detail'],
+        )
+    return json.loads(x.text)
 
 
 def update(game: str, code: str, command: str):