From 746605fc62f12b3c1ad62ff8c5386bc3a0e8c6ba Mon Sep 17 00:00:00 2001
From: Daniel <d.brummerloh@gmail.com>
Date: Wed, 22 Nov 2023 14:51:08 +0100
Subject: [PATCH] change dir logic

---
 src/dirmanager.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/src/dirmanager.py b/src/dirmanager.py
index 4fa4844..dfc149d 100644
--- a/src/dirmanager.py
+++ b/src/dirmanager.py
@@ -21,18 +21,19 @@ class DirManager:
         self.tests_dir = tests_dir
         self.session_id = session_id
 
-        self.get_all_dirs()
-
-    def get_all_dirs(self):
-        self.session_dir = self.tests_dir / f"test-{self.session_id}"
-        self.session_sub_dirs = self.get_subdirs(self.session_dir)
+        self.dirs = self._get_all_dirs()
 
     def create_all_dirs(self):
         self.create_dirs(self.tests_dir, exist_ok=True)
-        self.create_dirs(self.session_dir)
-        self.create_dirs(self.session_sub_dirs)
+        self.create_dirs(self.dirs)
 
-    def get_subdirs(self, session_dir: Path):
+    def _get_all_dirs(self):
+        dirs = {}
+        dirs["session"] = self.tests_dir / f"test-{self.session_id}"
+        dirs.update(self._get_subdirs(session_dir=dirs["session"]))
+        return dirs
+
+    def _get_subdirs(self, session_dir: Path):
         return {
             "records": session_dir / Path("records"),
             "states": session_dir / Path("states"),