create payload of json string and then save it

server/Files.py

@@ -1,6 +1,7 @@
 import aiofiles.os
 import aiofiles
 import os
+import inspect
 
 async def read_file(path: str, callback=None):
   if not await aiofiles.os.path.exists(path):
@@ -8,7 +9,10 @@ async def read_file(path: str, callback=None):
 
   async with aiofiles.open(path, "r") as f:
     if callback:
-      return await callback(f)
+      result = callback(f)
+      if inspect.isawaitable(result):
+        return await result
+      return result
     return await f.read()
 
 
@@ -19,6 +23,8 @@ async def save_file(path: str, data, callback=None, *args, **kwargs):
 
   async with aiofiles.open(path, "w") as f:
     if callback:
-      await callback(data, f, *args, **kwargs)
+      result = callback(data, f, *args, **kwargs)
+      if inspect.isawaitable(result):
+        await result
     else:
       await f.write(data)

server/storage/LocalStorage.py

@@ -153,27 +153,25 @@ class LocalStorage:
       True if game_board.winner_snake_names and "me" in game_board.winner_snake_names else False
     )
 
-    await save_file(save_file_path, {
-        "winner": game_board.winner_snake_names,
-        "game": {
-          "url": game_board.url,
-          "id": game_board.id,
-          "final_turns": game_board.turn,
-          "map": game_board.map,
-          "type": game_type,
-          "ruleset": game_board.ruleset,
-        },
-        "moves": game_board.turns,
-        "snake": {
-          "type": game_board.snake_class.__class__.__name__,
-          "calculations": game_board.snake_class.get_history(),
-        },
-        "dataset": dataset,
+    payload = {
+      "winner": game_board.winner_snake_names,
+      "game": {
+        "url": game_board.url,
+        "id": game_board.id,
+        "final_turns": game_board.turn,
+        "map": game_board.map,
+        "type": game_type,
+        "ruleset": game_board.ruleset,
       },
-      callback=json.dump,
-      indent=2,
-      ensure_ascii=False,
-    )
+      "moves": game_board.turns,
+      "snake": {
+        "type": game_board.snake_class.__class__.__name__,
+        "calculations": game_board.snake_class.get_history(),
+      },
+      "dataset": dataset,
+    }
+
+    await save_file(save_file_path, json.dumps(payload, indent=2, ensure_ascii=False))
 
   def cleanup(self):
     pass