From 1e5f774f9183c963a395b134f63bfa09978b243a Mon Sep 17 00:00:00 2001
From: NoeBerdoz <berdoznoe@gmail.com>
Date: Sat, 12 Apr 2025 13:40:51 +0200
Subject: [PATCH] add json response store manager

---
 .env.example                        |  1 +
 .gitignore                          |  2 ++
 config.py                           |  3 ++-
 services/player.py                  | 11 ++++++++-
 utils/storage/__init__.py           |  0
 utils/storage/game_files_manager.py | 35 +++++++++++++++++++++++++++++
 6 files changed, 50 insertions(+), 2 deletions(-)
 create mode 100644 utils/storage/__init__.py
 create mode 100644 utils/storage/game_files_manager.py

diff --git a/.env.example b/.env.example
index 01ea49c..2260d4f 100644
--- a/.env.example
+++ b/.env.example
@@ -1,4 +1,5 @@
 API_URI=
 API_KEY=
 API_TEAM=
+GAME_FILES_DIR=/project_absolute_path/game_files
 GROQ_API_KEY=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
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
index 1fd7de5..995e414 100644
--- a/.gitignore
+++ b/.gitignore
@@ -273,3 +273,5 @@ devenv.lock
 
 # pre-commit
 .pre-commit-config.yaml
+
+game_files
diff --git a/config.py b/config.py
index 9e8a61a..5fdbb5f 100644
--- a/config.py
+++ b/config.py
@@ -5,4 +5,5 @@ from dotenv import load_dotenv
 load_dotenv()
 API_URI = str(os.getenv("API_URI") or "")
 API_KEY = str(os.getenv("API_KEY") or "")
-API_TEAM = str(os.getenv("API_TEAM") or "")
\ No newline at end of file
+API_TEAM = str(os.getenv("API_TEAM") or "")
+GAME_FILES_DIR = str(os.getenv("GAME_FILES_DIR") or "")
\ No newline at end of file
diff --git a/services/player.py b/services/player.py
index 3cae73f..3a142f2 100644
--- a/services/player.py
+++ b/services/player.py
@@ -5,6 +5,7 @@ from typing import Literal, Dict, Any
 import config
 from dto.requests import GameStartRequestDTO, GameDecisionRequestDTO
 from services.julius_baer_api_client import JuliusBaerApiClient
+from utils.storage.game_files_manager import store_game_round_data
 
 
 class Player:
@@ -33,8 +34,11 @@ class Player:
 
         status = ''
         decision = self.make_decision(start_response.client_data)
-        while status not in ['gameover', 'complete']:
 
+        decision_counter = 0
+        store_game_round_data(start_response, decision_counter, str(start_response.session_id))
+
+        while status not in ['gameover', 'complete']:
             payload = GameDecisionRequestDTO(
                 decision=decision,
                 session_id=start_response.session_id,
@@ -45,8 +49,13 @@ class Player:
             print(decision_response.status, decision_response.score)
             status = decision_response.status
             decision = self.make_decision(decision_response.client_data)
+
+            store_game_round_data(decision_response, decision_counter, str(start_response.session_id))
+
+            decision_counter += 1
             time.sleep(1.5)
 
+
     def make_decision(self, client_data: Dict[str, Any]) -> Literal["Accept", "Reject"]:
         # Do your magic!
 
diff --git a/utils/storage/__init__.py b/utils/storage/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/utils/storage/game_files_manager.py b/utils/storage/game_files_manager.py
new file mode 100644
index 0000000..d052d07
--- /dev/null
+++ b/utils/storage/game_files_manager.py
@@ -0,0 +1,35 @@
+import os
+import logging
+import config
+import json
+
+from dto.responses import GameStartResponseDTO, GameDecisionResponseDTO
+
+GAME_FILES_DIR = config.GAME_FILES_DIR
+
+# Define padding for round numbers (e.g., 6 digits for up to 999,999 rounds)
+FOLDER_ROUND_PADDING = 6
+
+def store_game_round_data(response: GameStartResponseDTO | GameDecisionResponseDTO, round_number: int, session_id: str):
+    """
+        Logs structured response data and saves associated client files.
+    """
+    logging.info(f"[+] Storing game round data in {GAME_FILES_DIR}")
+
+    try:
+        padded_round = str(round_number).zfill(FOLDER_ROUND_PADDING)
+        round_folder_name = f"{padded_round}_decision"
+
+        # Construct the directory path: base_dir / session_id / decision_XXXXXX
+        round_dir = os.path.join(GAME_FILES_DIR, str(session_id), round_folder_name)
+        os.makedirs(round_dir, exist_ok=True) # Create the directory structure if it doesn't exist
+
+        json_file_path = os.path.join(round_dir, f"{padded_round}_response.json")
+
+        with open(json_file_path, "w") as json_file:
+            json.dump(response.model_dump_json(), json_file)
+
+        logging.info(f"[+] Successfully saved API response JSON to: {json_file_path}")
+    except Exception as e:
+        logging.error(f"[!] Failed to save API response JSON: {e}")
+