From bd3bd7de2fdb6cdb181a9b02b627c98c712f3739 Mon Sep 17 00:00:00 2001
From: Karan Jayachandra <karan.jayachandra@nxp.com>
Date: Thu, 4 Apr 2024 21:06:37 +0200
Subject: Generating test data possible

---
 src/match_up/__init__.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/match_up/__init__.py')

diff --git a/src/match_up/__init__.py b/src/match_up/__init__.py
index 7b30c1b..c9c3d0b 100644
--- a/src/match_up/__init__.py
+++ b/src/match_up/__init__.py
@@ -8,6 +8,12 @@ class PlaySession:
     def __init__(self, database_location: PurePath):
         self.data = PlayerDatabase(database_location)
 
+    def end(self):
+        self.data.close()
+
+    def __del__(self):
+        self.end()
+
 
 def main():
     generate_test_database()
@@ -15,5 +21,5 @@ def main():
     for player in s.data.get_players():
         print(player)
     sleep(2)
-    s.data._database.close()
+    s.end()
     remove_test_database()
-- 
cgit v1.3.1