game analysis csv logging

Author: Felixleh

.gitignore

@@ -134,3 +134,6 @@ config_local.py
 
 # ignore local todo files
 todo.txt
+
+# ignore local analysis data
+.data

python-visualizer/config.py

@@ -34,5 +34,4 @@
     from config_local import *
 
 # game analysis paths
-GAME_ANALYSIS_SIZE_PATH = "../.data/size"
-GAME_ANALYSIS_TIME_PATH = "../.data/time"
+GAME_ANALYSIS_DIRECTORY_PATH = "../.data/"

python-visualizer/game_analysis.py

@@ -2,13 +2,22 @@
 import json
 import time
 from datetime import datetime, timedelta
-
+import os
 import websockets
 import requests
 from csv import writer
 
 import config
 
+
+MAX_ROUNDS = 100
+
+def append_to_csv(file_name: str, elements):
+    with open(file_name, 'a+', newline='') as write:
+        csv_writer = writer(write)
+        csv_writer.writerow(elements)
+
+
 client_time = datetime.now()
 time_request_result = requests.get(config.TIME_URL)
 
@@ -28,31 +37,39 @@
 def transform_to_server_time(time_value: datetime):
     return time_value + time_offset
 
+
 async def play():
+
+    if not os.path.exists(config.GAME_ANALYSIS_DIRECTORY_PATH):
+        os.makedirs(config.GAME_ANALYSIS_DIRECTORY_PATH)
+
+    board_size_file_name = config.GAME_ANALYSIS_DIRECTORY_PATH + "sizes" + ".csv"
+    time_file_name = config.GAME_ANALYSIS_DIRECTORY_PATH + "times-" \
+                     + datetime.now().strftime("%Y-%m-%d-%H-%M-%S") + ".csv"
+
     url = config.URL
     key = config.API_KEY
 
     async with websockets.connect(f"{url}?key={key}") as websocket:
-        started = False
+        rounds = 0
         print("Waiting for initial state...", flush=True)
 
         while True:
             state_json = await websocket.recv()
             state = json.loads(state_json)
 
-            if not started:
+            if rounds == 0:
                 print(f"board -- width: %d -- height: %d" % (state["width"], state["height"]))
-                started = True
+                append_to_csv(board_size_file_name, [state["width"], state["height"]])
+            rounds += 1
 
-            if not state["running"]:
+            if not state["running"] or rounds >= MAX_ROUNDS:
                 break
 
             deadline = datetime.strptime(state["deadline"], '%Y-%m-%dT%H:%M:%SZ')
             available_millis = (deadline - transform_to_server_time(datetime.now())) / timedelta(milliseconds=1)
-
             print(f"\tavailable milliseconds: %d" % available_millis)
-
-            time.sleep(0.5)
+            append_to_csv(time_file_name, [available_millis])
 
             action_json = json.dumps({"action": "change_nothing"})
             await websocket.send(action_json)
@@ -61,5 +78,6 @@ async def play():
 while True:
     try:
         asyncio.get_event_loop().run_until_complete(play())
-    except Exception:
+    except Exception as e:
+        print(e)
         print("Connection was interrupted!")