ai-snake-lab 0.4.6__py3-none-any.whl → 0.4.8__py3-none-any.whl

ai_snake_lab/ai/ReplayMemory.py

@@ -14,10 +14,11 @@ import os
 from collections import deque
 import random
 import sqlite3, pickle
+import tempfile
+import shutil
 
 from ai_snake_lab.constants.DReplayMemory import MEM_TYPE
-from ai_snake_lab.constants.DFile import DFile
-from ai_snake_lab.constants.DDir import DDir
+from ai_snake_lab.constants.DDef import DDef
 
 
 class ReplayMemory:
@@ -31,11 +32,6 @@ class ReplayMemory:
         self.max_states = 15000
         self.max_shuffle_games = 40
         self.max_games = 500
-        self.db_file = os.path.join(DDir.AI_SNAKE_LAB, DDir.DB, DFile.REPLAY_DB)
-
-        # Delete the replay memory file, if it exists
-        if os.path.exists(self.db_file):
-            os.remove(self.db_file)
 
         if self._mem_type == MEM_TYPE.SHUFFLE:
             # States are stored in a deque and a random sample will be returned
@@ -46,13 +42,33 @@ class ReplayMemory:
             # A complete game will be returned
             self.cur_memory = []
 
+        # Get a temporary directory for the DB file
+        self._tmpfile = tempfile.NamedTemporaryFile(suffix=DDef.DOT_DB, delete=False)
+        self.db_file = self._tmpfile.name
+
         # Connect to SQLite
         self.conn = sqlite3.connect(self.db_file, check_same_thread=False)
+
+        # Get a cursor
         self.cursor = self.conn.cursor()
+
+        # We don't need the file handle anymore
+        self._tmpfile.close()
+
+        # Intialize the schema
         self.init_db()
 
-    def __len__(self):
-        return len(self.memories)
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def __del__(self):
+        try:
+            self.close()
+        except Exception:
+            pass  # avoid errors on interpreter shutdown
 
     def append(self, transition):
         """Add a transition to the current game."""
@@ -75,7 +91,12 @@ class ReplayMemory:
 
     def close(self):
         """Close the database connection."""
-        self.conn.close()
+        if getattr(self, "conn", None):
+            self.conn.close()
+            self.conn = None
+        if getattr(self, "db_file", None) and os.path.exists(self.db_file):
+            os.remove(self.db_file)
+            self.db_file = None
 
     def get_random_game(self):
         """Return a random full game from the database."""

ai_snake_lab/constants/DDef.py

@@ -15,4 +15,5 @@ class DDef(ConstGroup):
     """Defaults"""
 
     APP_TITLE: str = "AI Snake Game Lab"
+    DOT_DB: str = ".db"
     MOVE_DELAY: float = 0.0

ai_snake_lab/constants/DDir.py

@@ -15,5 +15,4 @@ class DDir(ConstGroup):
     """Directories"""
 
     AI_SNAKE_LAB: str = "ai_snake_lab"
-    DB: str = "db"
     UTILS: str = "utils"

ai_snake_lab/constants/DFile.py

@@ -15,4 +15,3 @@ class DFile(ConstGroup):
     """Files"""
 
     CSS_FILE: str = "AISim.tcss"
-    REPLAY_DB: str = "replay_mem.db"

ai_snake_lab/ui/AISim.py

@@ -63,7 +63,7 @@ class AISim(App):
     """A Textual app that has an AI Agent playing the Snake Game."""
 
     TITLE = DDef.APP_TITLE
-    CSS_PATH = os.path.join(DDir.UTILS, DFile.CSS_FILE)
+    CSS_PATH = DFile.CSS_FILE
 
     ## Runtime values
     # Current epsilon value (degrades in real-time)

{ai_snake_lab-0.4.6.dist-info → ai_snake_lab-0.4.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-snake-lab
-Version: 0.4.6
+Version: 0.4.8
 Summary: Interactive reinforcement learning sandbox for experimenting with AI agents in a classic Snake Game environment.
 License: GPL-3.0
 License-File: LICENSE

{ai_snake_lab-0.4.6.dist-info → ai_snake_lab-0.4.8.dist-info}/RECORD

@@ -1,15 +1,15 @@
 ai_snake_lab/ai/AIAgent.py,sha256=p_O7zC15s_tBcI7oqu60X2-3AbrzAWlGXWkrLVC4UsQ,3005
 ai_snake_lab/ai/AITrainer.py,sha256=ssX6B03yZLEKhNCJv9D83iFAEEhU3_XbW4sA0z5QMRM,3304
 ai_snake_lab/ai/EpsilonAlgo.py,sha256=Q2U_Ow28ZRQn1hlLybWEhB2Gn101x99mrQWupAooRjk,2142
-ai_snake_lab/ai/ReplayMemory.py,sha256=dtF6CkYWBvjqnKPdNA_0Dgr1DWQYAE-KMpB9IIxkoCk,4020
+ai_snake_lab/ai/ReplayMemory.py,sha256=yX_dTEkzXxJyK6OjWHDUDoWHTB2sah1Vgg-X9MpK4e4,4529
 ai_snake_lab/ai/models/ModelL.py,sha256=hK7MoPyIF1_K43R_9xW_MYuaweQGo9qSMUIVrHDQZnQ,1249
 ai_snake_lab/ai/models/ModelRNN.py,sha256=Ky63BUqJEmwe1PbM4gtjvqd7SmSy_2XA1n-yu8DNexI,1104
 ai_snake_lab/constants/DDb4EPlot.py,sha256=2g8SdDVZWrTVSTw-rpO84OrNfl7ToBYtf6J7sYB7q8w,442
-ai_snake_lab/constants/DDef.py,sha256=XABLL9P_WUkCiNr2L-phiNL6cnxc_E0uD4vW6WhtZs0,378
-ai_snake_lab/constants/DDir.py,sha256=e8pNvZJpR4R6Cm3VRGljry_g-GAODYkz1ZrU8OudXuQ,395
+ai_snake_lab/constants/DDef.py,sha256=InS6P0tUCvwZE_xiQtcxrrmLrts0HpO6CpU75UNXVnQ,402
+ai_snake_lab/constants/DDir.py,sha256=6XiS6OTsqL-3AhYyZAZApziMFpw8kNd2QY4v0GLrb-Q,376
 ai_snake_lab/constants/DEpsilon.py,sha256=HuhMGxy3WIebcFQ2XZvlOpl03JwPYhHZdO40Bzph9yM,420
 ai_snake_lab/constants/DFields.py,sha256=0MV21hLy-Ojnn4Qa3C9_dQLPjC-3VYBjes_YU5V1Bio,521
-ai_snake_lab/constants/DFile.py,sha256=jYnYgrMqw3BMa3XEJw5XtC5buHfE0v7QFLv25BKNHW0,378
+ai_snake_lab/constants/DFile.py,sha256=VAVH5qf2sFY62J6jcIJhfDnbGuEGpndqnwZr-ysFb7E,341
 ai_snake_lab/constants/DLabels.py,sha256=Nv7vCs8pxpF6Px5FD3JFkI7KEPWy6G8wLqzv31_nXxw,1501
 ai_snake_lab/constants/DLayout.py,sha256=XhRMpOqgzOkL0kcFz4pfS-HuVsYZZLBC6GIAq-E1_Ik,1616
 ai_snake_lab/constants/DModelL.py,sha256=VzAK5Uqkb0ovMPKTo_DJ2lQmIJkpRQyF-z3g3y4-j20,506
@@ -20,12 +20,12 @@ ai_snake_lab/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 ai_snake_lab/game/GameBoard.py,sha256=iNgain-8YQg2Ky0GkmGX-ViVmL8P7zPiZYU6Qwrp9Cc,8465
 ai_snake_lab/game/GameElements.py,sha256=9vcNH2YWAzYyBZosvj1DzO6ZmUCUMUd8MaIhORWQ8go,470
 ai_snake_lab/game/SnakeGame.py,sha256=ADJSVD-bXBruSHf0rzQFh2P9hPWUAnCkQ95B-RFk1vI,6506
-ai_snake_lab/ui/AISim.py,sha256=_XUJtiQ_CMWB7beiXq3o9wgmIL-3QGwjHxy8KbSmd_c,15173
+ai_snake_lab/ui/AISim.py,sha256=xAesy3utDuGaVzdTGLYeqIB4bZ6kg4EL16CMvRzMEOg,15147
+ai_snake_lab/ui/AISim.tcss,sha256=XnhqjuxtOiNQHDyXFrbAYSlCw-6ch4e1o52nbytoeF0,2118
 ai_snake_lab/ui/Db4EPlot.py,sha256=pcEb0ydXNX2wL0EFBSrqhIoTEhryk4GD0Ua8FFEaZHY,5352
-ai_snake_lab/utils/AISim.tcss,sha256=XnhqjuxtOiNQHDyXFrbAYSlCw-6ch4e1o52nbytoeF0,2118
 ai_snake_lab/utils/ConstGroup.py,sha256=ZYyQxFd9PudBUZmc_NsNvWCp___utOe1MptqD3eyVH8,1174
-ai_snake_lab-0.4.6.dist-info/METADATA,sha256=shfnaQS9q3tefXcTuISZbAJRVCGUQ79ivMNrHgDQkKE,3689
-ai_snake_lab-0.4.6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-ai_snake_lab-0.4.6.dist-info/entry_points.txt,sha256=ThFx-0yPF5pdwOXUbEo95FEXLUUbOcjfhgMv67PHuIw,59
-ai_snake_lab-0.4.6.dist-info/licenses/LICENSE,sha256=f-FHFu0xzHH8O_mvKTw2jUZKhTpw6obpmVOI9rnpKeU,35151
-ai_snake_lab-0.4.6.dist-info/RECORD,,
+ai_snake_lab-0.4.8.dist-info/METADATA,sha256=DbfR2kfsRDLU6ZEJ5cMzYgqc3cphigWSjgFYp2bSEpw,3689
+ai_snake_lab-0.4.8.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+ai_snake_lab-0.4.8.dist-info/entry_points.txt,sha256=ThFx-0yPF5pdwOXUbEo95FEXLUUbOcjfhgMv67PHuIw,59
+ai_snake_lab-0.4.8.dist-info/licenses/LICENSE,sha256=f-FHFu0xzHH8O_mvKTw2jUZKhTpw6obpmVOI9rnpKeU,35151
+ai_snake_lab-0.4.8.dist-info/RECORD,,